def test_get_submit_from_xml(self, par): w = SubmitWalker('a') w.contest_id = '7' w.database = Mock() w.database.get_submit_info.return_value = Mock(problem_id='1', user_id='11', lang_id='2', timestamp=42) res = w._get_submit_from_xml('filename') self.assertEqual(res.submit_id, '5') self.assertEqual(res.problem_id, ('7', '1')) self.assertEqual(res.user_id, '11') self.assertEqual(res.lang_id, '2') self.assertEqual(res.outcome, 'OK') self.assertEqual(res.scoring, 'ACM') self.assertEqual(res.timestamp, 42) self.assertEqual((res.runs[0].problem_id, res.runs[0].submit_id, res.runs[0].case_id, res.runs[0].real_time, res.runs[0].time, res.runs[0].outcome), (('7', '1'), '5', 1, '2', '3', 'OK')) self.assertEqual((res.runs[1].problem_id, res.runs[1].submit_id, res.runs[1].case_id, res.runs[1].real_time, res.runs[1].time, res.runs[1].outcome), (('7', '1'), '5', 2, '4', '5', 'WA')) par.assert_called_once_with('filename')
def test_get_submit_from_xml_error(self, par): w = SubmitWalker('a') res = w._get_submit_from_xml('filename') self.assertIsNone(res)