python + selenium 实战

* 项目目标

 对百度登录页面(https://passport.baidu.com/v2/?login)进行登录测试

 python + selenium 实战_第1张图片

* 功能实现

  • 自动运行用例
  • 自动生成测试报告
  • 自动断言与测试截图
  • 自动将最新测试报告发送到指定邮箱
  • PageObject + Unittest + 数据驱动

 * 工程目录

 python + selenium 实战_第2张图片

  • base:基础共享包,通用定位元素方式封装
  • config:配置文件、配置信息存放的地方
  • util:存放第三方模块,发送邮件、读取报告等数据操作
  • page_obj:实现登录时,目标元素定位、自动发送信息等
  • business:执行page_obj包中相关py文件,逻辑操作等
  • case:内置程序main文件,真正case用例
  • report:测试报告 + 测试error截图

 * 各模块实现

 1、config模块

1)配置文件:LocalElement.ini

[LoginElement]
user_name=id>TANGRAM__PSP_3__userName
password=id>TANGRAM__PSP_3__password
login_error=id>TANGRAM__PSP_3__error
login_submit=id>TANGRAM__PSP_3__submit
login_loginbtn=id>TANGRAM__PSP_3__footerULoginBtn

 

2)配置信息:setting.py

import os

base_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) # 项目首路径
# report_path = os.path.join(base_dir,'report','first_case.html')  # 测试报告路径
config_ini = os.path.join(base_dir,'config','localElement.ini')  # localElement.ini 配置文件路径
report_path = os.path.join(base_dir,'report')  # report 配置文件路径

 

2、util模块

1)read_ini.py:读取配置文件内容

from config.setting import config_ini
import configparser

class Read_Ini(object):
    """读取配置文件信息"""
    def __init__(self,node = None):
        if node:
            self.node = node
        else:
            self.node = 'LoginElement'  # 配置文件中的某个节点
        self.cf = self.load_ini()

    def load_ini(self):  # 加载文件
        cf = configparser.ConfigParser()  # 使用 configparser模块读取配置文件信息
        cf.read(config_ini)  # 配置文件所在路径
        return cf

    def get_value(self,key): # 获取配置文件中key的value值
        data = self.cf.get(self.node,key)
        return data

# if __name__ == '__main__':
#     read_init = Read_Ini()
#     print(read_init.get_value('user_name'))
rea_ini.py

 

2)get_last_report.py:获取最近创建的测试报告 

import os
from config import setting

class GetLastReport(object):
    """获取最近一次测试报告"""
    def last_report_file(self):
        lists = os.listdir(setting.report_path)
        last_report_file = self.get_last_report_file(lists)
        # print(last_report_file)
        return last_report_file

    def get_last_report_file(self,lists):
        report_list = []
        for l_list in lists:
            if l_list.endswith('html'):
                report_list.append(l_list)
        last_report_file = report_list[-1]
        # print(report_list)
        return last_report_file

if __name__ == '__main__':
    get_last_report = GetLastReport()
    get_last_report.last_report_file()
get_last_report.py

 

3)send_email.py:测试完成后自动发送邮件 -->测试报告.html 

import smtplib
from email.mime.text import MIMEText
from email.header import Header
from util.get_last_report import GetLastReport
from config import setting
import os


last_report = GetLastReport()
report_file = last_report.last_report_file()

