单元测试模块在脚本中时，sys.argv [1]有问题

Question

我有一个脚本，可以执行各种操作并使用sys.argv访问参数，但是当脚本到达代码的unittest部分时，它表示没有用于此目的的模块。我拥有的脚本是：

class MyScript():

    def __init__(self):
        self.value = sys.argv[1]

    def hello(self):
        print self.value

    def suite(self):
        modules_to_test = ('external_sanity_onvif', 'starttest')
        alltests = unittest.TestSuite()
        for module in map(__import__, modules_to_test):
            alltests.addTest(unittest.findTestCases(module))
        return alltests


if __name__ == '__main__': 
    Run = MyScript()
    Run.hello()
    log_file = 'log_file.txt'
    test_file = open(log_file, "w") 
    runner = unittest.TextTestRunner(test_file)
    unittest.main(defaultTest='Run.suite', testRunner=runner)

假设我./script.py Hello在命令行中输入。我得到的错误是：

AttributeError: 'module' object has no attribute 'Hello'

如果我删除unittest模块，它将起作用。另外，如果我删除testrunner日志并将其保留在：

unittest.main(defaultTest='Run.suite')

这仍然行不通。

谁能帮忙。

谢谢

我尝试了这种技巧，但仍然尝试阅读sys.argv。

project = sys.argv[4:]
sys.argv = sys.argv[0:4]

我已经尝试过了，argv但是仍然很难读取额外的参数。

Answer 1

问题是unittest.main()要让您宝贵的argv供自己使用！它使用您提供的argv作为函数参数，或者sys.argv使用您未明确提供argv的参数，并尝试加载名为您提供的参数的测试。在这种情况下，这意味着它将在模块“脚本”中查找名为的子模块，名为Hello的TestCase类Hello，位于的测试用例类中的测试用例方法Hello或Hello返回TestCase或的TestSuite实例的可调用对象。

有几种方法可以解决此问题：

• 自己绕过unittest.main()并调用较低级别的unittest函数，以设置和运行您所想到的测试用例。
• 删除代码对的依赖sys.argv，并充分利用该unittest.main()行为。如果您的模块 除了 作为单元测试 之外 不打算独立运行，这可能是有道理的，因为模块的调用者可能不希望您从其argv中读取内容！
• 将测试代码和主例程分离到一个单独的测试模块中。您仍然需要弄清楚如何通过测试模块将正确的argv导入代码中。
• 指定argv=[sys.argv[0]]作为参数unittest.main()；应该可以避免尝试阅读您的文章。