selenium2+python_自动化测试实例_登录&下拉框

本篇博客主要是想记录一下自己在学习过程中编写的代码，将selenium与单元测试结合实现了百度网盘的登录测试、排序下拉框的测试，并使用测试套件组织测试模块执行，将结果生成html文件。代码中使用包、模块、unitest、TestSuite、HTMLTestRunner、输出文件名带上时间戳。

代码中包含一个以百度网盘为基础的两个测试用例，还有两个是在学习中单纯为了测试一些概念的毫无意义的测试用例。废话不多说，直接看代码吧，一些注意点会在代码中注释。

首先看一下项目中的目录结构：



selenium_result：存放HTMLTestRunner的输出结果

testCase：是百度网盘测试用例中用到的方法

—— __init__.py ：作用是定义此文件夹是个包：【备注1】

—— 其他就省略了，看代码就知道是什么用处了

——login_test.py：与本篇文章中的整体无关

allTestCase.py ：使用TestSuite与HTMLTestRunner

baidu_unitest.py：百度网盘的单元测试用例

testCase\__init__.py

#coding:utf-8
import login_test
import toolbarCase
import init

testCase\init.py

# coding:utf-8
import time
from selenium import webdriver
home_title = u'百度网盘-全部文件'
username = u'幸福_小乖'
toobar_order_list = [u"文件名", u"大小", u"修改日期"]
toobar_order_list_len = len(toobar_order_list)
cookieslist = []  #这个需要自己获得，参考【备注2】
def init(driver): # 初始化浏览器
driver.get("https://pan.baidu.com/")
driver.set_window_size(1280, 960)
driver.implicitly_wait(30)
for cookie in cookieslist:
driver.add_cookie(cookie)

def get_title_and_username(driver, url):  # 获取登录后的title和用户名
driver.get(url)
time.sleep(5)
title = driver.title
user = driver.find_element_by_class_name("user-name").get_attribute("innerHTML")
return [title, user]

testCase\toolbarCase

#coding:utf-8
from selenium.webdriver import ActionChains

def get_orderlist(driver):
url = "https://pan.baidu.com/disk/home"
driver.get(url)
toobar = driver.find_element_by_css_selector("#layoutMain").find_element_by_class_name("DxdbeCb")
order = toobar.find_element_by_class_name("EzLavy")
#showtype = toobar.find_element_by_class_name("ryvhP50g")
orderlistshow = order.find_element_by_class_name("efdtEWV0")
ActionChains(driver).move_to_element(orderlistshow).perform()
orderlist = order.find_element_by_class_name("sDwvAgb").find_elements_by_css_selector("span")

listpro = []
for orderem in orderlist:
emtext = orderem.get_attribute("innerText")
listpro.append(emtext)

return [orderlist, listpro]

def order(driver):
# ====================文件列表================================
filelists = driver.find_element_by_css_selector("#layoutMain").find_element_by_class_name(
"vdAfKMb").find_elements_by_css_selector("dd")
print len(filelists)
for file in filelists:
filehref = file.find_element_by_css_selector("div.file-name>div>a")
filetitle = filehref.get_attribute("title")
print filetitle

baidu_unitest.py

# coding:utf-8
import unittest
from selenium import webdriver
from testCase import * #由于在testCase包中的__init__.py中使用了导入语句，所以才可以这样写

class BaiduTestCase(unittest.TestCase):
# 使用修饰器，定义一个测试类的方法，这样可以不用每次执行一个测试用例就重新启动一个浏览器
@classmethod
def setUpClass(self):
self.driver = webdriver.Chrome()
init.init(self.driver)

# 测试百度云登录首页的title以及与账号对应的用户名
def test_01_title_and_username(self):
u"""测试百度云登录首页的title以及与账号对应的用户名"""
url = "https://pan.baidu.com/disk/home"
title = init.get_title_and_username(self.driver, url)[0]
username = init.get_title_and_username(self.driver, url)[1]
print title + "----" + username
self.assertEqual(title, init.home_title, "Title错误")
self.assertEqual(username, init.username, "用户名错误")

# 测试首页导航栏中的排序图标的下拉列表
def test_02_ordericon(self):
u"""测试首页导航栏中的排序图标的下拉列表"""
orderlist = toolbarCase.get_orderlist(self.driver)[0]
orderlisttext = toolbarCase.get_orderlist(self.driver)[1]
print len(orderlist)
self.assertEqual(len(orderlist), init.toobar_order_list_len, "排序列表中的选项不为3")
self.assertListEqual(orderlisttext, init.toobar_order_list, "排序下拉列表不正确")

# 原理同setUpClass()
@classmethod
def tearDownClass(self):
self.driver.close()

if __name__ == "__main__":
unittest.main()

a.py

import unittest

class A(unittest.TestCase):
def setUp(self):
self.x = 3
self.y = 4

def test_3(self):
self.assertEqual(self.x+self.y, 7)
print "\nA.test_3"

def test_4(self):
self.assertEqual(self.y + self.x, 1)  # 这个断言是错误的，为了查看测试报告中的错误显示情况
print "\nA.test_4"

def tearDown(self):
pass

if __name__ == "__main":
unittest.main()

b.py

import unittest
class B(unittest.TestCase):
def setUp(self):
self.x = 3
self.y = 4

def test_1(self):
self.assertEqual(self.x+self.y, 7)
print "\nB.test_1"

def test_2(self):
self.assertEqual(self.y - self.x, 1)
print "\nB.test_2"

def tearDown(self):
pass

if __name__ == "__main":
unittest.main()

allTestCase.py

# coding:utf-8
import unittest
import HTMLTestRunner
import a, b,  baidu_unitest
import time, os

def testcasesuite():

# 用测试套件加载测试类，按加载的顺序执行各个测试类，但是测试类中的测试方法还是根据名称来排序的。
testunit = unittest.TestSuite()
testunit.addTest(unittest.makeSuite(a.A))
testunit.addTest(unittest.makeSuite(b.B))
testunit.addTest(unittest.makeSuite(baidu_unitest.BaiduTestCase))

timestring = time.strftime("%Y-%m-%d_%H_%M_%S", time.localtime())
# 这块是获取项目路径，后面在生成report时会放到改目录下
current_path = os.getcwd()

# 要先建立下面的文件夹selenium_result，否则会报错[Errno 22] invalid mode ('wb') or filename
result_path = current_path + r'\selenium_result'

filename = result_path +r'\result_'+ timestring + '.html'
print filename

fp = file(filename, 'wb')
runner = HTMLTestRunner.HTMLTestRunner(
stream=fp,
title=u'selenium自动化测试学习',
description=u'用例执行结果:'
)
runner.run(testunit)
file.close(fp)

if __name__ == '__main__':
print '开始执行测试用例'
testcasesuite()
print '测试用例执行完毕'

测试位置：



测试结果：





备注1：当导入一个包时，实际上是导入它的__init__.py文件，在__init__.py文件中添加导入语句，可以在添加新的测试方法时，直接修改__init__.py文件就可以了，对于引用它的文件不必做任何修改。

备注2：http://blog.csdn.net/xingfu_xiaoguai/article/details/79230588

备注3：测试类中的每个测试方法的执行顺序是按照方法名称排序的，并不是编写测试方法的先后顺序！

备注4：HTMLTestRunner 需要下载，是一个.py文件，需要放到程序的Interpreter的Lib目录下（windows）