Fix test_availability tests

TabbycatDebate · Sep 14, 2016 · 8f05f0b · 8f05f0b
1 parent c09fc60
commit 8f05f0b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tabbycat/availability/tests/test_availability.py b/tabbycat/availability/tests/test_availability.py
@@ -1,4 +1,4 @@
-from participants.models import Adjudicator
+from participants.models import Adjudicator, Institution
 from tournaments.models import Round
 from utils.tests import BaseDebateTestCase
 
@@ -27,15 +27,19 @@ def test_one_disabled(self):
         self.assertEqual(7, self.round.active_adjudicators.count())
 
     def test_activate_all(self):
+        Adjudicator.objects.create(institution=Institution.objects.get(code="INS0"), name="Unattached")
         self.t.preferences['league_options__share_adjs'] = False
+        self.t.preferences['league_options__share_venues'] = False
         activate_all(self.round)
         self.assertEqual(8, self.round.active_adjudicators.count())
         self.assertEqual(12, self.round.active_teams.count())
         self.assertEqual(8, self.round.active_venues.count())
 
     def test_activate_relevant(self):
+        Adjudicator.objects.create(institution=Institution.objects.get(code="INS0"), name="Unattached")
         self.t.preferences['league_options__share_adjs'] = True
+        self.t.preferences['league_options__share_venues'] = True
         activate_all(self.round)
-        self.assertEqual(8, self.round.active_adjudicators.count())
+        self.assertEqual(9, self.round.active_adjudicators.count())
         self.assertEqual(12, self.round.active_teams.count())
         self.assertEqual(16, self.round.active_venues.count())