class TestCompany(unittest.TestCase): def setUp(self): self.info = { "uid": "", "name": "Test Company", "address": "new york, NY", "website": "www.testing.com", "phone": "555-123-4567"} self.company = Company( self.info["uid"], self.info["name"], self.info["address"], self.info["website"], self.info["phone"], testing=True ) self.new_uid = "6" self.new_name = "Testing!!!" self.total_num_of_companies = 5 #Reset the database db = SQL(TEST_DB) db.create_db() db.add_test_data_to_db() def test_properties(self): self.assertEqual(self.company.uid, self.info["uid"]) self.assertEqual(self.company.name, self.info["name"]) self.assertEqual(self.company.address, self.info["address"]) self.assertEqual(self.company.website, self.info["website"]) self.assertEqual(self.company.phone, self.info["phone"]) def test_get_all_companies(self): self.assertEqual(5, len(self.company.get_all_companies())) def test_get_a_company(self): data = self.company.get_a_company(1)[0] self.assertEqual("Google", data.name) def test_add_company_to_db(self): self.company.add_company_to_db() self.assertTrue(self.company.uid) def test_update_company_in_db(self): test_class = self.company.get_a_company(1)[0] test_class.name = self.new_name test_class._testing = True test_class.update_company_in_db() test_class = self.company.get_a_company(1)[0] self.assertEqual(test_class.name, self.new_name) def test_delete_company_in_db(self): test_class = self.company.get_a_company(1)[0] test_class._testing = True test_class.delete_company_in_db() list_of_companies = self.company.get_all_companies() self.assertEqual(len(list_of_companies), self.total_num_of_companies -1)
def setUp(self): self.info = { "uid": "", "name": "Test Company", "address": "new york, NY", "website": "www.testing.com", "phone": "555-123-4567"} self.company = Company( self.info["uid"], self.info["name"], self.info["address"], self.info["website"], self.info["phone"], testing=True ) self.new_uid = "6" self.new_name = "Testing!!!" self.total_num_of_companies = 5 #Reset the database db = SQL(TEST_DB) db.create_db() db.add_test_data_to_db()