def test_set_policy(self):
'''
Test if it set a policy based on rabbitmqctl set_policy.
'''
mock_run = MagicMock(return_value='saltstack')
with patch.dict(rabbitmq.__salt__, {'cmd.run': mock_run}):
self.assertDictEqual(rabbitmq.set_policy('/', 'HA', '.*',
'{"ha-mode": "all"}'),
{'Set': 'saltstack'})
def test_set_policy(self):
'''
Test if it set a policy based on rabbitmqctl set_policy.
'''
mock_run = MagicMock(return_value='saltstack')
with patch.dict(rabbitmq.__salt__, {'cmd.run': mock_run}):
self.assertDictEqual(
rabbitmq.set_policy('/', 'HA', '.*', '{"ha-mode": "all"}'),
{'Set': 'saltstack'})
def test_set_policy(self):
"""
Test if it set a policy based on rabbitmqctl set_policy.
"""
mock_run = MagicMock(
return_value={"retcode": 0, "stdout": "saltstack", "stderr": ""}
)
with patch.dict(rabbitmq.__salt__, {"cmd.run_all": mock_run}):
self.assertDictEqual(
rabbitmq.set_policy("/", "HA", ".*", '{"ha-mode": "all"}'),
{"Set": "saltstack"},
)