def test_get_header(self, field_names_to_extract): field_names_to_extract.return_value = [ "patient_id", "name", "consistency_token", "id" ] renderer = extract.PatientSubrecordCsvRenderer( PatientColour, models.Episode.objects.all(), self.user) self.assertEqual(["Episode", "Patient", "Name"], renderer.get_headers())
def test_get_rows(self, field_names_to_extract): field_names_to_extract.return_value = [ "patient_id", "name", "consistency_token", "id" ] renderer = extract.PatientSubrecordCsvRenderer( PatientColour, models.Episode.objects.all(), self.user) rendered = list(renderer.get_rows()) self.assertEqual([["1", "1", "blue"]], rendered)
def test_get_rows_same_patient(self, field_names_to_extract): self.patient.create_episode() first_episode = self.patient.episode_set.first() last_episode = self.patient.episode_set.last() field_names_to_extract.return_value = [ "patient_id", "name", "consistency_token", "id" ] renderer = extract.PatientSubrecordCsvRenderer( PatientColour, models.Episode.objects.all(), self.user) rendered = list(renderer.get_rows()) self.assertEqual( [[str(first_episode.id), str(self.patient.id), "blue"], [str(last_episode.id), str(self.patient.id), "blue"]], rendered)