def test_highstate(self): ''' Test to retrieve the state data from the salt master for the minion and execute it ''' arg = "whitelist=sls1.sls" mock = MagicMock(side_effect=[True, False, False, False]) with patch.object(state, '_disabled', mock): self.assertDictEqual(state.highstate("whitelist=sls1.sls"), {'comment': 'Disabled', 'name': 'Salt highstate run is disabled. ' 'To re-enable, run state.enable highstate', 'result': 'False'}) mock = MagicMock(side_effect=["A", None, None]) with patch.object(state, '_check_queue', mock): self.assertEqual(state.highstate("whitelist=sls1.sls"), "A") with patch.dict(state.__opts__, {"test": "A"}): mock = MagicMock(return_value={'test': True}) with patch.object(state, '_get_opts', mock): self.assertRaises(SaltInvocationError, state.highstate, "whitelist=sls1.sls", pillar="A") mock = MagicMock(return_value=True) with patch.dict(state.__salt__, {'config.option': mock}): mock = MagicMock(return_value="A") with patch.object(state, '_filter_running', mock): mock = MagicMock(return_value=True) with patch.object(state, '_filter_running', mock): mock = MagicMock(return_value=True) with patch.object(salt.payload, 'Serial', mock): with patch.dict(state.__opts__, {"cachedir": "D"}): with patch.object(os.path, 'join', mock): with patch.object( state, '_set' '_retcode', mock): self.assertTrue(state. highstate (arg))
def test_highstate(self): ''' Test to retrieve the state data from the salt master for the minion and execute it ''' arg = "whitelist=sls1.sls" mock = MagicMock(side_effect=[True, False, False, False]) with patch.object(state, '_disabled', mock): self.assertDictEqual(state.highstate("whitelist=sls1.sls"), {'comment': 'Disabled', 'name': 'Salt highstate run is disabled. ' 'To re-enable, run state.enable highstate', 'result': 'False'}) mock = MagicMock(side_effect=["A", None, None]) with patch.object(state, '_check_queue', mock): self.assertEqual(state.highstate("whitelist=sls1.sls"), "A") with patch.dict(state.__opts__, {"test": "A"}): mock = MagicMock(return_value={'test': True}) with patch.object(state, '_get_opts', mock): self.assertRaises(SaltInvocationError, state.highstate, "whitelist=sls1.sls", pillar="A") mock = MagicMock(return_value=True) with patch.dict(state.__salt__, {'config.option': mock}): mock = MagicMock(return_value="A") with patch.object(state, '_filter_running', mock): mock = MagicMock(return_value=True) with patch.object(state, '_filter_running', mock): mock = MagicMock(return_value=True) with patch.object(salt.payload, 'Serial', mock): with patch.dict(state.__opts__, {"cachedir": "D"}): with patch.object(os.path, 'join', mock): with patch.object( state, '_set' '_retcode', mock): self.assertTrue(state. highstate (arg))
def test_highstate(): """ Test to retrieve the state data from the salt master for the minion and execute it """ arg = "whitelist=sls1.sls" mock = MagicMock(side_effect=[True, False, False, False]) with patch.object(state, "_disabled", mock): assert state.highstate("whitelist=sls1.sls") == { "comment": "Disabled", "name": "Salt highstate run is disabled. " "To re-enable, run state.enable highstate", "result": "False", } mock = MagicMock(side_effect=["A", None, None]) with patch.object(state, "_check_queue", mock): assert state.highstate("whitelist=sls1.sls") == "A" with patch.dict(state.__opts__, {"test": "A"}): mock = MagicMock(return_value={"test": True}) with patch.object(salt.utils.state, "get_sls_opts", mock): pytest.raises( SaltInvocationError, state.highstate, "whitelist=sls1.sls", pillar="A", ) mock = MagicMock(return_value="A") with patch.object(state, "_filter_running", mock): mock = MagicMock(return_value=True) with patch.object(state, "_filter_running", mock): mock = MagicMock(return_value=True) with patch.object(salt.payload, "Serial", mock): with patch.object(os.path, "join", mock): with patch.object(state, "_set" "_retcode", mock): assert state.highstate(arg)
def test_lock_saltenv(): """ Tests lock_saltenv in each function which accepts saltenv on the CLI """ lock_msg = "lock_saltenv is enabled, saltenv cannot be changed" empty_list_mock = MagicMock(return_value=[]) with patch.dict(state.__opts__, {"lock_saltenv": True}), patch.dict( state.__salt__, {"grains.get": empty_list_mock} ), patch.object(state, "running", empty_list_mock): # Test high with pytest.raises(CommandExecutionError, match=lock_msg): state.high([{"vim": {"pkg": ["installed"]}}], saltenv="base") # Test template with pytest.raises(CommandExecutionError, match=lock_msg): state.template("foo", saltenv="base") # Test template_str with pytest.raises(CommandExecutionError, match=lock_msg): state.template_str("foo", saltenv="base") # Test apply_ with SLS with pytest.raises(CommandExecutionError, match=lock_msg): state.apply_("foo", saltenv="base") # Test apply_ with Highstate with pytest.raises(CommandExecutionError, match=lock_msg): state.apply_(saltenv="base") # Test "test" with SLS with pytest.raises(CommandExecutionError, match=lock_msg): state.test("foo", saltenv="base") # Test "test" with Highstate with pytest.raises(CommandExecutionError, match=lock_msg): state.test(saltenv="base") # Test highstate with pytest.raises(CommandExecutionError, match=lock_msg): state.highstate(saltenv="base") # Test sls with pytest.raises(CommandExecutionError, match=lock_msg): state.sls("foo", saltenv="base") # Test top with pytest.raises(CommandExecutionError, match=lock_msg): state.top("foo.sls", saltenv="base") # Test show_highstate with pytest.raises(CommandExecutionError, match=lock_msg): state.show_highstate(saltenv="base") # Test show_lowstate with pytest.raises(CommandExecutionError, match=lock_msg): state.show_lowstate(saltenv="base") # Test sls_id with pytest.raises(CommandExecutionError, match=lock_msg): state.sls_id("foo", "bar", saltenv="base") # Test show_low_sls with pytest.raises(CommandExecutionError, match=lock_msg): state.show_low_sls("foo", saltenv="base") # Test show_sls with pytest.raises(CommandExecutionError, match=lock_msg): state.show_sls("foo", saltenv="base") # Test show_top with pytest.raises(CommandExecutionError, match=lock_msg): state.show_top(saltenv="base") # Test single with pytest.raises(CommandExecutionError, match=lock_msg): state.single("foo.bar", name="baz", saltenv="base") # Test pkg with pytest.raises(CommandExecutionError, match=lock_msg): state.pkg( "/tmp/salt_state.tgz", "760a9353810e36f6d81416366fc426dc", "md5", saltenv="base", )
def test_lock_saltenv(self): ''' Tests lock_saltenv in each function which accepts saltenv on the CLI ''' lock_msg = 'lock_saltenv is enabled, saltenv cannot be changed' empty_list_mock = MagicMock(return_value=[]) with patch.dict(state.__opts__, {'lock_saltenv': True}), \ patch.dict(state.__salt__, {'grains.get': empty_list_mock}), \ patch.object(state, 'running', empty_list_mock): # Test high with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.high( [{"vim": {"pkg": ["installed"]}}], saltenv='base') # Test template with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.template('foo', saltenv='base') # Test template_str with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.template_str('foo', saltenv='base') # Test apply_ with SLS with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.apply_('foo', saltenv='base') # Test apply_ with Highstate with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.apply_(saltenv='base') # Test highstate with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.highstate(saltenv='base') # Test sls with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.sls('foo', saltenv='base') # Test top with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.top('foo.sls', saltenv='base') # Test show_highstate with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.show_highstate(saltenv='base') # Test show_lowstate with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.show_lowstate(saltenv='base') # Test sls_id with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.sls_id('foo', 'bar', saltenv='base') # Test show_low_sls with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.show_low_sls('foo', saltenv='base') # Test show_sls with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.show_sls('foo', saltenv='base') # Test show_top with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.show_top(saltenv='base') # Test single with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.single('foo.bar', name='baz', saltenv='base') # Test pkg with self.assertRaisesRegex(CommandExecutionError, lock_msg): state.pkg( '/tmp/salt_state.tgz', '760a9353810e36f6d81416366fc426dc', 'md5', saltenv='base')