def test_get_row(self): with self.drtb_import(IMPORT_ROWS, "mumbai") as (csv_file, csv_rows): output = DRTBImportHistoryCommand.handle_get_row("some bad id", csv_file) csv_file.seek(0) self.assertEqual(output, "case not found\n") row_case_ids = csv_rows[0]['case_ids'].split(",") for case_id in row_case_ids: output = DRTBImportHistoryCommand.handle_get_row(case_id, csv_file) csv_file.seek(0) self.assertEqual(output, "row: 2\n")
def test_get_outcome(self): with self.drtb_import(IMPORT_ROWS, "mumbai") as (csv_file, csv_rows): output = DRTBImportHistoryCommand.handle_get_outcome("2", csv_file) csv_file.seek(0) # Confirm that the outcome contains all the expected case ids self.assertNotIn( "Traceback", output, "Expected list of case ids, but output was: {}".format(output)) self.assertEqual(len(output.split()), 5 + len(DRUG_MAP)) output = DRTBImportHistoryCommand.handle_get_outcome("3", csv_file) csv_file.seek(0) # Confirm that the outcome contains the exception raised for the row self.assertTrue("person_name is required" in output)