class SendEmail:
    def __init__(self,receivers = None):
        self.mail_host = "smtp.163.com"  # 设置服务器
        self.mail_user = "13*******[email protected]"  # 用户名
        self.mail_pass = "密令"  # 密令/密码

        self.sender = '13*******[email protected]'
        if receivers:
            self.receivers = receivers
        else:
            self.receivers = ['[email protected]']  # 接收邮件,可设置为你的QQ邮箱或者其他邮箱

    def get_report_file(self,report_file):
        """获取最近一次测试报告"""
        report_path = os.path.join(setting.report_path,report_file)
        with open(report_path,'rb') as f:
        # f = open(report_path,'rb')
            mail_content = f.read()
        return mail_content

    def send_email(self,report_file):
        content = self.get_report_file(report_file)
        message = MIMEText(content, _subtype='html', _charset='utf-8')  # 邮件内容
        subject = '百度登录-自动化测试报告'
        message['Subject'] = Header(subject, 'utf-8')  # 邮件主题
        message['From'] = self.sender  # 发送人,必填,邮箱格式
        message['To'] = ";".join(self.receivers)  # 收件人,必填,邮箱格式

        server = smtplib.SMTP()
        server.connect(self.mail_host, 25)  # 连接服务器
        server.login(self.mail_user, self.mail_pass)  # 登录
        try:
            server.sendmail(self.sender, self.receivers, message.as_string())
            print('发送成功')
        except smtplib.SMTPException as e:
            print("Error: 无法发送邮件")
        server.close()

if __name__ == "__main__":
    send_email = SendEmail()
    send_email.send_email(report_file)
send_email.py

 

 

4)HTMLTestRunner.py:用于生成测试报告,HTML格式 

"""
A TestRunner for use with the Python unit testing framework. It
generates a HTML report to show the result at a glance.

The simplest way to use this is to invoke its main method. E.g.

    import unittest
    import HTMLTestRunner

    ... define your tests ...

    if __name__ == '__main__':
        HTMLTestRunner.main()


For more customization options, instantiates a HTMLTestRunner object.
HTMLTestRunner is a counterpart to unittest's TextTestRunner. E.g.

    # output to a file
    fp = file('my_report.html', 'wb')
    runner = HTMLTestRunner.HTMLTestRunner(
                stream=fp,
                title='My unit test',
                description='This demonstrates the report output by HTMLTestRunner.'
                )

    # Use an external stylesheet.
    # See the Template_mixin class for more customizable options
    runner.STYLESHEET_TMPL = ''

    # run the test
    runner.run(my_test_suite)


------------------------------------------------------------------------
Copyright (c) 2004-2007, Wai Yip Tung
All rights reserved.

Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are
met:

* Redistributions of source code must retain the above copyright notice,
  this list of conditions and the following disclaimer.
* Redistributions in binary form must reproduce the above copyright
  notice, this list of conditions and the following disclaimer in the
  documentation and/or other materials provided with the distribution.
* Neither the name Wai Yip Tung nor the names of its contributors may be
  used to endorse or promote products derived from this software without
  specific prior written permission.

THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
"""

# URL: http://tungwaiyip.info/software/HTMLTestRunner.html

__author__ = "Wai Yip Tung"
__version__ = "0.8.2"

"""
Change History

Version 0.8.2
* Show output inline instead of popup window (Viorel Lupu).

Version in 0.8.1
* Validated XHTML (Wolfgang Borgert).
* Added description of test classes and test cases.

Version in 0.8.0
* Define Template_mixin class for customization.
* Workaround a IE 6 bug that it does not treat 

%(heading)s
%(report)s
%(ending)s



"""
    # variables: (title, generator, stylesheet, heading, report, ending)

    # ------------------------------------------------------------------------
    # Stylesheet
    #
    # alternatively use a  for external style sheet, e.g.
    #   

    STYLESHEET_TMPL = """

"""

    # ------------------------------------------------------------------------
    # Heading
    #

    HEADING_TMPL = """

%(title)s

%(parameters)s

%(description)s

""" # variables: (title, parameters, description) HEADING_ATTRIBUTE_TMPL = """

%(name)s: %(value)s

""" # variables: (name, value) # ------------------------------------------------------------------------ # Report # REPORT_TMPL = """

Show Summary Failed All

