You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.6 KiB
48 lines
1.6 KiB
7 months ago
|
"""Tests for base_sysinfo."""
|
||
|
|
|
||
|
|
import mock
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import common
|
||
|
|
from autotest_lib.client.common_lib import autotemp
|
||
|
|
from autotest_lib.client.bin import base_sysinfo
|
||
|
|
|
||
|
|
|
||
|
|
class LoggableTestException(Exception):
|
||
|
|
"""An exception thrown by the loggable used for testing."""
|
||
|
|
|
||
|
|
|
||
|
|
class BaseSysinfoTestCase(unittest.TestCase):
|
||
|
|
"""TestCase for free functions in the base_sysinfo module."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
self._output_dir = autotemp.tempdir()
|
||
|
|
self.addCleanup(self._output_dir.clean)
|
||
|
|
|
||
|
|
def test_run_loggables_with_no_exception(self):
|
||
|
|
"""Tests _run_loggables_ignoring_errors when no loggable throws"""
|
||
|
|
loggables = {
|
||
|
|
mock.create_autospec(base_sysinfo.loggable),
|
||
|
|
mock.create_autospec(base_sysinfo.loggable),
|
||
|
|
}
|
||
|
|
base_sysinfo._run_loggables_ignoring_errors(loggables, self._output_dir)
|
||
|
|
for log in loggables:
|
||
|
|
log.run.assert_called_once_with(self._output_dir)
|
||
|
|
|
||
|
|
def test_run_loggables_with_exception(self):
|
||
|
|
"""Tests _run_loggables_ignoring_errors when one loggable throws"""
|
||
|
|
failing_loggable = mock.create_autospec(base_sysinfo.loggable)
|
||
|
|
failing_loggable.run.side_effect = LoggableTestException
|
||
|
|
loggables = {
|
||
|
|
mock.create_autospec(base_sysinfo.loggable),
|
||
|
|
failing_loggable,
|
||
|
|
mock.create_autospec(base_sysinfo.loggable),
|
||
|
|
}
|
||
|
|
base_sysinfo._run_loggables_ignoring_errors(loggables, self._output_dir)
|
||
|
|
for log in loggables:
|
||
|
|
log.run.assert_called_once_with(self._output_dir)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|