Mocking a return value which is an object

user_1055165 0
Viewed 6727

I'd like to write some tests for the following code:

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

How can I mock out get_person(id) method so that it returns an object which has a value for first_name property?

In code:

@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)
python-unittest mocking unit-testing python
2 Answers

You don't need to create a one-off object for this. This is easier:

@patch('get_person')
def test_person_name_length(self, get_person_mock):
    get_person_mock.return_value = mock.Mock(first_name='Bob')
    self.assertEqual(person_name_length(1), 3)
user_1365980
Related