Python mockito -如何验证方法的字段

Question

我在单元测试中使用。我熟悉mockito的功能，如验证、模拟、捕获等，但我不知道如何验证方法文件的值。

我的制作代码。

class Dog(BaseModel):
    type: str
    age: int

    def bark(self, times: int) -> None:
        print(f"{self.type} {self.age}  {'ruf' * times}")


class FlowManager:

    def __init__(self, barks: int, dog_type: str, age: int):
        self.barks = barks
        self.dog_type = dog_type
        self.age = age

    def foo(self):
        # Some code....
        dog = Dog(type=self.dog_type, age=self.age)
        dog.bark(self.barks)
        # More some code...

这是包含"FlowManager“类的"foo”方法的单元测试。

from mockito import verify, when
class TestFlowManager(unittest.TestCase):

    def test_foo_happy_flow(self):
        # Arrange
        when(Dog).bark(...).thenReturn(None)

        # Act
        num_of_barks = 5
        dog_type = "bulldog"
        dog_age = 3
        FlowManager(num_of_barks, dog_type, dog_age).foo()

        # Assert
        verify(Dog).bark(num_of_barks)

我的问题是:如何断言Dog的属性。换句话说:我如何断言用dog_type==“牛头犬”和dog_age==3创建的Dog类？

谢谢!

Answer 1

看起来您没有将Dog注入到FlowManager或其方法中。在foo中，您调用一个模块全局Dog，所以您可能必须拦截它：

when(module_under_test).Dog(type="bulldog", age=3).thenReturn(mock())

由于when配置非常具体，所以您可能不需要验证任何内容。(foo的类型为None -> None；这只是意外的副作用:grin:)