def test_save_progress_success(self):
self.files_to_delete.append('progressf16c93d1167446f99a26837c0fdeac6fb73869794')
reader = LogReader(0, 'log.log', Log4jParser(), FakeReceiver(), progress_file_path_prefix='progress')
reader._make_progress_string = lambda: 'log.log 123g456 10 19'
reader._save_progress()
with open('progressf16c93d1167446f99a26837c0fdeac6fb73869794', 'rb') as f:
self.assertEqual(f.read(), 'log.log 123g456 10 19')
def test_maybe_do_housekeeping_first_time(self):
called = []
reader = LogReader(0, 'log.log', Log4jParser(), FakeReceiver(), progress_file_path_prefix='progress')
reader._ensure_file_is_good = lambda: called.append('_ensure_file_is_good')
reader._save_progress = lambda: called.append('_save_progress')
reader._adjust_loglevel_suppression = lambda: called.append('_adjust_loglevel_suppression')
reader._maybe_do_housekeeping(23)
self.assertEqual(called, ['_ensure_file_is_good', '_save_progress', '_adjust_loglevel_suppression'])
self.assertEqual(reader.last_ensure_file_is_good_call_timestamp, 23)
self.assertEqual(reader.last_save_progress_call_timestamp, 23)
self.assertEqual(reader.last_adjust_loglevel_suppression_call_timestamp, 23)
def test_save_progress_failure(self):
reader = LogReader(0, 'log.log', Log4jParser(), FakeReceiver(), progress_file_path_prefix='invalid\0path')
reader._make_progress_string = lambda: 'log.log 123g456 10 19'
reader._save_progress()
self.assertFalse(os.path.exists('progressf16c93d1167446f99a26837c0fdeac6fb73869794'))
self.assertEqual(self.fake_logging.log[-1], '[ERROR] Failed to save progress for log.log.')
def test_save_progress_no_data(self):
reader = LogReader(0, 'log.log', Log4jParser(), FakeReceiver(), progress_file_path_prefix='progress')
reader._make_progress_string = lambda: None
reader._save_progress()
self.assertFalse(os.path.exists('progressf16c93d1167446f99a26837c0fdeac6fb73869794'))
