class TestContact(TestCase): def setUp(self): self.contact = Contact(name='contact', email='*****@*****.**') def test_contact_default_values(self): self.assertNotEqual(self.contact.id, None) self.assertEqual(self.contact.phone, '') self.assertEqual(self.contact.enabled, True) def test_contacts_are_equal(self): self.assertEqual(self.contact, Contact(name='contact', email='*****@*****.**')) def test_contacts_are_not_equal(self): contact = Contact(name='contact', email='*****@*****.**') another_contact = Contact(name='another contact', email='*****@*****.**') self.assertNotEqual(contact, another_contact) def test_to_dict(self): d = Contact(name='contact', email='*****@*****.**').to_dict() keys = ['id', 'name', 'email', 'phone'] self.assertTrue(all([k in d for k in keys])) def test_check_as_list(self): self.assertEqual(type(self.contact.as_list()), list) self.assertEqual(len(self.contact.as_list()), 1)
def _get_test_contacts(self): return [ Contact(name='Hernan', email='hernan@invader'), Contact(name='Lucas', email='lucas@invader'), ]
def setUp(self): self.contact = Contact(name='contact', email='*****@*****.**')