模块化驱动测试实例

from selenium import webdriver
driver=webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")

def login():
    driver.find_element_by_xpath("//input[@name='email']").clear()
    driver.find_element_by_xpath("//input[@name='email']").send_keys("username")
    driver.find_element_by_name("password").clear()
    driver.find_element_by_name("password").send_keys("password")
    driver.find_element_by_id("dologin").click()
def logout():
    driver.find_element_by_link_text("退出").click()
    driver.quit()
login()
logout()

将登录的操作封装到login()函数中，将退出的操作封装到logout()函数中，用到时调用即可

public.py

class Login():
    def user_login(self,driver):
        driver.find_element_by_xpath("//input[@name='email']").clear()
        driver.find_element_by_xpath("//input[@name='email']").send_keys("username")
        driver.find_element_by_name("password").clear()
        driver.find_element_by_name("password").send_keys("password")
        driver.find_element_by_id("dologin").click()
    def user_logout(self,driver):
        driver.find_element_by_link_text("退出").click()
        driver.quit()

mailTest.py

from selenium import webdriver
from public import Login
driver=webdriver.Firefox()
driver.implicitly_wait(10)
driver.get("http://www.126.com")
Login().user_login(driver)
Login().user_logout(driver)

现在是将复用的代码独立保存，使用时导入调用即可