def test_add_vhost(self): ''' Test if it adds a vhost via rabbitmqctl add_vhost. ''' mock_run = MagicMock(return_value='saltstack') with patch.dict(rabbitmq.__salt__, {'cmd.run': mock_run}): self.assertDictEqual(rabbitmq.add_vhost('saltstack'), {'Added': 'saltstack'})
def test_add_vhost(): """ Test if it adds a vhost via rabbitmqctl add_vhost. """ mock_run = MagicMock(return_value={ "retcode": 0, "stdout": "saltstack", "stderr": "" }) with patch.dict(rabbitmq.__salt__, {"cmd.run_all": mock_run}): assert rabbitmq.add_vhost("saltstack") == {"Added": "saltstack"}