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)