def test__cannot_connect(self, module_patch, import_patch): mock_module = 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", rc=1)
def test__exception(self, module_patch, import_patch): mock_module = setup_module(module_patch) mock_root = self.setup_session_root(import_patch) self.setup_sample_domains(mock_root) self.l3_domains[1].save.side_effect = Exception("cannot save") main() import_patch.assert_called_once_with("vspk.v5_0") mock_module.fail_json.assert_called_once_with( msg="Could not set maintainance mode : cannot save", rc=1)
def test__cannot_import(self, module_patch, import_patch): mock_module = 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.', rc=1)
def test__no_changes(self, module_patch, import_patch): mock_module = setup_module(module_patch) mock_root = self.setup_session_root(import_patch) self.setup_sample_domains(mock_root) self.l3_domains[1].save.side_effect = self.get_mock_bambou_error( 400, "There are no attribute changes") main() import_patch.assert_called_once_with("vspk.v5_0") mock_module.fail_json.assert_not_called() mock_module.exit_json.assert_called_once_with( rc=0, changed=True, result="Maintainance mode is already enabled")
def test__bambou_error(self, module_patch, import_patch): mock_module = setup_module(module_patch) mock_root = self.setup_session_root(import_patch) self.setup_sample_domains(mock_root) self.l3_domains[1].save.side_effect = self.get_mock_bambou_error( 400, "save error") main() import_patch.assert_called_once_with("vspk.v5_0") mock_module.fail_json.assert_called_once_with( msg="Could not set maintainance mode : " "[HTTP 400(save error)] save error", rc=1)
def test_enable__success(self, module_patch, import_patch): mock_module = setup_module(module_patch) mock_root = self.setup_session_root(import_patch) self.setup_sample_domains(mock_root) main() self.validate_session(import_patch) self.validate_sample_domains(mock_root, "ENABLED") mock_module.fail_json.assert_not_called() mock_module.exit_json.assert_called_once_with( rc=0, changed=True, result="Maintainance mode for all non shared L3 domains-enabled, " "Maintainance mode for all non shared L2 domains-enabled,")
def test_enable__no_domains(self, module_patch, import_patch): mock_module = setup_module(module_patch) mock_root = self.setup_session_root(import_patch) self.setup_sample_domains(mock_root) mock_root.domains.get.return_value = list() mock_root.l2_domains.get.return_value = list() main() self.validate_session(import_patch) mock_module.fail_json.assert_not_called() mock_module.exit_json.assert_called_once_with( rc=0, changed=True, result="No L3 domains found to enabled maintainance mode, " "No L2 domains found to enabled maintainance mode")
def test_disable__success(self, module_patch, import_patch): params = dict(TEST_PARAMS) params["state"] = "disabled" mock_module = setup_module(module_patch, params) mock_root = self.setup_session_root(import_patch) self.setup_sample_domains(mock_root) main() self.validate_session(import_patch) self.validate_sample_domains(mock_root, "DISABLED") mock_module.fail_json.assert_not_called() mock_module.exit_json.assert_called_once_with( rc=0, changed=True, result="Maintainance mode for all non shared L3 domains-disabled, " "Maintainance mode for all non shared L2 domains-disabled,")