def test_knows_currently_open_survey(self):
country = LocationType.objects.create(name='Country', slug='country')
district = LocationType.objects.create(name='District',
slug='district')
uganda = Location.objects.create(name="Uganda", type=country)
kampala = Location.objects.create(name="Kampala",
type=district,
tree_parent=uganda)
open_survey = Survey.objects.create(name="open survey",
description="open survey")
closed_survey = Survey.objects.create(name="closed survey",
description="closed survey")
another_closed_survey = Survey.objects.create(
name="another closed survey", description="another closed survey")
open_batch = Batch.objects.create(order=1,
name="Open Batch",
survey=open_survey)
closed_batch = Batch.objects.create(order=2,
name="Closed Batch",
survey=closed_survey)
another_closed_batch = Batch.objects.create(
order=3, name="Another Closed Batch", survey=another_closed_survey)
open_batch.open_for_location(kampala)
self.assertEqual(open_survey, Survey.currently_open_survey())
self.assertNotEqual(closed_survey, Survey.currently_open_survey())
self.assertNotEqual(another_closed_survey,
Survey.currently_open_survey())
def test_knows_currently_open_survey(self):
country = LocationType.objects.create(name='Country', slug='country')
district = LocationType.objects.create(name='District', slug='district')
uganda = Location.objects.create(name="Uganda", type=country)
kampala = Location.objects.create(name="Kampala", type=district, tree_parent=uganda)
open_survey = Survey.objects.create(name="open survey", description="open survey")
closed_survey = Survey.objects.create(name="closed survey", description="closed survey")
another_closed_survey = Survey.objects.create(name="another closed survey", description="another closed survey")
open_batch = Batch.objects.create(order=1, name="Open Batch", survey=open_survey)
closed_batch = Batch.objects.create(order=2, name="Closed Batch", survey=closed_survey)
another_closed_batch = Batch.objects.create(order=3, name="Another Closed Batch", survey=another_closed_survey)
open_batch.open_for_location(kampala)
self.assertEqual(open_survey, Survey.currently_open_survey())
self.assertNotEqual(closed_survey, Survey.currently_open_survey())
self.assertNotEqual(another_closed_survey, Survey.currently_open_survey())
