def test_exclude_subrecords( self, write_data_dictionary, write_to_file, csv_renderer, subrecords ): subrecords.return_value = [Colour] extract.generate_csv_files( "somewhere", models.Episode.objects.all(), self.user ) self.assertEqual(csv_renderer.call_count, 0)
def test_exclude_subrecords( self, write_data_dictionary, write_to_file, csv_renderer, subrecords ): subrecords.return_value = [Colour] extract.generate_csv_files( "somewhere", models.Episode.objects.all(), self.user ) self.assertEqual(csv_renderer.call_count, 0)
def test_generate_csv_files(self, write_data_dictionary, write_to_file, subrecords): patient, episode = self.new_patient_and_episode_please() subrecords.return_value = [HatWearer, HouseOwner] HatWearer.objects.create(name="Indiana", episode=episode) HouseOwner.objects.create(patient=patient) results = extract.generate_csv_files("somewhere", models.Episode.objects.all(), self.user) expected = [('somewhere/data_dictionary.html', 'data_dictionary.html'), ('somewhere/episodes.csv', 'episodes.csv'), ('somewhere/hat_wearer.csv', 'hat_wearer.csv'), ('somewhere/house_owner.csv', 'house_owner.csv')] self.assertEqual(expected, results) self.assertEqual(write_data_dictionary.call_args[0][0], 'somewhere/data_dictionary.html') self.assertEqual(write_to_file.call_args[0], ('somewhere/house_owner.csv', ))
def test_generate_csv_files( self, write_data_dictionary, write_to_file, subrecords ): patient, episode = self.new_patient_and_episode_please() subrecords.return_value = [HatWearer, HouseOwner] HatWearer.objects.create(name="Indiana", episode=episode) HouseOwner.objects.create(patient=patient) results = extract.generate_csv_files( "somewhere", models.Episode.objects.all(), self.user ) expected = [ ('somewhere/data_dictionary.html', 'data_dictionary.html'), ('somewhere/episodes.csv', 'episodes.csv'), ('somewhere/hat_wearer.csv', 'hat_wearer.csv'), ('somewhere/house_owner.csv', 'house_owner.csv') ] self.assertEqual(expected, results) self.assertEqual( write_data_dictionary.call_args[0][0], 'somewhere/data_dictionary.html' ) self.assertEqual( write_to_file.call_args[0], ('somewhere/house_owner.csv',) )