大部分系统现在都有登录页面，本文主要尝试写一个登录的测试脚本，及另一个脚本调用它登录测试已登录的页面模块。
目标：
登录脚本：从excel里获取登录的测试数据（包括异常测试）→执行登录脚本→输出是否通过
A模块测试脚本：调用登录脚本，从excel里获取正确登录的测试数据→执行自动登录脚本→执行模块测试用例→输出是否通过
登录脚本auto.py：
# coding:utf-8
from selenium import webdriver
import time
import xlrd
from xlutils.copy import copy class login_test: def __init__(self, x, y, css):
self.x = x
self.y = y
self.css = css
# 函数功能：获取数据测试登录
def auto_login(self):
global driver, workbook
# 打开excel，初始化excel参数
workbook = xlrd.open_workbook(r'F:\test.xlsx') login = workbook.sheet_names()[1] sheet2 = workbook.sheet_by_name(login)
# 取excel数据
username = sheet2.cell(self.x, self.y).value
password = sheet2.cell(self.x, self.y + 1).value
ctype = sheet2.cell(self. x, self.y + 1).ctype if ctype == 2: # 若是数字则转化为整型，这一步可避免报错类型
password = int(password) # 初始化一个火狐浏览器实例：driver , 打开浏览器访问
driver = webdriver.Ie()
driver.maximize_window() # 最大化浏览器 driver.get("http://url") # 通过get()方法，打开一个url站点 time.sleep(2) # 输入登录账号密码，点击登录
driver.find_element_by_xpath("//input[@name='loginname']").send_keys(username)
driver.find_element_by_xpath("//input[@name='password']").send_keys(password)
driver.find_element_by_xpath(".//*[@id='btn']/input").click() time.sleep(2) # 调用判断元素是否存在来判断是否登录成功
self.element_exist() # 函数功能：保存并关闭浏览器 为什么这一步要单独分出来呢，因为测试其他模块的时候，被调用的时可以不用执行这几行代码。
def save_quit(self): wt = copy(workbook)
wt.get_sheet("login").write(self.x, self.y+2, k)
wt.save(r'F:\test.xlsx')
driver.quit() # 函数功能：判断页面是否存在元素 可调用该函数进行验证页面是否存在某个元素 def element_exist(self):
global k
s = driver.find_elements_by_css_selector(css_selector=self.css)
result = [u'通过', u'不通过']
if s:
k = result[0] print("通过")
else:
k = result[1] print("不通过")
return k
#这一步有待优化了，少了这一步，登录脚本被其他模块调用时会报driver未定义，多了这一步，又多开了个浏览器，所以我多加了一步，关闭浏览器，就当做测试下驱动
#如有优化建议，麻烦留言，谢谢！
driver = webdriver.Ie()
driver.quit() if __name__ == "__main__": for x in range(1, 3):
y = 0
test = login_test(x, y, css="#foot")
test.auto_login()
test.save_quit()
excel结果：
A模块脚本调用登录脚本automain.py：
# coding:utf-8
from selenium import webdriver
import time
# import xlrd
# from xlutils.copy import copy
import auto
from auto import driver auto.login_test(1, 0, css="#foot").auto_login() time.sleep(2) # 直接进行已登录页面的自动化测试操作
driver.find_element_by_xpath(".//*[@id='xxxx']/li[2]/a").click()
执行automain.py, 则从excel获取正确登录的数据，直接登录，无回写入excel（想写入也可以调），无关掉浏览器，直接等待下一步测试操作。
如有错误，请及时指正，谢谢