def test_get_rule_handle(self): ''' Test if it get the handle for a particular rule ''' self.assertEqual(nftables.get_rule_handle(), {'result': False, 'comment': 'Chain needs to be specified'}) self.assertEqual(nftables.get_rule_handle(chain='input'), {'result': False, 'comment': 'Rule needs to be specified'}) _ru = 'input tcp dport 22 log accept' ret = {'result': False, 'comment': 'Table filter in family ipv4 does not exist'} mock = MagicMock(return_value='') with patch.dict(nftables.__salt__, {'cmd.run': mock}): self.assertEqual(nftables.get_rule_handle(chain='input', rule=_ru), ret) ret = {'result': False, 'comment': '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.get_rule_handle(chain='input', rule=_ru), ret) ret = {'result': False, 'comment': ('Rule input tcp dport 22 log accept chain' ' input in table filter in family ipv4 does not exist')} ret1 = {'result': False, 'comment': 'Could not find rule input tcp dport 22 log accept'} with patch.object(nftables, 'check_table', MagicMock(return_value={'result': True, 'comment': ''})): with patch.object(nftables, 'check_chain', MagicMock(return_value={'result': True, 'comment': ''})): _ret1 = {'result': False, 'comment': ('Rule input tcp dport 22 log accept' ' chain input in table filter in' ' family ipv4 does not exist')} _ret2 = {'result': True, 'comment': ''} with patch.object(nftables, 'check', MagicMock(side_effect=[_ret1, _ret2])): self.assertEqual(nftables.get_rule_handle(chain='input', rule=_ru), ret) _ru = 'input tcp dport 22 log accept' mock = MagicMock(return_value='') with patch.dict(nftables.__salt__, {'cmd.run': mock}): self.assertEqual(nftables.get_rule_handle(chain='input', rule=_ru), ret1)
def test_get_rule_handle(self): ''' Test if it get the handle for a particular rule ''' self.assertEqual(nftables.get_rule_handle(), 'Error: Chain needs to be specified') self.assertEqual(nftables.get_rule_handle(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.get_rule_handle(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.get_rule_handle(chain='input', rule=_ru), ret) ret = ('Error: rule input tcp dport 22 log accept chain input' ' in table filter in family ipv4 does not exist') ret1 = 'Error: could not find rule input tcp dport 22 log accept' with patch.object(nftables, 'check_table', MagicMock(return_value=True)): with patch.object(nftables, 'check_chain', MagicMock(return_value=True)): with patch.object(nftables, 'check', MagicMock(side_effect=[False, True])): self.assertEqual( nftables.get_rule_handle(chain='input', rule=_ru), ret) _ru = 'input tcp dport 22 log accept' mock = MagicMock(return_value='') with patch.dict(nftables.__salt__, {'cmd.run': mock}): self.assertEqual( nftables.get_rule_handle(chain='input', rule=_ru), ret1)
def test_get_rule_handle(self): ''' Test if it get the handle for a particular rule ''' self.assertEqual(nftables.get_rule_handle(), 'Error: Chain needs to be specified') self.assertEqual(nftables.get_rule_handle(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.get_rule_handle(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.get_rule_handle(chain='input', rule=_ru), ret) ret = ('Error: rule input tcp dport 22 log accept chain input' ' in table filter in family ipv4 does not exist') ret1 = 'Error: could not find rule input tcp dport 22 log accept' with patch.object(nftables, 'check_table', MagicMock(return_value=True)): with patch.object(nftables, 'check_chain', MagicMock(return_value=True)): with patch.object(nftables, 'check', MagicMock(side_effect=[False, True])): self.assertEqual(nftables.get_rule_handle(chain='input', rule=_ru), ret) _ru = 'input tcp dport 22 log accept' mock = MagicMock(return_value='') with patch.dict(nftables.__salt__, {'cmd.run': mock}): self.assertEqual(nftables.get_rule_handle(chain='input', rule=_ru), ret1)
def test_get_rule_handle(self): """ Test if it get the handle for a particular rule """ self.assertEqual( nftables.get_rule_handle(), {"result": False, "comment": "Chain needs to be specified"}, ) self.assertEqual( nftables.get_rule_handle(chain="input"), {"result": False, "comment": "Rule needs to be specified"}, ) _ru = "input tcp dport 22 log accept" ret = {"result": False, "comment": "Table filter in family ipv4 does not exist"} mock = MagicMock(return_value="") with patch.dict(nftables.__salt__, {"cmd.run": mock}): self.assertEqual(nftables.get_rule_handle(chain="input", rule=_ru), ret) ret = { "result": False, "comment": "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.get_rule_handle(chain="input", rule=_ru), ret) ret = { "result": False, "comment": ( "Rule input tcp dport 22 log accept chain" " input in table filter in family ipv4 does not exist" ), } ret1 = { "result": False, "comment": "Could not find rule input tcp dport 22 log accept", } with patch.object( nftables, "check_table", MagicMock(return_value={"result": True, "comment": ""}), ): with patch.object( nftables, "check_chain", MagicMock(return_value={"result": True, "comment": ""}), ): _ret1 = { "result": False, "comment": ( "Rule input tcp dport 22 log accept" " chain input in table filter in" " family ipv4 does not exist" ), } _ret2 = {"result": True, "comment": ""} with patch.object( nftables, "check", MagicMock(side_effect=[_ret1, _ret2]) ): self.assertEqual( nftables.get_rule_handle(chain="input", rule=_ru), ret ) _ru = "input tcp dport 22 log accept" mock = MagicMock(return_value="") with patch.dict(nftables.__salt__, {"cmd.run": mock}): self.assertEqual( nftables.get_rule_handle(chain="input", rule=_ru), ret1 )