def test_db(self, execute, select, table_exists, connect):
table_exists.return_value = True
select.return_value = [["1", "", read_file_as_string("../files/data.pkl")]]
execute.return_value = True
res = IM.get_data_from_db("mysql://*****:*****@server/db_name")
self.assertEqual(len(res), 1)
inf = InfrastructureInfo()
inf.id = "1"
success = IM.save_data_to_db("mysql://*****:*****@server/db_name", {"1": inf})
self.assertTrue(success)
