Эталонный объект, созданный в середине функции с макетом unittest

Question

Если у меня есть такая функция:

def get_vcard():
    new_vcard = vobject.vCard()
    new_vcard.add('fn')
    new_card.fn.value = 'First Last'
    work_phone = new_vcard.add('tel')
    work_phone.value = '[phone]'
    mobile_phone = new_vcard.add('tel')
    mobile_phone.value = '[phone]'

И такой тест:

@patch('myapp.vobject.vCard', autospec=True)
def test_create_card(self, mock_vcard_constructor):
    mock_vcard = mock_vcard_constructor.return_value
    myapp.get_vcard()
    self.assertEqual('First Last', mock_vcard.fn.value)

Я также хочу сослаться на эти разные объекты телефонных номеров, чтобы проверить, правильно ли они установлены. Я не уверен, как я могу получить ссылку на них.

Answer 1

Вы можете получить доступ ко всем детям vCard с .getChildren()Например:

def get_vcard(new_vcard=vobject.vCard()):
    new_vcard.add('fn')
    new_vcard.fn.value = 'First Last'
    work_phone = new_vcard.add('tel')
    work_phone.value = '[phone]'
    mobile_phone = new_vcard.add('tel')
    mobile_phone.value = '[phone]'

    return new_vcard

Теперь ваш тестовый модуль будет выглядеть так:

def has_phone(vcard, value):
    for child in vcard.getChildren():
        if child.value == value:
            return True
    return False

@patch('myapp.vobject.vCard', autospec=True)
def test_create_card(self, mock_vcard_constructor):
    mock_vcard = get_vcard()

    self.assertEqual('First Last', mock_vcard.fn.value)
    self.assertTrue(has_phone(mock_vcard, '[phone]'))
    self.assertTrue(has_phone(mock_vcard, '[phone]'))

Еще одна возможность доступа ко всем телефонным номерам с vcard.contents['tel'] а затем пообщаться над ними.