3

我需要测试一种可以移动文件的方法。我不希望文件移动操作实际发生，我只需要知道被测方法正确调用shutil.move

什么是最好的修补方法，shutil.move以便该方法可以在不执行实际文件操作的情况下调用它？

我是这样做的，但是很丑，我想使用模拟库来做到这一点：

    real_move = ftp2email.shutil.move
    move_operations = []
    def fake_move(src, dst):
        move_operations.append((src, dst))
    ftp2email.shutil.move = fake_move

    msg_id = '/path/to/message.xml'
    self.ch.mark_message(msg_id)
    self.assertEqual(move_operations,
                        [('/path/to/message.xml', '/path/to/archived/message.xml')])

    ftp2email.shutil.move = real_move
4

2 回答 2

1

您可以尝试使用mox，它可以做那种事情。

来自mox 食谱。

def testFoo(self):
    self.mox.StubOutWithMock(module_to_mock, 'FunctionToMock')
    module_to_mock.FunctionToMock().AndReturn(foo)

    self.mox.ReplayAll()
    ...
    self.mox.VerifyAll()
于 2012-10-26T22:51:13.883 回答
0

我已经使用Mock 库解决了它

@mock.patch('%s.ftp2email.shutil.move' % __name__)
def test_mark_message(self, move_mock):

    self.ch.mark_message('/path/to/message.xml')
    move_mock.assert_called_with('/path/to/message.xml',
                                 '/path/to/archived/message.xml')

这只是一种方法，我也可以使用“with 语句”或调用模拟方法 start 和 stop 来完成。

于 2012-10-30T14:26:30.823 回答