我有一个脚本,它充当使用Python unittest
模块编写的一些单元测试的包装器。除了清理一些文件,创建输出流和生成一些代码之外,它还使用
unittest.TestLoader().loadTestsFromTestCase()
我已经使用optparse
提取了几个用于确定输出位置的命令行参数,是否重新生成代码以及是否进行一些清理。我还想传递一个配置变量,即端点URI,以便在测试用例中使用。
我意识到我可以在TestCase的setUp方法中添加OptionParser
,但我希望将选项传递给setUp
。这可以使用loadTestsFromTestCase()
吗?我可以迭代返回的TestSuite
TestCases
,但是我可以在TestCases
上手动调用setUp吗?
**编辑**
我想指出,如果我遍历测试并手动调用setUp
,我可以将参数传递给setUp
,如下所示:
(options, args) = op.parse_args()
suite = unittest.TestLoader().loadTestsFromTestCase(MyTests.TestSOAPFunctions)
for test in suite:
test.setUp(options.soap_uri)
但是,我正在使用xmlrunner
,其run方法以TestSuite
为参数。我假设它将运行setUp方法本身,所以我需要XMLTestRunner
中可用的参数。
我希望这是有道理的。
答案 0 :(得分:47)
好吧,我想做同样的事情,并且自己会问这个问题。我想改进以下代码,因为它有重复。它允许我发送参数来测试TestCase但是:
import unittest
import helpspot
class TestHelpSpot(unittest.TestCase):
"A few simple tests for HelpSpot"
def __init__(self, testname, path, user, pword):
super(TestHelpSpot, self).__init__(testname)
self.hs = helpspot.HelpSpot(path, user, pword)
def test_version(self):
a = self.hs.version()
b = self.hs.private_version()
self.assertEqual(a, b)
def test_get_with_param(self):
a = self.hs.filter_get(xFilter=1)
def test_unknown_method(self):
self.assertRaises(helpspot.HelpSpotError, self.hs.private_wuggienorple)
if __name__ == '__main__':
import sys
user = sys.argv[1]
pword = sys.argv[2]
path = sys.argv[3]
test_loader = unittest.TestLoader()
test_names = test_loader.getTestCaseNames(TestHelpSpot)
suite = unittest.TestSuite()
for test_name in test_names:
suite.addTest(TestHelpSpot(test_name, path, user, pword))
result = unittest.TextTestRunner().run(suite)
sys.exit(not result.wasSuccessful())
答案 1 :(得分:7)
if __name__ == '__main__':
from optparse import OptionParser
parser = OptionParser()
parser.add_option("-z", "--zebra",
action="store_true", dest="zebra", default=False,
help="run like a zebra")
(options, args) = parser.parse_args()
if options.zebra:
zebrafy()
# remove our args because we don't want to send them to unittest
for x in sum([h._long_opts+h._short_opts for h in parser.option_list],[]):
if x in sys.argv:
sys.argv.remove(x)
unittest.main()
答案 2 :(得分:5)
我肯定会建议不要像这样向setUp传递参数; setUp旨在在运行测试时隐式调用,因此您不应该像这样显式调用它。
解决这个问题的一种方法是在全局可访问的“上下文”模块中设置您需要设置为环境变量或值的值,这将允许测试用例根据需要访问它们。我会去使用环境变量,因为它在运行测试方面更灵活(你不再依赖于命令行参数)。
答案 3 :(得分:0)
如果在init方法中定义属性,那么你可以简单地在构造函数中传递它们,就像这样..
import unittest
import helpspot
class TestHelpSpot(unittest.TestCase):
"A few simple tests for HelpSpot"
def __init__(self, testname, path, user, pword):
super(TestHelpSpot, self).__init__(testname)
self.path = path
self.user = user
self.pword = pword
....
....
....
if __name__ == '__main__':
True
suite = unittest.TestSuite()
suite.addTest(TestHelpSpot("test_version", path, user, pword))
unittest.TextTestRunner().run(suite)