def test_smoke_read(self): stream = io.StringIO(self._TEXT) reader = IOLogRecordReader(stream) record1 = reader.read_record() self.assertEqual(record1, self._RECORD) record2 = reader.read_record() self.assertEqual(record2, None) reader.close() with self.assertRaises(ValueError): stream.getvalue()
def test_iter_read(self): stream = io.StringIO(self._TEXT) reader = IOLogRecordReader(stream) record_list = list(reader) self.assertEqual(record_list, [self._RECORD])