[Backend] Add SQS period to serializer (#4238)

gchhablani · web-flow · commit 6de637c31aa4 · 2023-12-12T11:33:01.000+05:30
diff --git a/apps/challenges/serializers.py b/apps/challenges/serializers.py
@@ -93,7 +93,8 @@ class Meta:
             "ephemeral_storage",
             "evaluation_module_error",
             "worker_image_url",
-            "worker_instance_type"
+            "worker_instance_type",
+            "sqs_retention_period"
         )
 
 
@@ -309,7 +310,8 @@ class Meta:
             "ec2_storage",
             "ephemeral_storage",
             "evaluation_module_error",
-            "worker_image_url"
+            "worker_image_url",
+            "sqs_retention_period"
         )
 
 
diff --git a/tests/unit/challenges/test_views.py b/tests/unit/challenges/test_views.py
@@ -194,6 +194,7 @@ def test_get_challenge(self):
                 "ephemeral_storage": self.challenge.ephemeral_storage,
                 "worker_image_url": self.challenge.worker_image_url,
                 "worker_instance_type": self.challenge.worker_instance_type,
+                "sqs_retention_period": self.challenge.sqs_retention_period,
             }
         ]
 
@@ -548,6 +549,7 @@ def test_get_particular_challenge(self):
             "ephemeral_storage": self.challenge.ephemeral_storage,
             "worker_image_url": self.challenge.worker_image_url,
             "worker_instance_type": self.challenge.worker_instance_type,
+            "sqs_retention_period": self.challenge.sqs_retention_period,
         }
         response = self.client.get(self.url, {})
         self.assertEqual(response.data, expected)
@@ -649,6 +651,7 @@ def test_update_challenge_when_user_is_its_creator(self):
             "ephemeral_storage": self.challenge.ephemeral_storage,
             "worker_image_url": self.challenge.worker_image_url,
             "worker_instance_type": self.challenge.worker_instance_type,
+            "sqs_retention_period": self.challenge.sqs_retention_period,
         }
         response = self.client.put(
             self.url, {"title": new_title, "description": new_description}
@@ -776,6 +779,7 @@ def test_particular_challenge_partial_update(self):
             "ephemeral_storage": self.challenge.ephemeral_storage,
             "worker_image_url": self.challenge.worker_image_url,
             "worker_instance_type": self.challenge.worker_instance_type,
+            "sqs_retention_period": self.challenge.sqs_retention_period,
         }
         response = self.client.patch(self.url, self.partial_update_data)
         self.assertEqual(response.data, expected)
@@ -852,6 +856,7 @@ def test_particular_challenge_update(self):
             "ephemeral_storage": self.challenge.ephemeral_storage,
             "worker_image_url": self.challenge.worker_image_url,
             "worker_instance_type": self.challenge.worker_instance_type,
+            "sqs_retention_period": self.challenge.sqs_retention_period,
         }
         response = self.client.put(self.url, self.data)
         self.assertEqual(response.data, expected)
@@ -1444,6 +1449,7 @@ def test_get_past_challenges(self):
                 "ephemeral_storage": self.challenge3.ephemeral_storage,
                 "worker_image_url": self.challenge3.worker_image_url,
                 "worker_instance_type": self.challenge3.worker_instance_type,
+                "sqs_retention_period": self.challenge3.sqs_retention_period,
             }
         ]
         response = self.client.get(self.url, {}, format="json")
@@ -1526,6 +1532,7 @@ def test_get_present_challenges(self):
                 "ephemeral_storage": self.challenge.ephemeral_storage,
                 "worker_image_url": self.challenge.worker_image_url,
                 "worker_instance_type": self.challenge.worker_instance_type,
+                "sqs_retention_period": self.challenge.sqs_retention_period,
             }
         ]
         response = self.client.get(self.url, {}, format="json")
@@ -1608,6 +1615,7 @@ def test_get_future_challenges(self):
                 "ephemeral_storage": self.challenge4.ephemeral_storage,
                 "worker_image_url": self.challenge4.worker_image_url,
                 "worker_instance_type": self.challenge4.worker_instance_type,
+                "sqs_retention_period": self.challenge4.sqs_retention_period,
             }
         ]
         response = self.client.get(self.url, {}, format="json")
@@ -1690,6 +1698,7 @@ def test_get_all_challenges(self):
                 "ephemeral_storage": self.challenge3.ephemeral_storage,
                 "worker_image_url": self.challenge3.worker_image_url,
                 "worker_instance_type": self.challenge3.worker_instance_type,
