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