def test_default(self): contest = Contest.objects.get(id="c") collector = SubmissionsWithUserDataCollector(contest) submission_data_list = collector.collect_list() submissions = [s.submission_id for s in submission_data_list] self.assertEqual(submissions, [1, 2, 3, 4]) self.assert_correct_submission_data(submission_data_list)
def test_specific_round(self): contest = Contest.objects.get(id="c") round = Round.objects.get(id=3) collector = SubmissionsWithUserDataCollector(contest, round=round) submission_data_list = collector.collect_list() submissions = [s.submission_id for s in submission_data_list] self.assertEqual(submissions, [3]) self.assert_correct_submission_data(submission_data_list)
def test_not_only_final(self): contest = Contest.objects.get(id="c") collector = SubmissionsWithUserDataCollector(contest, only_final=False) submission_data_list = collector.collect_list() submissions = [s.submission_id for s in submission_data_list] # actually all are final self.assertEqual(submissions, [1, 2, 3, 4]) self.assert_correct_submission_data(submission_data_list)