Fix #4524: Add test cases for host models (#4525)

* add test cases for host models

* Format test_models.py

---------

Co-authored-by: Rishabh Jain <[email protected]>

Author: Zahed-Riyaz

tests/unit/hosts/test_models.py

@@ -42,3 +42,43 @@ def test__str__(self):
             "{}: {}".format(team_name, created_by),
             self.challenge_host_team.__str__(),
         )
+
+    def test_get_all_challenge_host_email(self):
+        first_user = User.objects.create(
+            username="user1", email="[email protected]", password="password"
+        )
+        second_user = User.objects.create(
+            username="user2", email="[email protected]", password="password"
+        )
+        ChallengeHost.objects.create(
+            user=first_user,
+            team_name=self.challenge_host_team,
+            status=ChallengeHost.ACCEPTED,
+            permissions=ChallengeHost.READ,
+        )
+        ChallengeHost.objects.create(
+            user=second_user,
+            team_name=self.challenge_host_team,
+            status=ChallengeHost.PENDING,
+            permissions=ChallengeHost.WRITE,
+        )
+
+        other_team = ChallengeHostTeam.objects.create(
+            team_name="Other Team", created_by=self.user
+        )
+        other_user = User.objects.create(
+            username="other", email="[email protected]", password="password"
+        )
+        ChallengeHost.objects.create(
+            user=other_user,
+            team_name=other_team,
+            status=ChallengeHost.ACCEPTED,
+            permissions=ChallengeHost.ADMIN,
+        )
+        emails = self.challenge_host_team.get_all_challenge_host_email()
+        self.assertEqual(len(emails), 3)
+        self.assertIn("[email protected]", emails)
+        self.assertIn("[email protected]", emails)
+        self.assertIn("[email protected]", emails)
+        self.assertNotIn("[email protected]", emails)
+        self.assertIsInstance(emails, list)