def test_check_system_bad_state_do_not_quit(
self,
kernel_status,
docker_version_status,
m_check_etcd_version,
m_check_docker_version,
m_check_kernel_modules,
m_enforce_root,
):
"""
Test for check_system when one of the system checks fails
This test does not quit if there is an error -
Assert that the function returns False
:param kernel_status: return_value for _check_modules
:param docker_version_status: return_value for _check_docker_version
"""
# Set up mock objects
m_check_kernel_modules.return_value = kernel_status
m_check_docker_version.return_value = docker_version_status
# Call method under test without exiting if error detected
test_return = check_system(quit_if_error=False)
# Assert
self.assertIn(False, test_return)
def test_check_system_bad_state_do_not_quit(
self, kernel_status, ip_forwarding_status, docker_version_status):
"""
Test for check_system when one of the system checks fails
This test does not quit if there is an error -
Assert that the function returns False
:param kernel_status: return_value for _check_kernel_modules
:param ip_forwarding_status: return_value for _check_ip_forwarding
:param docker_version_status: return_value for _check_docker_version
"""
with patch('calico_ctl.checksystem.enforce_root', autospec=True) \
as m_enforce_root, \
patch('calico_ctl.checksystem._check_kernel_modules', autospec=True) \
as m_check_kernel_modules, \
patch('calico_ctl.checksystem._check_ip_forwarding', autospec=True) \
as m_check_ip_forwarding, \
patch('calico_ctl.checksystem._check_docker_version', autospec=True) \
as m_check_docker_version:
# Set up mock objects
m_check_kernel_modules.return_value = kernel_status
m_check_ip_forwarding.return_value = ip_forwarding_status
m_check_docker_version.return_value = docker_version_status
# Call method under test without exiting if error detected
test_return = check_system(fix=False, quit_if_error=False)
# Assert
m_enforce_root.assert_called_once_with()
self.assertFalse(test_return)
def test_check_system(self, m_check_docker_version,
m_check_kernel_modules, m_enforce_root):
"""
Test for check_system when all checks pass
Assert that the function returns True
"""
# Call method under test
test_return = check_system(quit_if_error=True)
# Assert
m_enforce_root.assert_called_once_with()
m_check_kernel_modules.assert_called_once_with()
m_check_docker_version.assert_called_once_with(False)
self.assertTrue(test_return)
def test_check_system(self, m_check_docker_version, m_check_kernel_modules,
m_enforce_root):
"""
Test for check_system when all checks pass
Assert that the function returns True
"""
# Call method under test
test_return = check_system(quit_if_error=True)
# Assert
m_enforce_root.assert_called_once_with()
m_check_kernel_modules.assert_called_once_with()
m_check_docker_version.assert_called_once_with(False)
self.assertTrue(test_return)
def test_check_system_bad_state_do_not_quit(
self, kernel_status, docker_version_status, m_check_etcd_version,
m_check_docker_version, m_check_kernel_modules, m_enforce_root):
"""
Test for check_system when one of the system checks fails
This test does not quit if there is an error -
Assert that the function returns False
:param kernel_status: return_value for _check_modules
:param docker_version_status: return_value for _check_docker_version
"""
# Set up mock objects
m_check_kernel_modules.return_value = kernel_status
m_check_docker_version.return_value = docker_version_status
# Call method under test without exiting if error detected
test_return = check_system(quit_if_error=False)
# Assert
self.assertIn(False, test_return)
