def test_init_host(self, mock_DriverManager):
mock_driver = mock.Mock()
mock_DriverManager.return_value = mock_driver
host_manager = host.HostmonitorManager()
host_manager.init_host()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
def test_init_host_exception(self, mock_DriverManager, mock_exit):
mock_DriverManager.side_effect = Exception("Test exception.")
mock_exit.return_value = None
host_manager = host.HostmonitorManager()
host_manager.init_host()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
mock_exit.assert_called_once_with(1)
def test_main_exception(self, mock_DriverManager):
mock_driver = mock.Mock()
mock_DriverManager.return_value = mock_driver
mock_driver.driver.monitor_hosts.side_effect = \
Exception("Test exception.")
host_manager = host.HostmonitorManager()
host_manager.init_host()
ret = host_manager.main()
mock_DriverManager.assert_called_once_with(
namespace='hostmonitor.driver',
name='default',
invoke_on_load=True,
invoke_args=(),
)
mock_driver.driver.monitor_hosts.assert_called_once_with()
self.assertIsNone(ret)