def test__cannot_connect(self, module_patch, import_patch): mock_module = self.setup_module(module_patch) self.setup_session_root(import_patch) self.mock_session.start.side_effect = Exception("cannot connect") main() import_patch.assert_called_once_with("vspk.v5_0") mock_module.fail_json.assert_called_once_with( msg="Could not establish connection to VSD cannot connect")
def test__cannot_import(self, module_patch, import_patch): mock_module = self.setup_module(module_patch) import_patch.side_effect = ImportError("cannot import") main() import_patch.assert_called_once_with("vspk.v5_0") mock_module.fail_json.assert_called_once_with( msg="vspk is required for this module, or " "API version specified does not exist.")
def test_get__exception(self, module_patch, import_patch): params = dict(TEST_PARAMS) params['get_gateway_purge_time'] = True mock_module = self.setup_module(module_patch, params) mock_root = self.setup_session_root(import_patch) mock_root.system_configs = MagicMock() mock_root.system_configs.get_first.side_effect = Exception("Test") main() self.validate_session(import_patch) mock_module.fail_json.assert_called_once_with( msg="Could not retrieve gateway purge timer : Test")
def test_get__success(self, module_patch, import_patch): params = dict(TEST_PARAMS) params['get_gateway_purge_time'] = True mock_module = self.setup_module(module_patch, params) mock_root = self.setup_session_root(import_patch) mock_sys_config = self.setup_sys_config(mock_root) mock_sys_config.ad_gateway_purge_time = 12345 main() self.validate_session(import_patch) mock_module.fail_json.assert_not_called() mock_module.exit_json.assert_called_once_with(changed=True, result=12345)
def test_set__update_error(self, module_patch, import_patch): mock_module = self.setup_module(module_patch) mock_root = self.setup_session_root(import_patch) mock_sys_config = self.setup_sys_config(mock_root) mock_sys_config.save.side_effect = Exception("Test") main() gw_purge = TEST_PARAMS['gateway_purge_time'] self.validate_session(import_patch) assert mock_sys_config.ad_gateway_purge_time == gw_purge mock_sys_config.save.assert_called_once_with() mock_module.fail_json.assert_called_once_with( msg="Could not update gateway purge timer : Test")
def test_set__success(self, module_patch, import_patch): mock_module = self.setup_module(module_patch) mock_root = self.setup_session_root(import_patch) mock_sys_config = self.setup_sys_config(mock_root) main() gw_purge = TEST_PARAMS['gateway_purge_time'] self.validate_session(import_patch) assert mock_sys_config.ad_gateway_purge_time == gw_purge mock_sys_config.save.assert_called_once_with() mock_module.fail_json.assert_not_called() mock_module.exit_json.assert_called_once_with( changed=True, result="Gateway purge time set to %ssec" % gw_purge)
def test_set__already_set(self, module_patch, import_patch): mock_module = self.setup_module(module_patch) mock_root = self.setup_session_root(import_patch) mock_sys_config = self.setup_sys_config(mock_root) mock_sys_config.save.side_effect = self.get_mock_bambou_error( 400, "There are no attribute changes") main() gw_purge = TEST_PARAMS['gateway_purge_time'] self.validate_session(import_patch) assert mock_sys_config.ad_gateway_purge_time == gw_purge mock_sys_config.save.assert_called_once_with() mock_module.fail_json.assert_not_called() mock_module.exit_json.assert_called_once_with( changed=True, result="Gateway time is already updated to %s" % gw_purge)