def test_common(self, some, useless, args): cursor = Mock() fill_db_from_contest_xml('asdf', cursor, 'lksh') resulting_requests = [] for one_call in cursor.execute.call_args_list: format_string, data_tuple = tuple(one_call)[0] resulting_requests.append(self.prepare_request(format_string, data_tuple)) good_requests = ['UPDATE Contests SET name = I am contest name #1 WHERE origin = lksh AND contest_id = 123456', 'UPDATE Contests SET name = Contest name #2 WHERE origin = lksh AND contest_id = 179179', 'UPDATE Contests SET name = Yo, i am contest name #3! WHERE origin = lksh AND contest_id = 031337'] self.assertEqual(resulting_requests, good_requests)
def fill_contests_names(sqlite_cursor, contests_info_dir, origin): if contests_info_dir: logging.info('Filling contests names') fill_db_from_contest_xml(contests_info_dir, sqlite_cursor, origin) logging.info('Contests names were filled')