python+selenium-登錄封裝與調用

一、登錄封裝

把登錄寫成一個類，里面寫登錄的方法，保存文件為loginclass.py

class login_go():
    u'''登錄類封裝'''
    def __init__(self,driver):
        u'''初始化driver參數'''
        self.driver = driver

    def login(self,username,password):
        u'''輸入用戶名和密碼,點擊登錄'''
        self.driver.find_element_by_id("liger-textbox-user").clear()
        self.driver.find_element_by_id("liger-textbox-user").send_keys(username)
        self.driver.find_element_by_id("liger-textbox-pwd_old").clear()
        self.driver.find_element_by_id("liger-textbox-pwd").clear()
        self.driver.find_element_by_id("liger-textbox-pwd").send_keys(password)
        self.driver.find_element_by_id("go").click()

二、調用

from selenium import webdriver
import unittest
from loginclass import login_go
from time import sleep
class Login(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(10)
        self.driver.maximize_window()
        self.driver.get(login_url)
    def tearDown(self):
        self.driver.quit()
    def case_login(self):
        login_go(self.driver).login("chen","chen")#調用login方法
        sleep(3)
        result1 = self.driver.find_element_by_xpath(".//*[@id='l-topmenu-r-bottm']/span[2]").text
        print(result1)
        result2 = "歡迎您"
        self.assertIn(result2,result1,msg="失敗原因：%s中沒有發現%s"%(result1,result2))
        sleep(2)
if __name__ == "__main__":
    login_url = "http://www.abc.com"
    unittest.main()