def test_03_apps_by_country(self): apps = ApplicationFixtureFactory.make_application_spread(APPLICATION_YEAR_OUTPUT, "year") for a in apps: a.save() time.sleep(2) outfiles = reporting.content_reports("1970-01-01T00:00:00Z", dates.now(), TMP_DIR) assert len(outfiles) == 1 assert os.path.exists(outfiles[0]) table = [] with codecs.open(outfiles[0], "rb", "utf-8") as f: reader = clcsv.UnicodeReader(f) for row in reader: table.append(row) expected = self._as_output(APPLICATION_YEAR_OUTPUT) assert table == expected
def test_04_background(self): provs = ProvenanceFixtureFactory.make_action_spread(MONTH_EDIT_OUTPUT, "edit", "month") for p in provs: p.save() apps = ApplicationFixtureFactory.make_application_spread(APPLICATION_YEAR_OUTPUT, "year") for a in apps: a.save() time.sleep(2) job = reporting.ReportingBackgroundTask.prepare("system", outdir=TMP_DIR, from_date="1970-01-01T00:00:00Z", to_date=dates.now()) reporting.ReportingBackgroundTask.submit(job) time.sleep(2) job = models.BackgroundJob.pull(job.id) prov_outfiles = job.reference["reporting__provenance_outfiles"] cont_outfiles = job.reference["reporting__content_outfiles"] assert len(prov_outfiles) == 4 assert len(cont_outfiles) == 1