class TestLocal(TestCase):
# Wesley
def setUp(self):
self.db = DBLocal()
self.db.connect("TestLocal")
self.db.create_table()
# Wesley
def tearDown(self):
self.db.drop_table()
self.db.close()
self.db = None
# Wesley
def test_db_insert_values(self):
"""Create a table, insert data and check the return data is what was inserted"""
expected = [(1, 'dskjhflkasdjlfkj23543'),
(2, 'dskjhflkasdjlasdfdsafkj23543')]
self.db.insert_record("dskjhflkasdjlfkj23543")
self.db.insert_record("dskjhflkasdjlasdfdsafkj23543")
self.db.commit()
result = self.db.get_db()
self.assertEqual(expected, result)
def test_db_get(self):
"""Getting an empty database"""
expected = list()
result = self.db.get_db()
self.assertEqual(expected, result)
def test_db_insert_dict(self):
"""Insert a dictionary into the database"""
expected = [(1, 'dskjhflkasdjlfkj23543'),
(2, "dskjhflkasdjlasdfdsafkj23543")]
data = {1: 'dskjhflkasdjlfkj23543', 2: "dskjhflkasdjlasdfdsafkj23543"}
self.db.insert_dictionary(data)
result = self.db.get_db()
self.assertEqual(expected, result)
def test_db_update(self):
"""Insert a dictionary into the database"""
expected = [(1, 'dskjhflkasdjlfkj23543'), (2, "changed")]
data = {1: 'dskjhflkasdjlfkj23543', 2: "dskjhflkasdjlasdfdsafkj23543"}
self.db.insert_dictionary(data)
self.db.update_record(2, "changed")
result = self.db.get_db()
self.assertEqual(expected, result)
def test_db_delete(self):
"""Insert a dictionary into the database"""
expected = [(1, 'dskjhflkasdjlfkj23543')]
data = {1: 'dskjhflkasdjlfkj23543', 2: "dskjhflkasdjlasdfdsafkj23543"}
self.db.insert_dictionary(data)
self.db.delete_record("2")
result = self.db.get_db()
self.assertEqual(expected, result)
class TestUnpickler(TestCase):
def setUp(self):
self.local = DBLocal()
self.local.connect()
self.local.create_table()
self.maxDiff = None
def tearDown(self):
self.local.close()
self.local = None
def test_unpickle_dict(self):
expected = {
1: {
"1ID": "A23",
"Gender": "Male",
"Age": 22,
"Sales": 2445,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
2: {
"IhD": "A2f3",
"Gender": "Male",
"Age": 23,
"Sales": 2565,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
3: {
"IjD": "Aa23",
"Gender": "Female",
"Age": 25,
"Sales": 25,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
},
4: {
"IgD": "A23",
"Gender": "Female",
"Age": 26,
"Sales": 225,
"BMI": "normal",
"salary": 20,
"Birthday": "24/06/1995"
}
}
pickle = Pickler.pickle_dictionary_values(expected)
self.local.insert_dictionary(pickle)
result = Unpickler.unpickle_dictionary(self.local.get_db())
self.assertEqual(expected, result)