def test_append(self):
'''
Test if it append a rule to the specified table & chain.
'''
self.assertEqual(nftables.append(),
'Error: Chain needs to be specified')
self.assertEqual(nftables.append(chain='input'),
'Error: Rule needs to be specified')
_ru = 'input tcp dport 22 log accept'
ret = 'Error: table filter in family ipv4 does not exist'
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.append(chain='input', rule=_ru), ret)
ret = 'Error: chain input in table filter in family ipv4 does not exist'
mock = MagicMock(return_value='table ip filter')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.append(chain='input', rule=_ru), ret)
r_val = 'table ip filter chain input {{ input tcp dport 22 log accept #'
mock = MagicMock(return_value=r_val)
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertTrue(nftables.append(chain='input', rule=_ru))
def test_append_rule(self):
'''
Test if it append a rule to the specified table & chain.
'''
_ru = 'input tcp dport 22 log accept'
mock = MagicMock(side_effect=['1', ''])
with patch.dict(nftables.__salt__, {'cmd.run': mock}), \
patch('salt.modules.nftables.check',
MagicMock(return_value={'result': False,
'comment': ''})), \
patch('salt.modules.nftables.check_chain',
MagicMock(return_value={'result': True,
'comment': ''})), \
patch('salt.modules.nftables.check_table',
MagicMock(return_value={'result': True,
'comment': ''})):
_expected = {
'comment':
'Failed to add rule "{0}" chain input in table filter in family ipv4.'
.format(_ru),
'result':
False
}
self.assertEqual(nftables.append(chain='input', rule=_ru),
_expected)
_expected = {
'comment':
'Added rule "{0}" chain input in table filter in family ipv4.'.
format(_ru),
'result':
True
}
self.assertEqual(nftables.append(chain='input', rule=_ru),
_expected)
def test_append_rule(self):
"""
Test if it append a rule to the specified table & chain.
"""
_ru = "input tcp dport 22 log accept"
mock = MagicMock(side_effect=["1", ""])
with patch.dict(nftables.__salt__, {"cmd.run": mock}), patch(
"salt.modules.nftables.check",
MagicMock(return_value={"result": False, "comment": ""}),
), patch(
"salt.modules.nftables.check_chain",
MagicMock(return_value={"result": True, "comment": ""}),
), patch(
"salt.modules.nftables.check_table",
MagicMock(return_value={"result": True, "comment": ""}),
):
_expected = {
"comment": 'Failed to add rule "{}" chain input in table filter in family ipv4.'.format(
_ru
),
"result": False,
}
self.assertEqual(nftables.append(chain="input", rule=_ru), _expected)
_expected = {
"comment": 'Added rule "{}" chain input in table filter in family ipv4.'.format(
_ru
),
"result": True,
}
self.assertEqual(nftables.append(chain="input", rule=_ru), _expected)
def test_append(self):
'''
Test if it append a rule to the specified table & chain.
'''
self.assertEqual(nftables.append(),
'Error: Chain needs to be specified')
self.assertEqual(nftables.append(chain='input'),
'Error: Rule needs to be specified')
_ru = 'input tcp dport 22 log accept'
ret = 'Error: table filter in family ipv4 does not exist'
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.append(chain='input', rule=_ru), ret)
ret = 'Error: chain input in table filter in family ipv4 does not exist'
mock = MagicMock(return_value='table ip filter')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.append(chain='input', rule=_ru), ret)
r_val = 'table ip filter chain input {{ input tcp dport 22 log accept #'
mock = MagicMock(return_value=r_val)
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertTrue(nftables.append(chain='input', rule=_ru))
def test_append(self):
'''
Test if it append a rule to the specified table & chain.
'''
self.assertEqual(nftables.append(),
{'result': False,
'comment': 'Chain needs to be specified'})
self.assertEqual(nftables.append(chain='input'),
{'result': False,
'comment': 'Rule needs to be specified'})
_ru = 'input tcp dport 22 log accept'
ret = {'comment': 'Table filter in family ipv4 does not exist',
'result': False}
mock = MagicMock(return_value='')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.append(chain='input', rule=_ru), ret)
ret = {'comment': 'Chain input in table filter in family ipv4 does not exist',
'result': False}
mock = MagicMock(return_value='table ip filter')
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.append(chain='input', rule=_ru), ret)
r_val = 'table ip filter chain input {{ input tcp dport 22 log accept #'
mock = MagicMock(return_value=r_val)
_expected = {'comment': 'Rule input tcp dport 22 log accept chain input in table filter in family ipv4 already exists',
'result': False}
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertEqual(nftables.append(chain='input',
rule=_ru),
_expected)
def test_append_rule(self):
'''
Test if it append a rule to the specified table & chain.
'''
_ru = 'input tcp dport 22 log accept'
mock = MagicMock(side_effect=['1', ''])
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertFalse(nftables.append(chain='input', rule=_ru))
self.assertTrue(nftables.append(chain='input', rule=_ru))
def test_append_rule(self):
'''
Test if it append a rule to the specified table & chain.
'''
_ru = 'input tcp dport 22 log accept'
mock = MagicMock(side_effect=['1', ''])
with patch.dict(nftables.__salt__, {'cmd.run': mock}):
self.assertFalse(nftables.append(chain='input', rule=_ru))
self.assertTrue(nftables.append(chain='input', rule=_ru))
def test_append_rule(self):
'''
Test if it append a rule to the specified table & chain.
'''
_ru = 'input tcp dport 22 log accept'
mock = MagicMock(side_effect=['1', ''])
with patch.dict(nftables.__salt__, {'cmd.run': mock}), \
patch('salt.modules.nftables.check', MagicMock(return_value=False)), \
patch('salt.modules.nftables.check_chain', MagicMock(return_value=True)), \
patch('salt.modules.nftables.check_table', MagicMock(return_value=True)):
self.assertFalse(nftables.append(chain='input', rule=_ru))
self.assertTrue(nftables.append(chain='input', rule=_ru))
def test_append(self):
"""
Test if it append a rule to the specified table & chain.
"""
self.assertEqual(
nftables.append(),
{
"result": False,
"comment": "Chain needs to be specified"
},
)
self.assertEqual(
nftables.append(chain="input"),
{
"result": False,
"comment": "Rule needs to be specified"
},
)
_ru = "input tcp dport 22 log accept"
ret = {
"comment": "Table filter in family ipv4 does not exist",
"result": False
}
mock = MagicMock(return_value="")
with patch.dict(nftables.__salt__, {"cmd.run": mock}):
self.assertEqual(nftables.append(chain="input", rule=_ru), ret)
ret = {
"comment":
"Chain input in table filter in family ipv4 does not exist",
"result": False,
}
mock = MagicMock(return_value="table ip filter")
with patch.dict(nftables.__salt__, {"cmd.run": mock}):
self.assertEqual(nftables.append(chain="input", rule=_ru), ret)
r_val = "table ip filter chain input {{ input tcp dport 22 log accept #"
mock = MagicMock(return_value=r_val)
_expected = {
"comment":
"Rule input tcp dport 22 log accept chain input in table filter in family ipv4 already exists",
"result": False,
}
with patch.dict(nftables.__salt__, {"cmd.run": mock}):
self.assertEqual(nftables.append(chain="input", rule=_ru),
_expected)
