def test_filter_documentation_org_org_admin(self): group_org_admin = factories.Group(name=ROLE_ORGANIZATION_ADMIN) self.tola_user.user.groups.add(group_org_admin) country1 = factories.Country() wflvl1_1 = factories.WorkflowLevel1( name='WorkflowLevel1_1', country=[country1], organization=self.tola_user.organization) WorkflowTeam.objects.create(workflow_user=self.tola_user, workflowlevel1=wflvl1_1) documentation1 = factories.Documentation(name='Document 1', workflowlevel1=wflvl1_1) country2 = factories.Country(country='Brazil', code='BR') wflvl1_2 = factories.WorkflowLevel1( name='WorkflowLevel1_2', country=[country2], organization=self.tola_user.organization) WorkflowTeam.objects.create(workflow_user=self.tola_user, workflowlevel1=wflvl1_2) factories.Documentation(name='Document 2', workflowlevel1=wflvl1_2) request = self.factory.get('/api/documentation' '/?workflowlevel1__country__country=%s' % country1.country) request.user = self.tola_user.user view = DocumentationViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['name'], documentation1.name)
def test_filter_collecteddata_country_org_admin(self): group_org_admin = factories.Group(name=ROLE_ORGANIZATION_ADMIN) self.tola_user.user.groups.add(group_org_admin) country1 = factories.Country(country='Brazil', code='BR') country2 = factories.Country(country='Germany', code='DE') wkflvl1_1 = factories.WorkflowLevel1(country=[country1]) wkflvl1_2 = factories.WorkflowLevel1(country=[country2]) WorkflowTeam.objects.create(workflow_user=self.tola_user, workflowlevel1=wkflvl1_1) WorkflowTeam.objects.create(workflow_user=self.tola_user, workflowlevel1=wkflvl1_2) indicator1 = factories.Indicator(workflowlevel1=[wkflvl1_1]) indicator2 = factories.Indicator(workflowlevel1=[wkflvl1_2]) collecteddata1 = factories.CollectedData(data_uuid=111, indicator=indicator1) factories.CollectedData(data_uuid=222, indicator=indicator2) request = self.factory.get( '/api/collecteddata/?indicator__workflowlevel1__country__country=%s' % country1.country) request.user = self.tola_user.user view = CollectedDataViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['data_uuid'], str(collecteddata1.data_uuid))
def test_filter_checklist_country_org_admin(self): group_org_admin = factories.Group(name=ROLE_ORGANIZATION_ADMIN) self.tola_user.user.groups.add(group_org_admin) country1 = factories.Country(country='Brazil', code='BR') country2 = factories.Country(country='Germany', code='DE') wkflvl1_1 = factories.WorkflowLevel1( organization=self.tola_user.organization, country=[country1]) wkflvl1_2 = factories.WorkflowLevel1( organization=self.tola_user.organization, country=[country2]) wkflvl2_1 = factories.WorkflowLevel2(workflowlevel1=wkflvl1_1) wkflvl2_2 = factories.WorkflowLevel2(workflowlevel1=wkflvl1_2) checklist1 = factories.Checklist(name='Checklist A', workflowlevel2=wkflvl2_1) factories.Checklist(name='Checklist B', workflowlevel2=wkflvl2_2) request = self.factory.get( '/api/checklist' '/?workflowlevel2__workflowlevel1__country__country=%s' % country1.country) request.user = self.tola_user.user view = ChecklistViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['name'], checklist1.name)
def test_filter_level_normaluser(self): country1 = factories.Country(country='Brazil', code='BR') country2 = factories.Country() level1 = factories.Level(name='Level 1', country=country1, organization=self.tola_user.organization) factories.Level(name='Level 2', country=country2, organization=self.tola_user.organization) request = self.factory.get('/api/level/?country__country=%s' % country1.country) request.user = self.tola_user.user view = LevelViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['name'], level1.name)
def test_filter_strategicobjective_normaluser(self): country1 = factories.Country(country='Brazil', code='BR') country2 = factories.Country() s_obj1 = factories.StrategicObjective( name='StrategicObjective1', country=country1, organization=self.tola_user.organization) factories.StrategicObjective(name='StrategicObjective2', country=country2, organization=self.tola_user.organization) request = self.factory.get( '/api/strategicobjective/?country__country=%s' % country1.country) request.user = self.tola_user.user view = StrategicObjectiveViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['name'], s_obj1.name)
def setUp(self): self.country = factories.Country() SiteProfile.objects.bulk_create([ SiteProfile(name='SiteProfile1', country=self.country), SiteProfile(name='SiteProfile2', country=self.country), ]) factories.Indicator.create_batch(2) self.factory = APIRequestFactory() self.tola_user = factories.TolaUser()
def test_filter_workflowlevel2_wkflvl1_country_superuser(self): self.tola_user.user.is_staff = True self.tola_user.user.is_superuser = True self.tola_user.user.save() country1 = factories.Country(country='Brazil', code='BR') country2 = factories.Country(country='Germany', code='DE') wkflvl1_1 = factories.WorkflowLevel1(country=[country1]) wkflvl1_2 = factories.WorkflowLevel1(country=[country2]) wkflvl2 = factories.WorkflowLevel2(workflowlevel1=wkflvl1_1) factories.WorkflowLevel2(name='Develop brief survey', workflowlevel1=wkflvl1_2) request = self.factory.get( '/api/workflowlevel2/?workflowlevel1__country__country=%s' % country1.country) request.user = self.tola_user.user view = WorkflowLevel2ViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['name'], wkflvl2.name)
def test_filter_siteprofile_normaluser(self): site1 = factories.SiteProfile(name='SiteProfile1', country=self.country, organization=self.tola_user.organization) factories.SiteProfile(name='SiteProfile2', country=factories.Country(), organization=self.tola_user.organization) request = self.factory.get('/api/siteprofile/?country__country=%s' % self.country.country) request.user = self.tola_user.user view = SiteProfileViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['name'], site1.name)
def test_filter_siteprofile_superuser(self): self.tola_user.user.is_staff = True self.tola_user.user.is_superuser = True self.tola_user.user.save() site1 = factories.SiteProfile(name='SiteProfile1', country=self.country) factories.SiteProfile(name='SiteProfile2', country=factories.Country()) request = self.factory.get('/api/siteprofile/?country__country=%s' % self.country.country) request.user = self.tola_user.user view = SiteProfileViewSet.as_view({'get': 'list'}) response = view(request) self.assertEqual(response.status_code, 200) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]['name'], site1.name)
def test_restore_having_historical_dependent_data(self): organization = factories.Organization(name='Some org') country = factories.Country(country='Brazil', code='BR') factories.SiteProfile(organization=organization, country=country) siteprofile_history_id = SiteProfile.history.get( organization=organization).id args = ['--demo'] opts = {} call_command('loadinitialdata', *args, **opts) args = ['--restore'] opts = {} call_command('loadinitialdata', *args, **opts) self.assertRaises(SiteProfile.history.model.DoesNotExist, SiteProfile.history.get, id=siteprofile_history_id)
def setUp(self): self.country = factories.Country() self.factory = APIRequestFactory() self.tola_user = factories.TolaUser()
def setUp(self): self.country = factories.Country(country='Brazil', code='BR') self.factory = APIRequestFactory() self.tola_user = factories.TolaUser()
def _create_countries(self): factories.Country( country="Afghanistan", code="AF", latitude="34.5333", longitude="69.1333", ) factories.Country( country="Pakistan", code="PK", latitude="33.6667", longitude="73.1667", ) factories.Country( country="Jordan", code="JO", latitude="31.9500", longitude="35.9333", ) factories.Country( country="Lebanon", code="LB", latitude="33.9000", longitude="35.5333", ) factories.Country( country="Ethiopia", code="ET", latitude="9.0167", longitude="38.7500", ) factories.Country( country="Timor-Leste", code="TL", latitude="-8.3", longitude="125.5667", ) factories.Country( country="Kenya", code="KE", latitude="-1.2833", longitude="36.8167", ) factories.Country( country="Iraq", code="IQ", latitude="33.3333", longitude="44.4333", ) factories.Country( country="Nepal", code="NP", latitude="26.5333", longitude="86.7333", ) factories.Country( country="Mali", code="ML", latitude="17.6500", longitude="0.0000", ) factories.Country( country="United States", code="US", latitude="45", longitude="-120", ) factories.Country( country="Turkey", code="TR", latitude="39.9167", longitude="32.8333", ) factories.Country( country="Syrian Arab Republic", code="SY", latitude="33.5000", longitude="36.3000", ) factories.Country( country="China", code="CN", ) factories.Country( country="India", code="IN", ) factories.Country( country="Indonesia", code="ID", ) factories.Country( country="Mongolia", code="MN", ) factories.Country( country="Myanmar", code="MY", latitude="21.9162", longitude="95.9560", ) factories.Country( country="Palestine", code="PS", latitude="31.3547", longitude="34.3088", ) factories.Country( country="South Sudan", code="SS", latitude="6.8770", longitude="31.3070", ) factories.Country( country="Uganda", code="UG", latitude="1.3733", longitude="32.2903", ) factories.Country( country="Germany", code="DE", latitude="51.1657", longitude="10.4515", )
def setUp(self): self.hikaya_user = factories.TolaUser() self.org = factories.Organization() self.country = factories.Country()
def setUp(self): self.org = factories.Organization() self.country = factories.Country()