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, [])