1

我有一个要使用模拟测试的实例方法。我试图用另一个带参数的方法替换这个实例类中的一个方法。

要测试的类：

class ClassToTest():
    def compute(self):
        result = self._get_xml()
        return result

    def _get_xml(self):
        #get xml here
        return xml

测试文件：

from mock import patch
class ClassTest():

    @patch('classToTest._get_xml', _get_fake_xml)
    def computetest(self):
        test = ClassToTest()
        toassert = test.compute()

        #assert whatever
        #self.assert(...)

    def _get_fake_xml(self, objects, filename):
        py_file = os.path.abspath(__file__)
        py_dir = os.path.dirname(py_file)
        xml_file = os.path.join(py_dir, filename)
        xml_tree = objectify.parse(xml_file)
        return xml_tree.getroot()
        return xml

现在如何在补丁中添加 _get_fake_xml 方法的参数

我试过了 ： 

@patch('classToTest._get_xml', _get_fake_xml(mock.Mock(), 'filenam.xml'))

但这没有用。然后我尝试将文件名设为全局变量，但出现以下错误：

ClassToTest 实例中不存在 self.filename。

基本上我希望能够将 _get_fake_xml 与任何其他文件名重用。

有什么建议吗？

4

2 回答 2

2

所以，经过一番思考，我决定使用mock .return_value。

@patch('classToTest._get_xml')
    def computetest(self, get_xml):
        get_xml.return_value = _get_fake_xml('filenam.xml')
        test = ClassToTest()
        toassert = test.compute()

我想知道是否有办法在补丁装饰器中添加参数。

于 2012-12-03T15:15:53.313 回答
0

你能把它放在一个闭包里吗？您可以从您的_get_fake_xml方法中返回一个仍然可以访问_get_fake_xml' 范围的函数，如下所示：

from mock import patch
class ClassTest():

    @patch('classToTest._get_xml', _get_fake_xml(mock.Mock(), 'filenam.xml'))
    def computetest(self):
        test = ClassToTest()
        toassert = test.compute()

        #assert whatever
        #self.assert(...)

    def _get_fake_xml(self, objects, filename):
        py_file = os.path.abspath(__file__)
        py_dir = os.path.dirname(py_file)
        xml_file = os.path.join(py_dir, filename)
        xml_tree = objectify.parse(xml_file)
        # don't know much about mock, but I include *a in case this is passed self
        return lambda *a: xml_tree.getroot()
于 2012-10-26T17:29:21.500 回答