+                "sqs_retention_period": self.challenge3.sqs_retention_period,
             },
             {
                 "id": self.challenge3.pk,
@@ -1756,6 +1765,7 @@ def test_get_all_challenges(self):
                 "ephemeral_storage": self.challenge3.ephemeral_storage,
                 "worker_image_url": self.challenge3.worker_image_url,
                 "worker_instance_type": self.challenge3.worker_instance_type,
+                "sqs_retention_period": self.challenge3.sqs_retention_period,
             },
             {
                 "id": self.challenge2.pk,
@@ -1822,6 +1832,7 @@ def test_get_all_challenges(self):
                 "ephemeral_storage": self.challenge2.ephemeral_storage,
                 "worker_image_url": self.challenge2.worker_image_url,
                 "worker_instance_type": self.challenge2.worker_instance_type,
+                "sqs_retention_period": self.challenge2.sqs_retention_period,
             },
         ]
         response = self.client.get(self.url, {}, format="json")
@@ -1959,6 +1970,7 @@ def test_get_featured_challenges(self):
                 "ephemeral_storage": self.challenge3.ephemeral_storage,
                 "worker_image_url": self.challenge3.worker_image_url,
                 "worker_instance_type": self.challenge3.worker_instance_type,
+                "sqs_retention_period": self.challenge3.sqs_retention_period,
             }
         ]
         response = self.client.get(self.url, {}, format="json")
@@ -2120,6 +2132,7 @@ def test_get_challenge_by_pk_when_user_is_challenge_host(self):
             "ephemeral_storage": self.challenge3.ephemeral_storage,
             "worker_image_url": self.challenge3.worker_image_url,
             "worker_instance_type": self.challenge3.worker_instance_type,
+            "sqs_retention_period": self.challenge3.sqs_retention_period,
         }
 
         response = self.client.get(self.url, {})
@@ -2210,6 +2223,7 @@ def test_get_challenge_by_pk_when_user_is_participant(self):
             "ephemeral_storage": self.challenge4.ephemeral_storage,
             "worker_image_url": self.challenge4.worker_image_url,
             "worker_instance_type": self.challenge4.worker_instance_type,
+            "sqs_retention_period": self.challenge4.sqs_retention_period,
         }
 
         self.client.force_authenticate(user=self.user1)
@@ -2360,6 +2374,7 @@ def test_get_challenge_when_host_team_is_given(self):
                 "ephemeral_storage": self.challenge2.ephemeral_storage,
                 "worker_image_url": self.challenge2.worker_image_url,
                 "worker_instance_type": self.challenge2.worker_instance_type,
+                "sqs_retention_period": self.challenge2.sqs_retention_period,
             }
         ]
 
@@ -2438,6 +2453,7 @@ def test_get_challenge_when_participant_team_is_given(self):
                 "ephemeral_storage": self.challenge2.ephemeral_storage,
                 "worker_image_url": self.challenge2.worker_image_url,
                 "worker_instance_type": self.challenge2.worker_instance_type,
+                "sqs_retention_period": self.challenge2.sqs_retention_period,
             }
         ]
 
@@ -2516,6 +2532,7 @@ def test_get_challenge_when_mode_is_participant(self):
                 "ephemeral_storage": self.challenge2.ephemeral_storage,
                 "worker_image_url": self.challenge2.worker_image_url,
                 "worker_instance_type": self.challenge2.worker_instance_type,
+                "sqs_retention_period": self.challenge2.sqs_retention_period,
             }
         ]
 
@@ -2592,6 +2609,7 @@ def test_get_challenge_when_mode_is_host(self):
                 "ephemeral_storage": self.challenge.ephemeral_storage,
                 "worker_image_url": self.challenge.worker_image_url,
                 "worker_instance_type": self.challenge.worker_instance_type,
+                "sqs_retention_period": self.challenge.sqs_retention_period,
             },
             {
                 "id": self.challenge2.pk,
@@ -2658,6 +2676,7 @@ def test_get_challenge_when_mode_is_host(self):
                 "ephemeral_storage": self.challenge2.ephemeral_storage,
                 "worker_image_url": self.challenge2.worker_image_url,
                 "worker_instance_type": self.challenge2.worker_instance_type,
+                "sqs_retention_period": self.challenge2.sqs_retention_period,
             },
         ]
 
