使用嵌套方法对扭曲的 addCallback 进行单元测试

Question

我有一种情况要为下面的代码写一个单元测试，请在这里帮帮我。

def method1(self):
    self.method2()

def method2(self):
        def method2_success(result):
            if result.code == 200:
                self.state = 'running'
            else:
                self.state = 'failed'
                self.reactor.callLater(10, self.method2)

        def method2_failed(failure):
            self.state = 'failed'
            self.reactor.callLater(10, self.method2)

        self.state = 'in_progress'
        d = self.method3()
        d.addCallbacks(method2_success, method2_failed)
        return d

def method3(self):
    return some_deffered_object

如果有人知道如何使用 twisted 编写单元测试？

Answer 1

我不熟悉twisted，但你应该按照包本身的模式，比如class TCPConnectionTestsBuilder in src/twisted/internet/test/test_tcp.py.