053模块化驱动测试

# -*- coding:utf-8 -*-
from selenium import webdriver
from time import sleep

class login():
    def user_login(self,driver):
        driver.find_element_by_css_selector('div.main-select>ul>:nth-child(2)>a').click()
        driver.find_element_by_css_selector('#all').clear()
        driver.find_element_by_css_selector('#all').send_keys('手机号')    #账号
        driver.find_element_by_css_selector('#password-number').clear()
        driver.find_element_by_css_selector('#password-number').send_keys('密码')
        driver.find_element_by_css_selector('div.main-process-login>form>div>:nth-child(6)>div>button').click()
    def user_logout(self,driver):
        sleep(2)
        driver.quit()
if __name__ == '__main__':
    driver = webdriver.Chrome()
    driver.get('https://passport.csdn.net/login')
    driver.implicitly_wait(3)
    login().user_login(driver)
    sleep(2)
    login().user_logout(driver)