def test_init(self):
renderer = extract.EpisodeCsvRenderer(
models.Episode,
models.Episode.objects.all(),
self.user
)
self.assertEqual(renderer.model, models.Episode)
def test_get_row(self):
renderer = extract.EpisodeCsvRenderer(models.Episode,
models.Episode.objects.all(),
self.user)
self.episode.set_tag_names(["trees"], self.user)
# make sure we keep historic tags
self.episode.set_tag_names(["leaves"], self.user)
self.assertIn("trees;leaves", renderer.get_row(self.episode))
def test_headers(self):
expected = {
"Start", "End", "Created", "Updated", "Created By", "Updated By",
"Patient", "Tagging"
}
renderer = extract.EpisodeCsvRenderer(models.Episode,
models.Episode.objects.all(),
self.user)
self.assertEqual(len(expected - set(renderer.get_headers())), 0)
