def test_close_file(self):
with open('log.log', 'wb') as f:
f.write('Some file contents!')
reader = LogReader(0, 'log.log', Log4jParser(), FakeReceiver())
reader._open_file()
f = reader.logfile
reader._close_file()
self.assertTrue(f.closed)
self.assertEqual(reader.logfile, None)
def test_open_file_with_regular_file(self):
with open('log.log', 'wb') as f:
f.write('Some file contents!')
reader = LogReader(0, 'log.log', Log4jParser(), FakeReceiver())
reader._open_file()
try:
self.assertEqual(reader.logfile.name, 'log.log')
self.assertFalse(reader.logfile.closed)
self.assertNotEqual(reader.logfile_id, None)
self.assertEqual(reader.logfile.read(), 'Some file contents!')
finally:
reader.logfile.close()
def test_open_file_with_gzip_file(self):
self.files_to_delete.append('log.gz')
with gzip.open('log.gz', 'wb') as f:
f.write('Some file contents!')
reader = LogReader(0, 'log.gz', Log4jParser(), FakeReceiver())
reader._open_file()
try:
self.assertEqual(reader.logfile.name, 'log.gz')
self.assertFalse(reader.logfile.closed)
self.assertNotEqual(reader.logfile_id, None)
self.assertEqual(reader.logfile.read(), 'Some file contents!')
finally:
reader.logfile.close()
