test: added test for get_team_ids_by_team_submission_uuid (#159)

* test: added test for get_team_ids_by_team_submission_uuid
openedx · Mar 18, 2022 · 8510716 · 8510716
1 parent d27861e
commit 8510716
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/submissions/tests/test_team_api.py b/submissions/tests/test_team_api.py
@@ -653,3 +653,22 @@ def test_get_team_submission_student_ids__no_team_submission(self):
             team_api.get_team_submission_student_ids('aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee')
         with self.assertRaises(TeamSubmissionNotFoundError):
             team_api.get_team_submission_student_ids(None)
+
+    def test_get_team_ids_by_team_submission_uuid(self):
+        team_submissions = [
+            TeamSubmissionFactory.create() for _ in range(5)
+        ]
+        assert team_api.get_team_ids_by_team_submission_uuid([]) == {}
+
+        actual = team_api.get_team_ids_by_team_submission_uuid([
+            team_submissions[0].uuid,
+            team_submissions[1].uuid,
+            team_submissions[4].uuid,
+        ])
+        expected = {
+            str(team_submissions[0].uuid): team_submissions[0].team_id,
+            str(team_submissions[1].uuid): team_submissions[1].team_id,
+            str(team_submissions[4].uuid): team_submissions[4].team_id,
+        }
+
+        assert expected == actual