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

lsekfe

转自：博客园

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

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

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

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

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

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

测试用例编写

该模块用了Page模式，之前介绍过，这次只贴代码了

BasePage.py:

1. __author__ = 'xua'
   
2. 
   
3. #super class
   
4. class BasePage(object):
   
5.     def __init__(self, driver):
   
6.         self.driver = driver

复制代码

然后是各个web page继承BasePage，LoginPage.py:

from BasePage import BasePage
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys

class LoginPage(BasePage):
    """description of class"""

    #page element identifier
    usename = (By.ID,'username')
    password = (By.ID, 'password')
    dialogTitle = (By.XPATH,'//html/body/div[7]/div/div/div[1]/h3')
    cancelButton = (By.XPATH,'//html/body/div[7]/div/div/div[3]/button[2]')

    #Get username textbox and input username
    def set_username(self,username):
        name = self.driver.find_element(*LoginPage.usename)
        name.send_keys(username)
   
    #Get password textbox and input password, then hit return
    def set_password(self, password):
        pwd = self.driver.find_element(*LoginPage.password)
        pwd.send_keys(password + Keys.RETURN)

    #Get pop up dialog title
    def get_DiaglogTitle(self):
        digTitle = self.driver.find_element(*LoginPage.dialogTitle)
        return digTitle.text

    #Get "cancel" button and then click
    def click_cancel(self):
        cancelbtn = self.driver.find_element(*LoginPage.cancelButton)
        cancelbtn.click()

测试用例信息类：

TestCaseInfo.py

1. class TestCaseInfo(object):
   
2.     """description of class"""
   
3. 
   
4.     def __init__(self, id="",name="",owner="",result="Failed",starttime="",endtime="",errorinfo=""):
   
5.         self.id = id
   
6.         self.name = name
   
7.         self.owner = owner
   
8.         self.result = result
   
9.         self.starttime = starttime
   
10.         self.endtime = endtime
    
11.         self.errorinfo = errorinfo

复制代码

最后是每个测试用例的编写：（每个用例必须有自己的用例信息，这里有ID，Name等等信息，也会调用测试结果报告生成模块来添加测试结果）

Test_Login.py

1. __author__ = 'xua'
   
2. 
   
3. from selenium import webdriver
   
4. from selenium.webdriver.common.keys import Keys
   
5. from selenium.webdriver.common.alert import Alert
   
6. import unittest
   
7. import time
   
8. from LoginPage import LoginPage
   
9. from TestCaseInfo import TestCaseInfo
   
10. from TestReport import TestReport
    
11. 
    
12. class Test_Login(unittest.TestCase):
    
13. 
    
14.     #Setup
    
15.     def setUp(self):
    
16.         self.driver = webdriver.Chrome(r'C:\Users\xua\Downloads\chromedriver_win32\chromedriver.exe')
    
17.         self.driver.implicitly_wait(30)
    
18.         self.base_url = "http://10.222.30.145:9000/"
    
19.         #test case information
    
20.         self.testcaseinfo = TestCaseInfo(id="3",name="Login to floor manager lite using sbxadmin",owner="xua")
    
21.         self.testResult = TestReport()
    
22.   
    
23.     def test_Login(self):
    
24.         try:
    
25.             self.testcaseinfo.starttime = str(time.asctime())
    
26.             #Step1: open base site
    
27.             self.driver.get(self.base_url)
    
28.             #Step2: Open Login page
    
29.             login_page = LoginPage(self.driver)
    
30.             #Step3: Enter username
    
31.             login_page.set_username("sbXadmin")
    
32.             #Step4: Enter password
    
33.             login_page.set_password("IGTtest1")
    
34.             #Checkpoint1: Check popup dialog title
    
35.             self.assertEqual(login_page.get_DiaglogTitle(),"Sign in","Not Equal")
    
36.             #Step5: Cancel dialog
    
37.             login_page.click_cancel()
    
38.             self.testcaseinfo.result = "Pass"
    
39.         except Exception as err:
    
40.             self.testcaseinfo.errorinfo = str(err)
    
41.         finally:
    
42.             self.testcaseinfo.endtime = str(time.asctime())
    
43. 
    
44.     #tearDown
    
45.     def tearDown(self):
    
46.         self.driver.close()
    
47.         #write test result
    
48.         self.testResult.WriteHTML(self.testcaseinfo)
    
49. 
    
50. if __name__ == "__main__":
    
51.     unittest.main()

复制代码

用例执行模块

1. 借助外部文件记录需要执行的用例

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

1. Test_Login.py
   
2. Test_Login_2.py
   
3. #Test_Login_3.py
   
4. 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

TestReport.py

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

复制代码

ok，最后看一下生成的测试报表：

总结

在网上有很多关于Selenium自动化的Best Practice，当然大家也可以根据自己的需求来DIY自己的框架，不管简陋与否，好用才是硬道理!

SunnyLan

这例子看起来不错