python - 模拟一个对象的返回值

Question

我想为以下代码编写一些测试:

def person_name_length(id):
    person = get_person(id)
    return len(person.first_name)

如何模拟 get_person(id) 方法，以便它返回一个具有 first_name 属性值的对象？

在代码中:

@patch('get_person')
def test_person_name_length(self, get_person_mock):
    get_person_mock.return_value = # what goes here??? calling .first_name on it should return 'Bob'
    self.assertEqual(person_name_length(1), 3)

Answer 1

我认为你应该需要做这样的事情

class MockPerson(object):
    first_name = 'Bob'


@patch('get_person')
def test_person_name_length(self, get_person_mock):
    get_person_mock.return_value = MockPerson()
    self.assertEqual(person_name_length(1), 3)