Ejemplo n.º 1
0
    def test_create_and_delete_positive(self):
        """should raise TypeError when use a int as name"""
        phonebook = PhoneBook()
        create_from_input("jeif", "1234567890", phonebook)

        contact = phonebook.get_contacts_by_name("jeif")[0]

        delete_from_input(str(contact.get_contact_id()), phonebook)

        contact = phonebook.get_contacts_by_id(contact.get_contact_id())

        self.assertIsNone(contact)
    def test_add_contact(self):
        '''test if you can add a contact'''
        phonebook = PhoneBook()
        phonebook.initialize()

        expected_contact = Contact(phonebook.create_id(), "A", "0123456789")
        phonebook.save_contact(expected_contact)

        actual_contact = phonebook.get_contacts_by_name("A")[0]

        self.assertEqual(expected_contact.name, actual_contact.name)
        self.assertEqual(expected_contact.phonenumber,
                         actual_contact.phonenumber)