简易selenium自动化测试框架(Python)

小小糖

最近空闲时间在探索Selenium的自动化测试，简单的写了一个小框架来测试公司的一个web产品。该
框架包括以下模块：



1. Test case编写模式（page模式，参考之前的博文http://www.cnblogs.com/AlwinXu/p/5537955.html)

2. Test case的管理及执行 （主要是用nose）

　　该模块借助了一个外部txt文件来记录测试用例，每个用例为自身的文件名，如果不需要在本次执
行，只需在文件名前添加一个“#”标识符就可以跳过该用例的执行。

3. 测试报告的生成（xml和html两种格式）



对于自动化测试而言，这些模块应该是最基本的配置了，当然还有一些辅助模块比如日志，其他公
共库模块等需要根据具体的业务逐渐丰富。闲话少说，用代码交流吧。

测试用例编写
该模块用了Page模式，之前介绍过，这次只贴代码了

1. 
   
2. BasePage.py:
   
3. 
   
4. __author__ = 'xua'
   
5. 
   
6. #super class
   
7. class BasePage(object):
   
8.     def __init__(self, driver):
   
9.         self.driver = driver
   
10. 然后是各个web page继承BasePage，LoginPage.py:
    
11. 
    
12. 复制代码
    
13. from BasePage import BasePage
    
14. from selenium.webdriver.common.by import By
    
15. from selenium.webdriver.common.keys import Keys
    
16. 
    
17. class LoginPage(BasePage):
    
18.     """description of class"""
    
19. 
    
20.     #page element identifier
    
21.     usename = (By.ID,'username')
    
22.     password = (By.ID, 'password')
    
23.     dialogTitle = (By.XPATH,'//html/body/div[7]/div/div/div[1]/h3')
    
24.     cancelButton = (By.XPATH,'//html/body/div[7]/div/div/div[3]/button[2]')
    
25. 
    
26.     #Get username textbox and input username
    
27.     def set_username(self,username):
    
28.         name = self.driver.find_element(*LoginPage.usename)
    
29.         name.send_keys(username)
    
30.    
    
31.     #Get password textbox and input password, then hit return
    
32.     def set_password(self, password):
    
33.         pwd = self.driver.find_element(*LoginPage.password)
    
34.         pwd.send_keys(password + Keys.RETURN)
    
35. 
    
36.     #Get pop up dialog title
    
37.     def get_DiaglogTitle(self):
    
38.         digTitle = self.driver.find_element(*LoginPage.dialogTitle)
    
39.         return digTitle.text
    
40. 
    
41.     #Get "cancel" button and then click
    
42.     def click_cancel(self):
    
43.         cancelbtn = self.driver.find_element(*LoginPage.cancelButton)
    
44.         cancelbtn.click()
    
45. 复制代码
    
46. 测试用例信息类：
    
47. 
    
48. TestCaseInfo.py
    
49. 
    
50. 复制代码
    
51. class TestCaseInfo(object):
    
52.     """description of class"""
    
53. 
    
54.     def __init__(self, id="",name="",owner="",result="Failed",starttime="",endtime="",errorinfo=""):
    
55.         self.id = id
    
56.         self.name = name
    
57.         self.owner = owner
    
58.         self.result = result
    
59.         self.starttime = starttime
    
60.         self.endtime = endtime
    
61.         self.errorinfo = errorinfo
    
62. 复制代码
    
63. 
    
64. 
    
65. 最后是每个测试用例的编写：（每个用例必须有自己的用例信息，这里有ID，Name等等信息，也会
    
66. 调用测试结果报告生成模块来添加测试结果）
    
67. 
    
68. Test_Login.py
    
69. 
    
70. 复制代码
    
71. __author__ = 'xua'
    
72. 
    
73. from selenium import webdriver
    
74. from selenium.webdriver.common.keys import Keys
    
75. from selenium.webdriver.common.alert import Alert
    
76. import unittest
    
77. import time
    
78. from LoginPage import LoginPage
    
79. from TestCaseInfo import TestCaseInfo
    
80. from TestReport import TestReport
    
81. 
    
82. class Test_Login(unittest.TestCase):
    
