def test_find_by_last_name(self):
book = AddressBook()
book.add_person(Person('John', 'Doe', Address('New York'),
Email('*****@*****.**'),
PhoneNumber('88005553535')))
book.add_person(Person('Jane', 'Doe', Address('Chicago'),
Email('*****@*****.**'),
PhoneNumber('2128506')))
entries = book.find_by_name('Doe')
self.assertEqual(len(entries), 2)
self.assertEqual(entries[0].last_name, 'Doe')
self.assertEqual(entries[1].last_name, 'Doe')
self.assertEqual(len(book.find_by_name('Ivanov')), 0)
def test_find_by_first_name(self):
book = AddressBook()
book.add_person(Person('John', 'Doe', Address('New York'),
Email('*****@*****.**'),
PhoneNumber('88005553535')))
entries = book.find_by_name('John')
self.assertEqual(len(entries), 1)
entry = entries[0]
self.assertEqual(entry.first_name, 'John')
self.assertEqual(entry.last_name, 'Doe')
self.assertEqual(entry.street_addresses[0].address, 'New York')
self.assertEqual(entry.emails[0].email, '*****@*****.**')
self.assertEqual(entry.phone_numbers[0].number, '88005553535')
self.assertEqual(entry.groups, [])
