def test_location_created_after_facility_processing(self): start_date = datetime(2015, 9, 1) end_date = datetime.utcnow() with mock.patch('custom.ilsgateway.tanzania.warehouse.updater.default_start_date', return_value=start_date), \ mock.patch('custom.ilsgateway.tasks.default_start_date', return_value=start_date), \ mock.patch('custom.ilsgateway.tasks.get_start_date', return_value=start_date): run = ReportRun.objects.create(start=start_date, end=end_date, start_run=datetime.utcnow(), domain=self.domain.name) process_facility_warehouse_data(self.facility1, start_date, end_date, run) facility = make_loc(code='new_facility', name='New Facility', domain=self.domain.name, type='FACILITY', parent=self.district1, metadata={'group': 'A'}) process_non_facility_warehouse_data(self.district1, start_date, end_date, run) self.assertEqual( OrganizationSummary.objects.filter( date__lte=datetime(2015, 9, 3)).count(), 2) facility.delete()
def recalculate_on_group_change(location, last_run): OrganizationSummary.objects.filter(location_id=location.get_id).delete() process_facility_warehouse_data(location, default_start_date(), last_run.end) for parent in location.sql_location.get_ancestors(ascending=True): process_non_facility_warehouse_data(parent.couch_location, default_start_date(), last_run.end, strict=False)
def recalculate_on_group_change(location, last_run): OrganizationSummary.objects.filter(location_id=location.get_id).delete() process_facility_warehouse_data(location, default_start_date(), last_run.end) return { ancestor.location_type.name: {ancestor} for ancestor in location.sql_location.get_ancestors(ascending=True) }
def test_location_created_after_facility_processing(self): start_date = datetime(2015, 9, 1) end_date = datetime.utcnow() with mock.patch('custom.ilsgateway.tanzania.warehouse.updater.default_start_date', return_value=start_date), \ mock.patch('custom.ilsgateway.tasks.default_start_date', return_value=start_date), \ mock.patch('custom.ilsgateway.tasks.get_start_date', return_value=start_date): run = ReportRun.objects.create(start=start_date, end=end_date, start_run=datetime.utcnow(), domain=self.domain.name) process_facility_warehouse_data(self.facility1, start_date, end_date, run) facility = make_loc(code='new_facility', name='New Facility', domain=self.domain.name, type='FACILITY', parent=self.district1, metadata={'group': 'A'}) process_non_facility_warehouse_data(self.district1, start_date, end_date, run) self.assertEqual(OrganizationSummary.objects.filter(date__lte=datetime(2015, 9, 3)).count(), 2) facility.delete()