def setUp(self): self.info = { "uid": "", "first_name": "Marcus", "last_name": "Willock", "email": "*****@*****.**", "phone": "111-222-3333", "description": "Nothing interesting", "company_uid": 1 } self.recruiter = Recruiter( 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 = "Testing!!!" self.total_num_of_recruiters = 7 # Setting up the db db = SQL(TEST_DB) db.create_db() db.add_test_data_to_db()
class TestRecruiter(unittest.TestCase): def setUp(self): self.info = { "uid": "", "first_name": "Marcus", "last_name": "Willock", "email": "*****@*****.**", "phone": "111-222-3333", "description": "Nothing interesting", "company_uid": 1 } self.recruiter = Recruiter( 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 = "Testing!!!" self.total_num_of_recruiters = 7 # Setting up the db db = SQL(TEST_DB) db.create_db() db.add_test_data_to_db() def test_properties(self): self.assertEqual(self.recruiter.uid, self.info["uid"]) self.assertEqual(self.recruiter.first_name, self.info["first_name"]) self.assertEqual(self.recruiter.last_name, self.info["last_name"]) self.assertEqual(self.recruiter.email, self.info["email"]) self.assertEqual(self.recruiter.phone, self.info["phone"]) self.assertEqual(self.recruiter.company_uid, self.info["company_uid"]) def test_get_all_recruiters(self): self.assertEqual(7, len(self.recruiter.get_all_recruiters())) def test_a_recruiter(self): data = self.recruiter.get_a_recruiter(1)[0] self.assertEqual("Recruiter1", data.first_name) def test_add_recruiter_to_db(self): self.recruiter.add_recruiter_to_db() self.assertTrue(self.recruiter.uid) def test_update_recruiter_in_db(self): test_class = self.recruiter.get_a_recruiter(1)[0] test_class.first_name = self.new_first_name test_class._testing = True test_class.update_recruiter_in_db() test_class = self.recruiter.get_a_recruiter(1)[0] self.assertEqual(test_class.first_name, self.new_first_name) def test_delete_recruiter_in_db(self): test_class = self.recruiter.get_a_recruiter(1)[0] test_class._testing = True test_class.delete_recruiter_in_db() list_of_recruiters = self.recruiter.get_all_recruiters() self.assertEqual(len(list_of_recruiters), self.total_num_of_recruiters - 1)