blob: cc6a8365755638f81eb6c82425d17690296a43c2 (
plain) (
tree)
|
|
# This file is part of the sos project: https://github.com/sosreport/sos
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# version 2 of the GNU General Public License.
#
# See the LICENSE file in the source distribution for further information.
import random
import os
from sos_tests import StageOneReportTest, StageTwoReportTest
from string import ascii_uppercase, digits
from systemd import journal
class LogsPluginTest(StageOneReportTest):
"""Ensure common collections from the `logs` plugin are properly collected
:avocado: tags=stageone
"""
sos_cmd = '-o logs --all-logs'
def test_journalctl_collections(self):
self.assertFileCollected('sos_commands/logs/journalctl_--disk-usage')
self.assertFileCollected('sos_commands/logs/journalctl_--no-pager_--catalog_--boot')
def test_journal_runtime_collected(self):
self.assertFileGlobInArchive('/var/log/journal/*')
class LogsSizeLimitTest(StageTwoReportTest):
"""Test that journal size limiting is working and is independent of
--log-size
Note: this test will insert over 100MB of garbage into the test system's
journal
:avocado: tags=stagetwo
"""
sos_cmd = '-o logs'
packages = {
'rhel': 'python3-systemd',
'ubuntu': 'python3-systemd'
}
# override the stage2 mocking here to inject a string into the journal
def setup_mocking(self):
# write 20MB at a time to side-step rate/size limiting on some distros
# write over 100MB to ensure we will actually size limit inside sos,
# allowing for any compression or de-dupe systemd does
rsize = 20 * 1048576
for i in range(3):
# generate 20MB, write it, then write it in reverse.
# Spend less time generating new strings
rand = ''.join(random.choice(ascii_uppercase + digits) for _ in range(rsize))
journal.send(rand)
journal.send(rand[::-1])
def test_journal_size_limit(self):
journ = 'sos_commands/logs/journalctl_--no-pager_--catalog_--boot'
self.assertFileCollected(journ)
jsize = os.stat(self.get_name_in_archive(journ)).st_size
assert jsize <= 105906176, "Collected journal is larger than 100MB"
assert jsize > 27262976, "Collected journal limited by --log-size"
def test_journal_tailed_and_linked(self):
self.assertFileCollected('sos_strings/logs/journalctl_--no-pager_--catalog_--boot.tailed')
journ = self.get_name_in_archive('sos_commands/logs/journalctl_--no-pager_--catalog_--boot')
assert os.path.islink(journ), "Journal in sos_commands/logs is not a symlink"
|