83. 
    
84.     #Setup
    
85.     def setUp(self):
    
86.         self.driver = webdriver.Chrome(r'C:\Users\xua\Downloads\chromedriver_win32\chromedriver.
    
87. exe')
    
88.         self.driver.implicitly_wait(30)
    
89.         self.base_url = "http://10.222.30.145:9000/"
    
90.         #test case information
    
91.         self.testcaseinfo = TestCaseInfo(id="3",name="Login to floor manager lite using sbxadmin",own
    
92. er="xua")
    
93.         self.testResult = TestReport()
    
94.   
    
95.     def test_Login(self):
    
96.         try:
    
97.             self.testcaseinfo.starttime = str(time.asctime())
    
98.             #Step1: open base site
    
99.             self.driver.get(self.base_url)
    
100.             #Step2: Open Login page
     
101.             login_page = LoginPage(self.driver)
     
102.             #Step3: Enter username
     
103.             login_page.set_username("sbXadmin")
     
104.             #Step4: Enter password
     
105.             login_page.set_password("IGTtest1")
     
106.             #Checkpoint1: Check popup dialog title
     
107.             self.assertEqual(login_page.get_DiaglogTitle(),"Sign in","Not Equal")
     
108.             #Step5: Cancel dialog
     
109.             login_page.click_cancel()
     
110.             self.testcaseinfo.result = "Pass"
     
111.         except Exception as err:
     
112.             self.testcaseinfo.errorinfo = str(err)
     
113.         finally:
     
114.             self.testcaseinfo.endtime = str(time.asctime())
     
115. 
     
116.     #tearDown
     
117.     def tearDown(self):
     
118.         self.driver.close()
     
119.         #write test result
     
120.         self.testResult.WriteHTML(self.testcaseinfo)
     
121. 
     
122. if __name__ == "__main__":
     
123.     unittest.main()

复制代码

复制代码
用例执行模块
1. 借助外部文件记录需要执行的用例

testcases.txt(带“#”标识的用例不会被执行):

Test_Login.py
Test_Login_2.py
#Test_Login_3.py
Test_Login_4.py
2. 利用nose的nosetests命令执行各个用例：

复制代码

1. import subprocess
   
2. 
   
3. class RunTests(object):
   
4.     """description of class"""
   
5.     def __init__(self):
   
6.         self.testcaselistfile = "testcases.txt"
   
7.    
   
8.     #use nosetests command to execute test case list
   
9.     def LoadAndRunTestCases(self):
   
10.         f = open(self.testcaselistfile)
    
11.         testfiles = [test for test in f.readlines() if not test.startswith("#")]
    
12.         f.close()
    
13.         for item in testfiles:
    
14.             subprocess.call("nosetests "+str(item).replace("\\n",""),shell = True)
    
15. 
    
16. if __name__ == "__main__":
    
17.     newrun = RunTests()
    
18.     newrun.LoadAndRunTestCases()

复制代码

复制代码
测试结果报表生成模块
测试报表模块写了两种格式：xml和html

1. 
   
2. TestReport.py
   
3. 
   
4. 复制代码
   
5. from xml.etree import ElementTree as ET
   
6. import os
   
7. import lxml.etree as mytree
   
8. from lxml import html
   
9. 
   
10. class TestReport(object):
    
11.     """description of class"""
    
12.     def __init__(self):
    
13.         self.testreport = "TestResult.xml"
    
14. 
    
15.     #If there is no "TestResult.xml", then create one
    
16.     def CreateTestResultFile(self):
    
17.         if os.path.exists(self.testreport) == False:
    
18.             newElem = ET.Element("TestCases")
    
19.             newTree = ET.ElementTree(newElem)
    
20.             newTree.write(self.testreport)
    
21.                         
    
22.     #Write test result to xml
    
23.     def WriteResult(self,testcaseInfo):
    
24.         self.CreateTestResultFile()
    
25.         testResultFile = ET.parse(self.testreport)
    
26.         root = testResultFile.getroot()
    
27.         newElem = ET.Element("TestCase")
    
28.         newElem.attrib = {
    
29.             "ID":testcaseInfo.id,
    
30.             "Name":testcaseInfo.name,
    
31.             "Owner":testcaseInfo.owner,
    
32.             "Result":testcaseInfo.result,
    
33.             "StartTime":testcaseInfo.starttime,
    
34.             "EndTime":testcaseInfo.endtime,
    
35.             "ErrorInfo":testcaseInfo.errorinfo
    
36.             }
    
37.         root.append(newElem)
    
38. 
    
39.         testResultFile.write(self.testreport)
    
40. 
    
41.     #If there is no "TestResult.html" file exists, then create one with default style
    
42.     def CreateHtmlFile(self):
    
43.         if os.path.exists("TestResult.html") == False:
    
44.             f = open("TestResult.html",'w')
    
45.             message = """<html>
    
46.             <head>   
    
47.                 <title>Automation Test Result</title>
    
48.                 <style>
    
49.                     table {
    
50.                             border-collapse: collapse;
    
51.                             padding: 15px;
    
52.                             font-family: "Trebuchet MS", Arial, Helvetica, sans-serif;
    
53.                             }
    
54.                     th{
    
55.                         background-color: green;
    
56.                         color: white;
    
57.                         border: 1px solid #ddd;
    
58.                         padding-bottom: 15px;
    
59.                         padding-top: 15px;
    
60.                     }
    
61.                     tr{
    
62.                         border: 1px solid #008000;
    
63.                         padding-bottom: 8px;
    
64.                         padding-top: 8px;
    
65.                         text-align: left;
    
66.                     }
    
67.                     td{
    
68.                         border: 1px solid #008000;
    
69.                     }
    
70.                 </style>
    
71.             </head>
    
72.             <body>
    
73.                 <h1>Automation Test Result</h1>
    
74.                 <table>
    
75.                     <tr>
    
76.                         <th>ID</th>
    
77.                         <th>Name</th>
    
78.                         <th>Owner</th>
    
79.                         <th>Result</th>
    
80.                         <th>StartTime</th>
    
81.                         <th>EndTime</th>
    
82.                         <th>ErrorMessage</th>
    
83.                    </tr>
    
84.                 </table>
    
85.             </body>
    
86.             </html>
    
87.             """
    
88.             f.write(message)
    
89.             f.close()
    
90. 
    
91.     #append new test result to testresult file
    
92.     def WriteHTML(self,testcaseinfo):
    
93. 
    
94.         self.CreateHtmlFile()
    
95. 
    
96.         f = open("TestResult.html","r")
    
97.         
    
98.         htmlcontent = f.read()
    
99.         f.close()
    
100.         tree = html.fromstring(htmlcontent)
     
101.         tableElem = tree.find(".//table")
     
102.         if testcaseinfo.result == "Failed":
     
103.             mytablerow = "<tr><td>{0}</td><td>{1}</td><td>{2}</td><td bgcolor=\"#FF0000\">{3}</td><td>{4}</td><td>{5}</td><td>{6}</td></tr>".format(testcaseinfo.id,testcaseinfo.name,testcaseinfo.owner,testcaseinfo.result,testcaseinfo.starttime,testcaseinfo.endtime,testcaseinfo.errorinfo)
     
104.         else:
     
105.             mytablerow = "<tr><td>{0}</td><td>{1}</td><td>{2}</td><td>{3}</td><td>{4}</td><td>{5}</td><td>{6}</td></tr>".format(testcaseinfo.id,testcaseinfo.name,testcaseinfo.owner,testcaseinfo.result,testcaseinfo.starttime,testcaseinfo.endtime,testcaseinfo.errorinfo)
     
106.         tableElem.append(mytree.HTML(str(mytablerow)))
     
107. 
     
108.         f = open("TestResult.html","w")
     
109.         #html.tostring
     
110.         newContent = repr(html.tostring(tree,method="html",with_tail=False))
     
111.         newContent = newContent.replace(r"\n","").replace(r"\t","").replace('b\'',"")
     
112.         newContent = newContent[:len(newContent)-1]
     
113.         f.write(newContent)
     
114.         f.close()

复制代码

复制代码
ok，最后看一下生成的测试报表：

梦想家