Python+selenium實現登錄腳本

import unittest
from selenium import webdriver
from time import sleep
class LoginCase(unittest.TestCase):
 def setUp(self):
 self.dr = webdriver.Firefox()
 self.dr.maximize_window()
 self.dr.get('http://www.imooc.com/') #cnblog登錄頁面
 sleep(3)
 self.dr.find_element_by_id("js-signin-btn").click()
 sleep(3)

 #定義登錄方法
 def login(self, username, password):
 self.dr.find_element_by_name("email").send_keys(username)
 sleep(2)
 self.dr.find_element_by_name("password").send_keys(password)
 self.dr.find_element_by_css_selector("[class='btn-red btn-full xa-login']").click()
 def test_login_success(self):
 self.login('[e-mail]', 'gjhkjjhghj') # 正確用戶名和密碼
 sleep(3)
 self.dr.get_screenshot_as_file("D:\\login_success.jpg") # 截圖 可自定義截圖后的保存位置和圖片命名

 def test_login_pwd_error(self):
 self.login('[e-mail]', '123456')
 sleep(3)
 error_message = self.dr.find_element_by_id('signin-globle-error').text
 self.assertIn('密碼錯誤', error_message)
 self.dr.get_screenshot_as_file("D:\\login_success.jpg")

 def tearDown(self):
 self.dr.quit()
 sleep(2)
 print('自動測試完畢！')
if __name__=='__main__':
 unittest.main()

最后截圖保存到本地