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)
