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_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_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_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_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)
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)