diff --git a/tests/unit/participants/test_views.py b/tests/unit/participants/test_views.py
@@ -883,6 +883,7 @@ def test_get_teams_and_corresponding_challenges_for_a_participant(self):
                         "evaluation_module_error": self.challenge1.evaluation_module_error,
                         "worker_image_url": self.challenge1.worker_image_url,
                         "worker_instance_type": self.challenge1.worker_instance_type,
+                        "sqs_retention_period": self.challenge1.sqs_retention_period,
                     },
                     "participant_team": {
                         "id": self.participant_team.id,
@@ -976,6 +977,7 @@ def test_get_participant_team_challenge_list(self):
                 "evaluation_module_error": self.challenge1.evaluation_module_error,
                 "worker_image_url": self.challenge1.worker_image_url,
                 "worker_instance_type": self.challenge1.worker_instance_type,
+                "sqs_retention_period": self.challenge1.sqs_retention_period,
             }
         ]
 

Original file line number	Diff line number	Diff line change
`@@ -194,6 +194,7 @@ def test_get_challenge(self):`
`194`	`194`	`"ephemeral_storage": self.challenge.ephemeral_storage,`
`195`	`195`	`"worker_image_url": self.challenge.worker_image_url,`
`196`	`196`	`"worker_instance_type": self.challenge.worker_instance_type,`
	`197`	`+ "sqs_retention_period": self.challenge.sqs_retention_period,`
`197`	`198`	`}`
`198`	`199`	`]`
`199`	`200`
`@@ -548,6 +549,7 @@ def test_get_particular_challenge(self):`
`548`	`549`	`"ephemeral_storage": self.challenge.ephemeral_storage,`
`549`	`550`	`"worker_image_url": self.challenge.worker_image_url,`
`550`	`551`	`"worker_instance_type": self.challenge.worker_instance_type,`
	`552`	`+ "sqs_retention_period": self.challenge.sqs_retention_period,`
`551`	`553`	`}`
`552`	`554`	`response = self.client.get(self.url, {})`
`553`	`555`	`self.assertEqual(response.data, expected)`
`@@ -649,6 +651,7 @@ def test_update_challenge_when_user_is_its_creator(self):`
`649`	`651`	`"ephemeral_storage": self.challenge.ephemeral_storage,`
`650`	`652`	`"worker_image_url": self.challenge.worker_image_url,`
`651`	`653`	`"worker_instance_type": self.challenge.worker_instance_type,`
	`654`	`+ "sqs_retention_period": self.challenge.sqs_retention_period,`
`652`	`655`	`}`
`653`	`656`	`response = self.client.put(`
`654`	`657`	`self.url, {"title": new_title, "description": new_description}`
`@@ -776,6 +779,7 @@ def test_particular_challenge_partial_update(self):`
`776`	`779`	`"ephemeral_storage": self.challenge.ephemeral_storage,`
`777`	`780`	`"worker_image_url": self.challenge.worker_image_url,`
`778`	`781`	`"worker_instance_type": self.challenge.worker_instance_type,`
	`782`	`+ "sqs_retention_period": self.challenge.sqs_retention_period,`
`779`	`783`	`}`
`780`	`784`	`response = self.client.patch(self.url, self.partial_update_data)`
`781`	`785`	`self.assertEqual(response.data, expected)`
`@@ -852,6 +856,7 @@ def test_particular_challenge_update(self):`
`852`	`856`	`"ephemeral_storage": self.challenge.ephemeral_storage,`
`853`	`857`	`"worker_image_url": self.challenge.worker_image_url,`
`854`	`858`	`"worker_instance_type": self.challenge.worker_instance_type,`
	`859`	`+ "sqs_retention_period": self.challenge.sqs_retention_period,`
`855`	`860`	`}`
`856`	`861`	`response = self.client.put(self.url, self.data)`
`857`	`862`	`self.assertEqual(response.data, expected)`
`@@ -1444,6 +1449,7 @@ def test_get_past_challenges(self):`
`1444`	`1449`	`"ephemeral_storage": self.challenge3.ephemeral_storage,`
`1445`	`1450`	`"worker_image_url": self.challenge3.worker_image_url,`
`1446`	`1451`	`"worker_instance_type": self.challenge3.worker_instance_type,`
	`1452`	`+ "sqs_retention_period": self.challenge3.sqs_retention_period,`
