def test_add_rule(self):
'''
Test if it add a new firewall rule
'''
mock_cmd = MagicMock(return_value='Ok.')
with patch.dict(win_firewall.__salt__, {'cmd.run': mock_cmd}):
self.assertTrue(win_firewall.add_rule("test", "8080"))
def test_add_rule(self):
"""
Test if it add a new firewall rule
"""
mock_cmd = MagicMock(return_value="Ok.")
with patch.dict(win_firewall.__salt__, {"cmd.run": mock_cmd}):
self.assertTrue(win_firewall.add_rule("test", "8080"))
def test_add_rule(self):
'''
Test if it add a new firewall rule
'''
mock_cmd = MagicMock(return_value='Ok.')
with patch.dict(win_firewall.__salt__, {'cmd.run': mock_cmd}):
self.assertTrue(win_firewall.add_rule("test", "8080"))
mock_cmd.assert_called_once_with(['netsh', 'advfirewall', 'firewall', 'add', 'rule', 'name=test',
'protocol=tcp', 'dir=in', 'action=allow', 'localport=8080'],
python_shell=False)
def test_add_rule_icmp4_any(self):
'''
Test if it add a new firewall rule
'''
mock_cmd = MagicMock(return_value='Ok.')
with patch.dict(win_firewall.__salt__, {'cmd.run': mock_cmd}):
self.assertTrue(
win_firewall.add_rule("test", "1", protocol='icmpv4:any,any'))
mock_cmd.assert_called_once_with([
'netsh', 'advfirewall', 'firewall', 'add', 'rule', 'name=test',
'protocol=icmpv4:any,any', 'dir=in', 'action=allow',
'remoteip=any'
],
python_shell=False)
def test_add_rule_icmp4_any(self):
'''
Test if it add a new firewall rule
'''
mock_cmd = MagicMock(return_value='Ok.')
with patch.dict(win_firewall.__salt__, {'cmd.run': mock_cmd}):
self.assertTrue(win_firewall.add_rule("test", "1", protocol='icmpv4:any,any'))
mock_cmd.assert_called_once_with(['netsh', 'advfirewall', 'firewall', 'add', 'rule',
'name=test',
'protocol=icmpv4:any,any',
'dir=in',
'action=allow',
'remoteip=any'],
python_shell=False)
