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()