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)
