3.10 练习题1:模块导入（登录方法）

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()