def test_should_record_consumption_for_each_facility(self):
report = FakeReport()
report.ads = defaultdict(list)
report.pds = defaultdict(list)
report.cycle = "May - Jun 2015"
location = Location.migrate_from_dict(
{
"name": "location_one",
"IP": "ip_one",
"District": "district_one",
"Warehouse": "warehouse_one",
"Multiple": "",
"status": "",
"Web/Paper": "",
"scores": defaultdict(dict),
}
)
report.locs = [location]
report.cs = {
location: [
ConsumptionRecord.migrate_from_dict(
{"opening_balance": 20, "closing_balance": 12}
)
]
}
self.assertEqual(Consumption.objects.count(), 0)
persist_consumption(report)
self.assertEqual(Consumption.objects.count(), 1)
first_record = Consumption.objects.all()[0]
self.assertEqual(first_record.name, "location_one")
self.assertEqual(first_record.ip, "ip_one")
self.assertEqual(first_record.district, "district_one")
self.assertEqual(first_record.warehouse, "warehouse_one")
self.assertEqual(first_record.opening_balance, 20)
self.assertEqual(first_record.closing_balance, 12)
def test_should_record_paed_records_for_each_facility(self):
report = FakeReport()
report.ads = defaultdict(list)
report.pds = defaultdict(list)
report.cycle = "May - Jun 2015"
location_one = Location.migrate_from_dict(
{
"name": "location_one",
"IP": "ip_one",
"District": "district_one",
"Warehouse": "warehouse_one",
"Multiple": "",
"status": "",
"Web/Paper": "",
"scores": defaultdict(dict),
}
)
report.locs = [location_one]
report.pds = {
location_one: [PatientRecord.migrate_from_dict({"new": 20, "existing": 12})]
}
self.assertEqual(PAEDPatientsRecord.objects.count(), 0)
persist_paed_records(report)
self.assertEqual(PAEDPatientsRecord.objects.count(), 1)
first_record = PAEDPatientsRecord.objects.all()[0]
self.assertEqual(first_record.name, "location_one")
self.assertEqual(first_record.ip, "ip_one")
self.assertEqual(first_record.district, "district_one")
self.assertEqual(first_record.warehouse, "warehouse_one")
self.assertEqual(first_record.new, 20)
self.assertEqual(first_record.existing, 12)
def test_should_record_scores_for_each_facility(self):
report = FakeReport()
report.cycle = "May - Jun 2015"
location_one = Location.migrate_from_dict(
{
"name": "location_one",
"IP": "ip_one",
"District": "district_one",
"Warehouse": "warehouse_one",
}
)
location_two = Location.migrate_from_dict(
{
"name": "location_two",
"District": "district_one",
"IP": "ip_one",
"Warehouse": "warehouse_one",
}
)
report.locs = [location_one, location_two]
scores_cache = {
location_one: {
"WEB_BASED": {"DEFAULT": "YES"}, "REPORTING": {"DEFAULT": "NO"}
},
location_two: {
"WEB_BASED": {"DEFAULT": "YES"}, "REPORTING": {"DEFAULT": "NO"}
},
}
self.assertEqual(Score.objects.count(), 0)
persist_scores(scores_cache, report.cycle)
persist_scores(scores_cache, report.cycle)
self.assertEqual(Score.objects.count(), 2)
first_score = Score.objects.all()[0]
self.assertEqual(first_score.default_pass_count, 1)
self.assertEqual(first_score.default_fail_count, 1)
def test_dhis2_facility_as_location(self):
data = {
"level":
5,
"name":
"01 Commando HC II",
"id":
"Ugl9pT2K5B0",
"ancestors": [
{
"name": "MOH - Uganda",
"level": 1
},
{
"name": "Northern Region",
"level": 2
},
{
"name": "Otuke District",
"level": 3
},
{
"name": "Okwang Subcounty",
"level": 4
},
],
}
loc = Location.migrate_from_dict({
"name": "01 Commando HC II",
"District": "Otuke District"
})
location = dhis2_facility_as_location({"01 Commando HC II": "PT"},
{loc: loc}, {})(data)
self.assertEqual(location.facility, "01 Commando HC II")
self.assertEqual(location.district, "Otuke District")
self.assertEqual(location.status, "Reporting")
self.assertEqual(location.partner, "PT")