def test_export_with_talks_csv_file(self):
dev_data = DevData()
dev_data.SPEAKERS_PER_EVENT = 2
events = [Event.objects.current_event()]
dev_data.create_users_and_attendees(10, events=events)
dev_data.create_speakers(events)
dev_data.create_talk_formats()
dev_data.create_talks(events=events)
talks = Event.objects.current_event().talk_set
outbuf = io.StringIO()
errbuf = io.StringIO()
tempfile = NamedTemporaryFile(mode="w", delete=False)
try:
call_command("export_talks_for_committee",
csv_file=tempfile,
stdout=outbuf,
stderr=errbuf)
tempfile.close()
self.assertEqual("", outbuf.getvalue().strip())
self.assertEqual("", errbuf.getvalue().strip())
with open(tempfile.name, "r") as output:
output_lines = output.read().splitlines()
self.assertEqual(talks.count() + 1, len(output_lines))
finally:
os.unlink(tempfile.name)
def test_export_with_talks(self):
dev_data = DevData()
dev_data.SPEAKERS_PER_EVENT = 2
events = [Event.objects.current_event()]
dev_data.create_users_and_attendees(10, events=events)
dev_data.create_speakers(events)
dev_data.create_talk_formats()
dev_data.create_talks(events=events)
talks = Event.objects.current_event().talk_set
outbuf = io.StringIO()
errbuf = io.StringIO()
call_command("export_talks_for_committee",
stdout=outbuf,
stderr=errbuf)
output_lines = outbuf.getvalue().splitlines()
self.assertEqual(talks.count() + 1, len(output_lines))
self.assertEqual("", errbuf.getvalue().strip())