python+selenium个人学习笔记11-登录封装与调用

登录封装与调用

一、登录封装

把登录写成一个类，里面写登录的方法，保存文件为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.anenda.com" unittest.main()