`1447`	`1453`	`}`
`1448`	`1454`	`]`
`1449`	`1455`	`response = self.client.get(self.url, {}, format="json")`
`@@ -1526,6 +1532,7 @@ def test_get_present_challenges(self):`
`1526`	`1532`	`"ephemeral_storage": self.challenge.ephemeral_storage,`
`1527`	`1533`	`"worker_image_url": self.challenge.worker_image_url,`
`1528`	`1534`	`"worker_instance_type": self.challenge.worker_instance_type,`
	`1535`	`+ "sqs_retention_period": self.challenge.sqs_retention_period,`
`1529`	`1536`	`}`
`1530`	`1537`	`]`
`1531`	`1538`	`response = self.client.get(self.url, {}, format="json")`
`@@ -1608,6 +1615,7 @@ def test_get_future_challenges(self):`
`1608`	`1615`	`"ephemeral_storage": self.challenge4.ephemeral_storage,`
`1609`	`1616`	`"worker_image_url": self.challenge4.worker_image_url,`
`1610`	`1617`	`"worker_instance_type": self.challenge4.worker_instance_type,`
	`1618`	`+ "sqs_retention_period": self.challenge4.sqs_retention_period,`
`1611`	`1619`	`}`
`1612`	`1620`	`]`
`1613`	`1621`	`response = self.client.get(self.url, {}, format="json")`
`@@ -1690,6 +1698,7 @@ def test_get_all_challenges(self):`
`1690`	`1698`	`"ephemeral_storage": self.challenge3.ephemeral_storage,`
`1691`	`1699`	`"worker_image_url": self.challenge3.worker_image_url,`
`1692`	`1700`	`"worker_instance_type": self.challenge3.worker_instance_type,`
	`1701`	`+ "sqs_retention_period": self.challenge3.sqs_retention_period,`
`1693`	`1702`	`},`
`1694`	`1703`	`{`
`1695`	`1704`	`"id": self.challenge3.pk,`
`@@ -1756,6 +1765,7 @@ def test_get_all_challenges(self):`
`1756`	`1765`	`"ephemeral_storage": self.challenge3.ephemeral_storage,`
`1757`	`1766`	`"worker_image_url": self.challenge3.worker_image_url,`
`1758`	`1767`	`"worker_instance_type": self.challenge3.worker_instance_type,`
	`1768`	`+ "sqs_retention_period": self.challenge3.sqs_retention_period,`
`1759`	`1769`	`},`
`1760`	`1770`	`{`
`1761`	`1771`	`"id": self.challenge2.pk,`
`@@ -1822,6 +1832,7 @@ def test_get_all_challenges(self):`
`1822`	`1832`	`"ephemeral_storage": self.challenge2.ephemeral_storage,`
`1823`	`1833`	`"worker_image_url": self.challenge2.worker_image_url,`
`1824`	`1834`	`"worker_instance_type": self.challenge2.worker_instance_type,`
	`1835`	`+ "sqs_retention_period": self.challenge2.sqs_retention_period,`
`1825`	`1836`	`},`
`1826`	`1837`	`]`
`1827`	`1838`	`response = self.client.get(self.url, {}, format="json")`
`@@ -1959,6 +1970,7 @@ def test_get_featured_challenges(self):`
`1959`	`1970`	`"ephemeral_storage": self.challenge3.ephemeral_storage,`
`1960`	`1971`	`"worker_image_url": self.challenge3.worker_image_url,`
`1961`	`1972`	`"worker_instance_type": self.challenge3.worker_instance_type,`
	`1973`	`+ "sqs_retention_period": self.challenge3.sqs_retention_period,`
`1962`	`1974`	`}`
`1963`	`1975`	`]`
`1964`	`1976`	`response = self.client.get(self.url, {}, format="json")`
`@@ -2120,6 +2132,7 @@ def test_get_challenge_by_pk_when_user_is_challenge_host(self):`
`2120`	`2132`	`"ephemeral_storage": self.challenge3.ephemeral_storage,`
`2121`	`2133`	`"worker_image_url": self.challenge3.worker_image_url,`
`2122`	`2134`	`"worker_instance_type": self.challenge3.worker_instance_type,`
	`2135`	`+ "sqs_retention_period": self.challenge3.sqs_retention_period,`
`2123`	`2136`	`}`
`2124`	`2137`
`2125`	`2138`	`response = self.client.get(self.url, {})`
`@@ -2210,6 +2223,7 @@ def test_get_challenge_by_pk_when_user_is_participant(self):`
`2210`	`2223`	`"ephemeral_storage": self.challenge4.ephemeral_storage,`
`2211`	`2224`	`"worker_image_url": self.challenge4.worker_image_url,`
`2212`	`2225`	`"worker_instance_type": self.challenge4.worker_instance_type,`
	`2226`	`+ "sqs_retention_period": self.challenge4.sqs_retention_period,`
