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)