def test_delete(self): """ Test if it delete a rule from the specified table/chain """ with patch.object(iptables, "_has_option", MagicMock(return_value=True)): _rule = "m state --state RELATED,ESTABLISHED -j ACCEPT" self.assertEqual( iptables.delete(table="filter", chain=None, position=3, rule=_rule, family="ipv4"), "Error: Only specify a position or a rule, not both", ) mock = MagicMock(return_value=True) with patch.dict(iptables.__salt__, {"cmd.run": mock}): self.assertTrue( iptables.delete( table="filter", chain="INPUT", position=3, rule="", family="ipv4", ))
def test_delete(self): ''' Test if it delete a rule from the specified table/chain ''' _rule = 'm state --state RELATED,ESTABLISHED -j ACCEPT' self.assertEqual(iptables.delete(table='filter', chain=None, position=3, rule=_rule, family='ipv4'), 'Error: Only specify a position or a rule, not both') mock = MagicMock(return_value=True) with patch.dict(iptables.__salt__, {'cmd.run': mock}): self.assertTrue(iptables.delete(table='filter', chain='INPUT', position=3, rule='', family='ipv4'))