`2213`	`2227`	`}`
`2214`	`2228`
`2215`	`2229`	`self.client.force_authenticate(user=self.user1)`
`@@ -2360,6 +2374,7 @@ def test_get_challenge_when_host_team_is_given(self):`
`2360`	`2374`	`"ephemeral_storage": self.challenge2.ephemeral_storage,`
`2361`	`2375`	`"worker_image_url": self.challenge2.worker_image_url,`
`2362`	`2376`	`"worker_instance_type": self.challenge2.worker_instance_type,`
	`2377`	`+ "sqs_retention_period": self.challenge2.sqs_retention_period,`
`2363`	`2378`	`}`
`2364`	`2379`	`]`
`2365`	`2380`
`@@ -2438,6 +2453,7 @@ def test_get_challenge_when_participant_team_is_given(self):`
`2438`	`2453`	`"ephemeral_storage": self.challenge2.ephemeral_storage,`
`2439`	`2454`	`"worker_image_url": self.challenge2.worker_image_url,`
`2440`	`2455`	`"worker_instance_type": self.challenge2.worker_instance_type,`
	`2456`	`+ "sqs_retention_period": self.challenge2.sqs_retention_period,`
`2441`	`2457`	`}`
`2442`	`2458`	`]`
`2443`	`2459`
`@@ -2516,6 +2532,7 @@ def test_get_challenge_when_mode_is_participant(self):`
`2516`	`2532`	`"ephemeral_storage": self.challenge2.ephemeral_storage,`
`2517`	`2533`	`"worker_image_url": self.challenge2.worker_image_url,`
`2518`	`2534`	`"worker_instance_type": self.challenge2.worker_instance_type,`
	`2535`	`+ "sqs_retention_period": self.challenge2.sqs_retention_period,`
`2519`	`2536`	`}`
`2520`	`2537`	`]`
`2521`	`2538`
`@@ -2592,6 +2609,7 @@ def test_get_challenge_when_mode_is_host(self):`
`2592`	`2609`	`"ephemeral_storage": self.challenge.ephemeral_storage,`
`2593`	`2610`	`"worker_image_url": self.challenge.worker_image_url,`
`2594`	`2611`	`"worker_instance_type": self.challenge.worker_instance_type,`
	`2612`	`+ "sqs_retention_period": self.challenge.sqs_retention_period,`
`2595`	`2613`	`},`
`2596`	`2614`	`{`
`2597`	`2615`	`"id": self.challenge2.pk,`
`@@ -2658,6 +2676,7 @@ def test_get_challenge_when_mode_is_host(self):`
`2658`	`2676`	`"ephemeral_storage": self.challenge2.ephemeral_storage,`
`2659`	`2677`	`"worker_image_url": self.challenge2.worker_image_url,`
`2660`	`2678`	`"worker_instance_type": self.challenge2.worker_instance_type,`
	`2679`	`+ "sqs_retention_period": self.challenge2.sqs_retention_period,`
`2661`	`2680`	`},`
`2662`	`2681`	`]`
`2663`	`2682`
Original file line number	Diff line number	Diff line change
`@@ -883,6 +883,7 @@ def test_get_teams_and_corresponding_challenges_for_a_participant(self):`
`883`	`883`	`"evaluation_module_error": self.challenge1.evaluation_module_error,`
`884`	`884`	`"worker_image_url": self.challenge1.worker_image_url,`
`885`	`885`	`"worker_instance_type": self.challenge1.worker_instance_type,`
	`886`	`+ "sqs_retention_period": self.challenge1.sqs_retention_period,`
`886`	`887`	`},`
`887`	`888`	`"participant_team": {`
`888`	`889`	`"id": self.participant_team.id,`
`@@ -976,6 +977,7 @@ def test_get_participant_team_challenge_list(self):`
`976`	`977`	`"evaluation_module_error": self.challenge1.evaluation_module_error,`
`977`	`978`	`"worker_image_url": self.challenge1.worker_image_url,`
`978`	`979`	`"worker_instance_type": self.challenge1.worker_instance_type,`
	`980`	`+ "sqs_retention_period": self.challenge1.sqs_retention_period,`
`979`	`981`	`}`
`980`	`982`	`]`
`981`	`983`