def setUp(self): self.info = { "uid": "", "first_name": "Marcus", "last_name": "Willock", "email": "*****@*****.**", "phone": "xxx-xxx-xxxx", "description": "Creator", "company_uid": 1 } self.full_name = "{} {}".format(self.info["first_name"], self.info["last_name"]) self.contact = Contact( self.info["uid"], self.info["first_name"], self.info["last_name"], self.info["email"], self.info["phone"], self.info["description"], self.info["company_uid"], testing=True ) self.new_first_name = "Testin!!!" self.total_num_of_contacts = 5 # Setting up the test db db = SQL(TEST_DB) db.create_db() db.add_test_data_to_db()
class TestContact(unittest.TestCase): def setUp(self): self.info = { "uid": "", "first_name": "Marcus", "last_name": "Willock", "email": "*****@*****.**", "phone": "xxx-xxx-xxxx", "description": "Creator", "company_uid": 1 } self.full_name = "{} {}".format(self.info["first_name"], self.info["last_name"]) self.contact = Contact( self.info["uid"], self.info["first_name"], self.info["last_name"], self.info["email"], self.info["phone"], self.info["description"], self.info["company_uid"], testing=True ) self.new_first_name = "Testin!!!" self.total_num_of_contacts = 5 # Setting up the test db db = SQL(TEST_DB) db.create_db() db.add_test_data_to_db() def test_properties(self): self.assertTrue(self.contact) self.assertEqual(self.contact.uid, self.info["uid"]) self.assertEqual(self.contact.first_name, self.info["first_name"]) self.assertEqual(self.contact.last_name, self.info["last_name"]) self.assertEqual(self.contact.email, self.info["email"]) self.assertEqual(self.contact.phone, self.info["phone"]) self.assertEqual(self.contact.description, self.info["description"]) self.assertEqual(self.contact.company_uid, self.info["company_uid"]) def test_full_name(self): self.assertEqual(self.contact.full_name, self.full_name) def test_get_all_contacts(self): self.assertEqual(5, len(self.contact.get_all_contacts())) def test_a_contact(self): data = self.contact.get_a_contact(1)[0] self.assertEqual("Marcus", data.first_name) def test_add_contact_to_db(self): self.contact.add_contact_to_db() self.assertTrue(self.contact.uid) def test_update_contact_in_db(self): test_class = self.contact.get_a_contact(1)[0] test_class.first_name = self.new_first_name test_class._testing = True test_class.update_contact_in_db() test_class = self.contact.get_a_contact(1)[0] self.assertEqual(test_class.first_name, self.new_first_name) def test_delete_contact_in_db(self): test_class = self.contact.get_a_contact(1)[0] test_class._testing = True test_class.delete_contact_in_db() list_of_contacts = self.contact.get_all_contacts() self.assertEqual(len(list_of_contacts), self.total_num_of_contacts - 1)