def test_collect_and_store_output_in_a_different_location__directory(self): from infi.logs_collector.items import get_generic_os_items dst = mkdtemp() before = listdir(dst) logs_collector.run("test", get_generic_os_items(), datetime.now(), None, dst) after = listdir(dst) self.assertNotEqual(before, after)
def test_collect_and_store_output_in_a_different_location__filepath(self): from infi.logs_collector.items import get_generic_os_items dst = mkdtemp() before = listdir(dst) logs_collector.run("test", get_generic_os_items(), datetime.now(), None, path.join(dst, 'foo')) after = listdir(dst) self.assertNotEquals(before, after) self.assertEquals(after, ['foo'])
def test_collect_and_store_output_in_a_different_location__directory(self): from infi.logs_collector.items import get_generic_os_items dst = mkdtemp() before = listdir(dst) logs_collector.run("test", get_generic_os_items(), datetime.now(), None, dst) after = listdir(dst) self.assertNotEquals(before, after)
def test_windows_with_mocks(self): from infi.logs_collector.items import windows result, archive_path = logs_collector.run("test", windows(), None, None) self.assertTrue(path.exists(archive_path)) self.assertTrue(archive_path.endswith(".tar.gz")) archive = TarFile.open(archive_path, "r:gz")
def test_diretory_collector_timeout(self): def sleep(*args, **kwargs): from time import sleep sleep(5) with patch("re.match", new=sleep): items = [collectables.Directory("/tmp", "a", timeout_in_seconds=1)] result, archive_path = logs_collector.run("test", items, datetime.now(), None)
def test_windows_with_mocks(self): from infi.logs_collector.items import windows result, archive_path = logs_collector.run("test", windows(), datetime.now(), None) self.assertTrue(path.exists(archive_path)) self.assertTrue(archive_path.endswith(".tar.gz")) archive = TarFile.open(archive_path, "r:gz")
def test_run_collects_a_file_with_a_bad_st_size(self): fd, src = mkstemp() write(fd, b'\x00' * 4) close(fd) workaround_issue_10760 = logs_collector.workaround_issue_10760 def workaround_issue_10760_wrapper(*args, **kwargs): with patch("os.stat") as lstat: lstat.side_effect = fake_st_size_side_effect workaround_issue_10760(*args, **kwargs) items = [collectables.File(src)] with patch.object(logs_collector, "workaround_issue_10760", new=workaround_issue_10760_wrapper): result, archive_path = logs_collector.run("test", items, datetime.now(), None)
def test_run_collects_a_file_with_a_bad_st_size(self): fd, src = mkstemp() write(fd, '\x00' * 4) close(fd) workaround_issue_10760 = logs_collector.workaround_issue_10760 def workaround_issue_10760_wrapper(*args, **kwargs): with patch("os.stat") as lstat: lstat.side_effect = fake_st_size_side_effect workaround_issue_10760(*args, **kwargs) items = [collectables.File(src)] with patch.object(logs_collector, "workaround_issue_10760", new=workaround_issue_10760_wrapper): result, archive_path = logs_collector.run("test", items, datetime.now(), None)
def test_logging_handlers(self): import logging before = list(logging.root.handlers) result, archive_path = logs_collector.run("test", [], datetime.now(), None) self.assertEqual(before, logging.root.handlers)
def test_run_hits_OSError(self): with patch("os.path.islink") as islink: islink.side_effect = OSError() result, archive_path = logs_collector.run("test", [], datetime.now(), None)
def _test_real(self, items): result, archive_path = logs_collector.run("test", items, datetime.now(), None) self.assertTrue(path.exists(archive_path)) self.assertTrue(archive_path.endswith(".tar.gz")) archive = TarFile.open(archive_path, "r:gz")
def test_command_timeout(self): items = [collectables.Command("sleep", ["5"], wait_time_in_seconds=1)] result, archive_path = logs_collector.run("test", items, datetime.now(), None)
def test_logging_handlers(self): import logging before = list(logging.root.handlers) result, archive_path = logs_collector.run("test", [], datetime.now(), None) self.assertEquals(before, logging.root.handlers)