使用 Python 进行自动化测试的浏览器选择

Question

最初，我有 def setup() 选项，可以从另一个文件中调用特定的 Web 浏览器，该文件包含有关如何设置 Web 浏览器的所有信息。标记出未使用的浏览器。

    def setUp(self):
    # Choose the Web Browser to test with
    operabrowser(self)
    # chromebrowser(self)
   ...

def test_one()
    ...
def test_two()
    ...

我试图摆脱每次编辑测试文件，我希望设置一个将调用浏览器的控制台输入，并且我已添加到 def setup()。

    browser_choice = input ( """Choose your browser
    Opera, [Firefox], Chrome or Safari
    > """ ).lower ()
    if browser_choice == 'opera':
        operabrowser()
    else chromebrowser()

这适用于单个单元测试，但如果文件中有多个测试，它每次都会询问浏览器选择。

对于将在测试文件中运行的所有测试，如何只询问一次此选项？我尝试了其他几种方法来解决这一切都失败了。

Answer 1

测试应该定义为类方法，在您的代码中，它们似乎是独立的函数，与拥有setUp() 方法的类没有直接关系。你只需要像这样修改你的代码：

import unittest

class YouTests(unittest.TestCase):

    def setUp(self):
        q = 'Choose your browser Opera, [Firefox], Chrome or Safari >'
        browser_choice = input (q).lower ()
        if browser_choice == 'opera':
            operabrowser()
        else: 
            chromebrowser()

    def test_one(self):
        pass

    def test_two(self):
        pass

...

如果您有多个继承自 unittest.TestCase 的类，并且所有类都应使用相同的浏览器，则最好将浏览器选择插入您的 global 范围。例如：

browser_object = None
...

def operabrowser():
    global browser_object
    # Init opera object here

def chromebrowser():
    global browser_object
    # Init chrome object here

...
# Your tests using the browser_object
...

if __name__ == '__main__':
    q = 'Choose your browser Opera, [Firefox], Chrome or Safari >'
    browser_choice = input (q).lower ()
    if browser_choice == 'opera':
        operabrowser()
    elif browser_choice == 'firefox':
        ...
    else: 
        chromebrowser()

    unittest.main()