def testRaisesExceptionForNotEnoughVenues(self):
generate_objects.valid_venue()
for i in range(0, 20):
debate = generate_objects.valid_debate()
debate.save()
controller = VenueMapper()
with self.assertRaises(TournamentStateException):
controller.map_venues(1)
def testEachDebateHasVenue(self):
for i in range(0, 12):
venue = generate_objects.valid_venue()
venue.save()
debate = generate_objects.valid_debate()
debate.save()
controller = VenueMapper()
controller.map_venues(1)
for debate in Debate.objects.all().filter(round=1):
self.assertIsNotNone(debate.venue)