def test_finalize(self):
header = 'one,two,three\r\n'
recorder = CSVRecorder(self.temp)
# all calls do nothing
recorder.prepare(DummyRecord)
for x in range(10):
recorder.finalize()
self.assertMultiLineEqual(self.temp.getvalue(), header)
def test_exception_when_no_prepare(self):
values = [(5, 'pikos', 'apikos')]
recorder = CSVRecorder(self.temp)
with self.assertRaises(RecorderError):
recorder.record(values)
with self.assertRaises(RecorderError):
recorder.finalize()
def test_finalize(self):
fields = ('one', 'two', 'three')
header = 'one,two,three\r\n'
recorder = CSVRecorder(self.temp)
# all calls do nothing
recorder.prepare(fields)
for x in range(10):
recorder.finalize()
self.assertMultiLineEqual(self.temp.getvalue(), header)
def test_exception_when_no_prepare(self):
values = [(5, 'pikos', 'apikos')]
recorder = CSVRecorder(self.temp)
with self.assertRaises(RecorderError):
recorder.record(values)
with self.assertRaises(RecorderError):
recorder.finalize()
def test_exceptions(self):
fields = ('one', 'two','three')
values = [(5, 'pikos','apikos')]
recorder = CSVRecorder(self.temp)
with self.assertRaises(RecorderError):
recorder.record(values)
with self.assertRaises(RecorderError):
recorder.finalize()