%(test_list)s
Test Group/Test case Count Pass Fail Error View
Total %(count)s %(Pass)s %(fail)s %(error)s
""" # variables: (test_list, count, Pass, fail, error) REPORT_CLASS_TMPL = r""" %(desc)s %(count)s %(Pass)s %(fail)s %(error)s Detail """ # variables: (style, desc, count, Pass, fail, error, cid) REPORT_TEST_WITH_OUTPUT_TMPL = r"""
%(desc)s
%(status)s
""" # variables: (tid, Class, style, desc, status) REPORT_TEST_NO_OUTPUT_TMPL = r"""
%(desc)s
%(status)s
""" # variables: (tid, Class, style, desc, status) REPORT_TEST_OUTPUT_TMPL = r""" %(id)s: %(output)s """ # variables: (id, output) # ------------------------------------------------------------------------ # ENDING # ENDING_TMPL = """
""" # -------------------- The end of the Template class ------------------- TestResult = unittest.TestResult class _TestResult(TestResult): # note: _TestResult is a pure representation of results. # It lacks the output and reporting ability compares to unittest._TextTestResult. def __init__(self, verbosity=1): TestResult.__init__(self) self.stdout0 = None self.stderr0 = None self.success_count = 0 self.failure_count = 0 self.error_count = 0 self.verbosity = verbosity # result is a list of result in 4 tuple # ( # result code (0: success; 1: fail; 2: error), # TestCase object, # Test output (byte string), # stack trace, # ) self.result = [] def startTest(self, test): TestResult.startTest(self, test) # just one buffer for both stdout and stderr self.outputBuffer = io.StringIO() stdout_redirector.fp = self.outputBuffer stderr_redirector.fp = self.outputBuffer self.stdout0 = sys.stdout self.stderr0 = sys.stderr sys.stdout = stdout_redirector sys.stderr = stderr_redirector def complete_output(self): """ Disconnect output redirection and return buffer. Safe to call multiple times. """ if self.stdout0: sys.stdout = self.stdout0 sys.stderr = self.stderr0 self.stdout0 = None self.stderr0 = None return self.outputBuffer.getvalue() def stopTest(self, test): # Usually one of addSuccess, addError or addFailure would have been called. # But there are some path in unittest that would bypass this. # We must disconnect stdout in stopTest(), which is guaranteed to be called. self.complete_output() def addSuccess(self, test): self.success_count += 1 TestResult.addSuccess(self, test) output = self.complete_output() self.result.append((0, test, output, '')) if self.verbosity > 1: sys.stderr.write('ok ') sys.stderr.write(str(test)) sys.stderr.write('\n') else: sys.stderr.write('.') def addError(self, test, err): self.error_count += 1 TestResult.addError(self, test, err) _, _exc_str = self.errors[-1] output = self.complete_output() self.result.append((2, test, output, _exc_str)) if self.verbosity > 1: sys.stderr.write('E ') sys.stderr.write(str(test)) sys.stderr.write('\n') else: sys.stderr.write('E') def addFailure(self, test, err): self.failure_count += 1 TestResult.addFailure(self, test, err) _, _exc_str = self.failures[-1] output = self.complete_output() self.result.append((1, test, output, _exc_str)) if self.verbosity > 1: sys.stderr.write('F ') sys.stderr.write(str(test)) sys.stderr.write('\n') else: sys.stderr.write('F') class HTMLTestRunner(Template_mixin): """ """ def __init__(self, stream=sys.stdout, verbosity=1, title=None, description=None): self.stream = stream self.verbosity = verbosity if title is None: self.title = self.DEFAULT_TITLE else: self.title = title if description is None: self.description = self.DEFAULT_DESCRIPTION else: self.description = description self.startTime = datetime.datetime.now() def run(self, test): "Run the given test case or test suite." result = _TestResult(self.verbosity) test(result) self.stopTime = datetime.datetime.now() self.generateReport(test, result) # print >> sys.stderr, '\nTime Elapsed: %s' % (self.stopTime-self.startTime) print(sys.stderr, '\nTime Elapsed: %s' % (self.stopTime - self.startTime)) return result def sortResult(self, result_list): # unittest does not seems to run in any particular order. # Here at least we want to group them together by class. rmap = {} classes = [] for n, t, o, e in result_list: cls = t.__class__ if not cls in rmap: rmap[cls] = [] classes.append(cls) rmap[cls].append((n, t, o, e)) r = [(cls, rmap[cls]) for cls in classes] return r def getReportAttributes(self, result): """ Return report attributes as a list of (name, value). Override this to add custom attributes. """ startTime = str(self.startTime)[:19] duration = str(self.stopTime - self.startTime) status = [] if result.success_count: status.append('Pass %s' % result.success_count) if result.failure_count: status.append('Failure %s' % result.failure_count) if result.error_count: status.append('Error %s' % result.error_count) if status: status = ' '.join(status) else: status = 'none' return [ ('Start Time', startTime), ('Duration', duration), ('Status', status), ] def generateReport(self, test, result): report_attrs = self.getReportAttributes(result) generator = 'HTMLTestRunner %s' % __version__ stylesheet = self._generate_stylesheet() heading = self._generate_heading(report_attrs) report = self._generate_report(result) ending = self._generate_ending() output = self.HTML_TMPL % dict( title=saxutils.escape(self.title), generator=generator, stylesheet=stylesheet, heading=heading, report=report, ending=ending, ) self.stream.write(output.encode('utf8')) def _generate_stylesheet(self): return self.STYLESHEET_TMPL def _generate_heading(self, report_attrs): a_lines = [] for name, value in report_attrs: line = self.HEADING_ATTRIBUTE_TMPL % dict( name=saxutils.escape(name), value=saxutils.escape(value), ) a_lines.append(line) heading = self.HEADING_TMPL % dict( title=saxutils.escape(self.title), parameters=''.join(a_lines), description=saxutils.escape(self.description), ) return heading def _generate_report(self, result): rows = [] sortedResult = self.sortResult(result.result) for cid, (cls, cls_results) in enumerate(sortedResult): # subtotal for a class np = nf = ne = 0 for n, t, o, e in cls_results: if n == 0: np += 1 elif n == 1: nf += 1 else: ne += 1 # format class description if cls.__module__ == "__main__": name = cls.__name__ else: name = "%s.%s" % (cls.__module__, cls.__name__) doc = cls.__doc__ and cls.__doc__.split("\n")[0] or "" desc = doc and '%s: %s' % (name, doc) or name row = self.REPORT_CLASS_TMPL % dict( style=ne > 0 and 'errorClass' or nf > 0 and 'failClass' or 'passClass', desc=desc, count=np + nf + ne, Pass=np, fail=nf, error=ne, cid='c%s' % (cid + 1), ) rows.append(row) for tid, (n, t, o, e) in enumerate(cls_results): self._generate_report_test(rows, cid, tid, n, t, o, e) report = self.REPORT_TMPL % dict( test_list=''.join(rows), count=str(result.success_count + result.failure_count + result.error_count), Pass=str(result.success_count), fail=str(result.failure_count), error=str(result.error_count), ) return report def _generate_report_test(self, rows, cid, tid, n, t, o, e): # e.g. 'pt1.1', 'ft1.1', etc has_output = bool(o or e) tid = (n == 0 and 'p' or 'f') + 't%s.%s' % (cid + 1, tid + 1) name = t.id().split('.')[-1] doc = t.shortDescription() or "" desc = doc and ('%s: %s' % (name, doc)) or name tmpl = has_output and self.REPORT_TEST_WITH_OUTPUT_TMPL or self.REPORT_TEST_NO_OUTPUT_TMPL # o and e should be byte string because they are collected from stdout and stderr? if isinstance(o, str): # TODO: some problem with 'string_escape': it escape \n and mess up formating # uo = unicode(o.encode('string_escape')) # uo = o.decode('latin-1') uo = e else: uo = o if isinstance(e, str): # TODO: some problem with 'string_escape': it escape \n and mess up formating # ue = unicode(e.encode('string_escape')) # ue = e.decode('latin-1') ue = e else: ue = e script = self.REPORT_TEST_OUTPUT_TMPL % dict( id=tid, output=saxutils.escape(str(uo) + ue), ) row = tmpl % dict( tid=tid, Class=(n == 0 and 'hiddenRow' or 'none'), style=n == 2 and 'errorCase' or (n == 1 and 'failCase' or 'none'), desc=desc, script=script, status=self.STATUS[n], ) rows.append(row) if not has_output: return def _generate_ending(self): return self.ENDING_TMPL ############################################################################## # Facilities for running tests from the command line ############################################################################## # Note: Reuse unittest.TestProgram to launch test. In the future we may # build our own launcher to support more specific command line # parameters like test title, CSS, etc. class TestProgram(unittest.TestProgram): """ A variation of the unittest.TestProgram. Please refer to the base class for command line parameters. """ def runTests(self): # Pick HTMLTestRunner as the default test runner. # base class's testRunner parameter is not useful because it means # we have to instantiate HTMLTestRunner before we know self.verbosity. if self.testRunner is None: self.testRunner = HTMLTestRunner(verbosity=self.verbosity) unittest.TestProgram.runTests(self) main = TestProgram ############################################################################## # Executing this module from the command line ############################################################################## if __name__ == "__main__": main(module=None)
HTMLTestRunner.py

 

3、base模块

1)find_element.py:根据配置文件信息,整合定位元素位置 

#coding=utf-8
from util.read_ini import Read_Ini
class FindElement(object):
    """根据配置文件信息,获取元素所在位置"""
    def __init__(self,driver):
        self.driver = driver
    def get_element(self,key):
        read_ini = Read_Ini()
        data = read_ini.get_value(key)
        by,value = data.split('>')

        try:
            if by == 'id':
                return self.driver.find_element_by_id(value)
            elif by == 'name':
                return self.driver.find_element_by_name(value)
            elif by == 'className':
                return self.driver.find_element_by_class_name(value)
            elif by == 'xpath':
                return self.driver.find_element_by_xpath(value)
            elif by == 'css':
                return self.driver.find_elements_by_css_selector(value)[0]
            else:
                return self.driver.find_element_by_tag_name(value)
        except Exception as e:
            # print("find_element错误信息:",e)
            return None
find_element.py

 

* PO模式 ↓

 4、page_obj模块

1)login_page.py:实现元素定位

from base.find_element import FindElement


class LoginPage(object):
    """操作 -->获取元素所在位置"""

    def __init__(self, driver):
        self.fd = FindElement(driver)

    # 获取用户名元素
    def get_username_element(self):
        return self.fd.get_element("user_name")

    # 获取密码元素
    def get_password_element(self):
        return self.fd.get_element("password")

    # 获取登录按钮元素
    def get_submit_element(self):
        return self.fd.get_element("login_submit")

    # 获取登录错误元素
    def get_login_error_element(self):
        return self.fd.get_element("login_error")

    def get_login_qrcodebtn_element(self):
        return self.fd.get_element("login_loginbtn")
login_page.py

 

2)login_handle.py:元素定位后,自动填写测试内容 

# coding=utf-8
from page_obj.login_page import LoginPage

class LoginHandle(object):
    """打开页面后自动输入相应信息"""

    def __init__(self, driver):
        self.driver = driver
        self.login_p = LoginPage(self.driver)

    # 输入用户名
    def send_user_name(self, username):
        # self.loger.info("输入的用户名是:"+username)
        self.login_p.get_username_element().send_keys(username)

    # 输入密码
    def send_user_password(self, password):
        # self.loger.info("输入的密码是:"+password)
        self.login_p.get_password_element().send_keys(password)


    # 获取文字信息
    def get_login_error(self,info):
        try:  # 容错处理
            text_error = self.login_p.get_login_error_element()
        except:
            text_error = None
        return text_error

    # 点击登录按钮
    def click_submit_btn(self):
        self.login_p.get_submit_element().click()

    # 获取登录按钮文字
    def get_submit_btn_text(self):
        """如获取不到信息,表明页面已成功跳转"""
        return self.login_p.get_submit_element().text

    # 点击用户账号登录,转用户账号登录
    def click_login_qrcodebtn(self):
        return self.login_p.get_login_qrcodebtn_element().click()
login_handle.py

 


5、business模块

1)login_business.py:元素定位 + 自动填写数据实现

#coding=utf-8
from page_obj.login_handle import LoginHandle
class LoginBusiness:
    """测试注册页面form表单功能情况"""
    def __init__(self,driver):
        self.login_h = LoginHandle(driver)

    def user_base(self,username,password):
        self.login_h.click_login_qrcodebtn()  # 点击扫码登录,百度登录页面默认扫码登录,需点击并转到账号密码登录,才能进行下面的测试
        self.login_h.send_user_name(username)  # 定位并输入用户名
        self.login_h.send_user_password(password)  # 定位并输入密码
        self.login_h.click_submit_btn()   # 定位登录按钮,点击跳转
    
    def login_succes(self):
        if self.login_h.get_submit_btn_text() == None:
            # 登录成功
            return True
        else:
            return False

    # 数据驱动,只执行此条代码
    # 用户名、密码、错误提示信息
    def login_error_function(self,username,password):
        self.user_base(username,password) # 输入账号、密码,点击登录后,查看是否有错误信息提醒
        if self.login_h.get_login_error('帐号或密码错误,请重新输入或者找回密码') == None: # 没有错误信息,表示成功登录
            return True
        else:
            return False
login_business.py

 

 6、case模块

1):case测试用例,unittest + 数据驱动 

import ddt
import unittest
import os
from selenium import webdriver
from config import setting
from business.login_business import LoginBusiness
from util.get_last_report import GetLastReport
from util.send_email import SendEmail
from util.HTMLTestRunner import HTMLTestRunner
import datetime

# sys.path.append(setting_conf.base_dir)

@ddt.ddt
class FirstDdtCase(unittest.TestCase):
    # @classmethod
    # def setUpClass(cls):

    def setUp(self):
        self.driver = webdriver.Chrome()
        self.driver.get('https://passport.baidu.com/v2/?login')
        self.driver.refresh()
        self.driver.maximize_window()
        self.login = LoginBusiness(self.driver)
        self.driver.implicitly_wait(5)

        # self.login = RegisterBusiness(self.driver)

    def tearDown(self):
        for method_name, error in self._outcome.errors:  # case如果执行失败,错误会保存到_outcome.errors 中
            if error:
                case_name = self._testMethodName  # case名,即定义好的方法名
                report_error_name = case_name + '.png'
                report_error_path = os.path.join(setting.report_path,'screenshot', report_error_name)
                self.driver.save_screenshot(report_error_path)
        # self.driver.close()
        self.driver.quit()

    # @classmethod
    # def tearDownClass(cls):
    #     cls.driver.close()

    @ddt.data(('15816485972','562148562'),('136******57','45378652'))
    @ddt.unpack
    def test_login_case(self,username,pwd):   # ex_data:[[],[],..] 列表套列表
        login_error = self.login.login_error_function(username, pwd)
        self.assertFalse(login_error,"登录成功,测试失败!") # login_error如果不等于False,表示没有错误信息,登录成功。此条case测试失败


if __name__ == '__main__':
    suite = unittest.TestLoader().loadTestsFromTestCase(FirstDdtCase)
    report_name = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S") + ".html"
    report_file = os.path.join(setting.report_path,report_name)
    with open(report_file, 'wb') as f:
        runner = HTMLTestRunner(stream=f,title="This is the first ddt_case report1",description="这个是我们第一次测试报告 --数据驱动",verbosity=2)
        runner.run(suite)

    last_report = GetLastReport()
    report_file = last_report.last_report_file()  # 获取最近一次报告
    send_email = SendEmail()  # 发送邮件
    send_email.send_email(report_file)
frist_ddt_case.py

 


测试结果:

 python + selenium 实战_第3张图片



 

转载于:https://www.cnblogs.com/Eric15/articles/9896731.html

你可能感兴趣的:(python + selenium 实战)