def test_finalize(self):
header = 'one,two,three\n'
recorder = CSVFileRecorder(filename=self.filename)
# all calls do nothing
with self.finalizer(recorder):
recorder.prepare(DummyRecord)
for x in range(10):
recorder.finalize()
self.assertRecordedLines(header)
def test_exception_when_no_prepare(self):
records = [DummyRecord(5, 'pikos', 'apikos')]
recorder = CSVFileRecorder(filename=self.filename)
with self.assertRaises(RecorderError):
recorder.record(records)
with self.assertRaises(RecorderError):
recorder.finalize()
