3.10 练习题1:模块导入（登录方法）
以登录博客园为案例https://passport.cnblogs.com/user/signin
一、登录方法封装
1.我们可以把登录写成一个登录类，里面写个登录的方法，保存文件为login_pub.py
# coding:utf-8
'''
这里写了一个登录博客园的类，登录博客园方法
'''
class Login_Blog(): '''登录类封装''' def __init__(self, driver): '''初始化driver参数''' self.driver = driver def input_user(self, username): '''输入用户名''' self.driver.find_element_by_id("input1").clear() self.driver.find_element_by_id("input1").send_keys(username) def input_psw(self,psw): '''输入密码''' self.driver.find_element_by_id("input2").clear() self.driver.find_element_by_id("input2").send_keys(psw) def click_button(self): '''点击登录按钮''' self.driver.find_element_by_id("signin").click() def login(self, username, psw): '''登录公共方法''' self.input_user(username) self.input_psw(psw) self.click_button()
2.调用登录公共方法
# coding:utf-8
from selenium import webdriver
import unittest from login_pub import Login_Blog login_url = "https://passport.cnblogs.com/user/signin"
class TetsLogin(unittest.TestCase): def setUp(self): self.driver = webdriver.Firefox() self.driver.get(login_url) def tearDown(self): self.driver.quit() def test_login(self): # 调用登录类里面的login方法 Login_Blog(self.driver).login("xxx", "111") self.driver.find_element() # 后面接着的操作省略了
if __name__ == "__main__": unittest.main()