def test_process_submission_message_succesfully( self, mock_rs, mock_basename ): message = { "challenge_pk": self.challenge.pk, "phase_pk": self.challenge_phase.pk, "submission_pk": self.submission.pk, } user_annotation_file_path = ( "mocked/dir/submission_{}/user_annotation_file.txt".format( self.submission.pk ) ) with mock.patch( "scripts.workers.submission_worker.extract_submission_data" ) as mock_esd: submission_worker.process_submission_message(message) mock_esd.assert_called_with(self.submission.pk) submission_worker.process_submission_message(message) mock_rs.assert_called_with( self.challenge.pk, self.challenge_phase, self.submission, user_annotation_file_path, )
def test_process_submission_message_when_submission_does_not_exist( self, mock_esd): message = { "challenge_pk": self.challenge.pk, "phase_pk": self.challenge_phase.pk, "submission_pk": self.submission.pk } mock_esd.return_value = None submission_worker.process_submission_message(message) self.assertEqual(submission_worker.process_submission_message(message), None)
def test_process_submission_message_when_challenge_phase_does_not_exist( self, mock_logger, mock_esd): message = { "challenge_pk": self.challenge.pk, "phase_pk": self.challenge_phase.pk + 999, "submission_pk": self.submission.pk } mock_esd.return_value = self.submission phase_pk = self.challenge_phase.pk + 999 with self.assertRaises(Exception): submission_worker.process_submission_message(message) mock_logger.assert_called_with( "Challenge Phase {} does not exist".format(phase_pk))