def test_set_config(self): ''' test to show installed version of dnsmasq. ''' mock = MagicMock(return_value={'conf-dir': 'A'}) with patch.object(dnsmasq, 'get_config', mock): mock = MagicMock(return_value=['.', '~', 'bak', '#']) with patch.object(os, 'listdir', mock): self.assertDictEqual(dnsmasq.set_config(), {})
def test_set_config(self): """ test to show installed version of dnsmasq. """ mock = MagicMock(return_value={"conf-dir": "A"}) with patch.object(dnsmasq, "get_config", mock): mock = MagicMock(return_value=[".", "~", "bak", "#"]) with patch.object(os, "listdir", mock): self.assertDictEqual(dnsmasq.set_config(), {})
def test_set_config_filter_pub_kwargs(self): ''' Test that the kwargs returned from running the set_config function do not contain the __pub that may have been passed through in **kwargs. ''' mock_domain = 'local' mock_address = '/some-test-address.local/8.8.4.4' with patch.dict(dnsmasq.__salt__, {'file.append': MagicMock()}): ret = dnsmasq.set_config(follow=False, domain=mock_domain, address=mock_address, __pub_pid=8184, __pub_jid=20161101194639387946, __pub_tgt='salt-call') self.assertEqual(ret, {'domain': mock_domain, 'address': mock_address})
def test_set_config_filter_pub_kwargs(self): """ Test that the kwargs returned from running the set_config function do not contain the __pub that may have been passed through in **kwargs. """ with patch("salt.modules.dnsmasq.get_config", MagicMock(return_value={"conf-dir": "A"})): mock_domain = "local" mock_address = "/some-test-address.local/8.8.4.4" with patch.dict(dnsmasq.__salt__, {"file.append": MagicMock()}): ret = dnsmasq.set_config( follow=False, domain=mock_domain, address=mock_address, __pub_pid=8184, __pub_jid=20161101194639387946, __pub_tgt="salt-call", ) self.assertEqual(ret, { "domain": mock_domain, "address": mock_address })