从命令行运行python代码时如何读取配置文件作为输入

Question

我在 python 中有多个测试类和一个 basetest 类。 测试类继承自 basetest 类。

我想从命令行运行测试类，它需要一些输入参数，这些输入参数应该存储在配置文件中。该配置文件可以作为命令行的输入提供。

我怎样才能做到这一点，从命令行提供它，然后解析配置文件。我需要在 basetest 类中进行解析，因为这将用于所有测试类。

测试类：如果内容没有多大意义，您可以忽略它们，因为它只是一个示例类

from selenium import webdriver
import unittest

class TestCase():

    def setUp(self):
        super().setUp()

    def test(self):
        baseUrl = "https://www.google.com"
        driver = webdriver.Firefox()
        driver.get(baseUrl)
        elementByXpath = driver.find_element_by_xpath("//input[@id='name']")

        if elementByXpath is not None:
            print("We found an element by XPATH")

if __name__ == '__main__':
    unittest.main(verbosity=2)
    result = unittest.TestResult()

基本测试用例：

class BaseTestCase():

    def __init__(self):
        pass

    def setUp(self):
        # In the setup() or __init__(), config file should be parsed
        self.login.login("admin", "admin")

    def tearDown(self):
        self.log.info("Driver quit, session closed")

将从命令行运行的文件将是测试类。 像这样的东西： python testcase.py --configfile

我对 python 很陌生，非常感谢任何帮助。

我曾经使用 Java 和 TestNG，这真的很容易。我们可以在 TestNG xml 文件本身中提供任何配置。

我正在使用 python 中的 unittest，任何关于哪个框架可以比 unittest 更好的建议也很好。

这是一个 GUI 测试框架，我需要在不同的浏览器和平台上运行测试。

Answer 1

使用如下层次结构。

`
|
|- TestModule
       |
       |- main.py (your entry point to the test)
       |- conf.ini
       |- parse_config.py

现在在这里，您在parse_config.py 上拨打您的SafeConfigParser conf.ini。将其路径作为字符串传递给配置解析器。

在测试运行器的设置（或 before_all 挂钩）中实例化您在 parse_config 文件中创建的类。

class ParseConfig(object):
    def __init__(self):
        self.base_url = None
        ....

    def parse_file(self):
        parser = SafeConfigParser()
        parser.read("config.ini")
        self.base_url = parser.get("section", "url")
        ....

希望您大致了解要做什么

编辑

在parse_conf.py

import argparse

class ParseConfig(object):
    def __init__(self):
        self.base_url = None
        self.file_name = None

        self.parse_cli()  # This will parse the config file name from CLI
        self.parse_file() # This will parse the config file whose name you provide

    def parse_cli(self):
        parser = argparse.ArgumentParser(description="Parse conf file")
        parser.add_argument("--conf", "-c", metavar="Conf", help="Config file name")
        args = parser.parse_args()
        self.file_name = args.conf

    def parse_file(self):
        parser = SafeConfigParser()
        parser.read(self.file_name)
        self.base_url = parser.get("section", "url")
        ....

在你的main.py

from parse_config import ParseConfig
...
...
ParseConfig().parse_cli()
ParseConfig().parse_file()
...
...

这样你可以解析多个配置文件名。

现在调用你的脚本，

python main.py -c conf.ini