def test_chain_present(self): """ Test to verify the chain is exist. """ ret = {"name": "salt", "changes": {}, "result": True, "comment": ""} mock = MagicMock(side_effect=[True, False, False, False]) with patch.dict(iptables.__salt__, {"iptables.check_chain": mock}): ret.update( { "comment": ( "iptables salt chain is already exist in filter table for ipv4" ) } ) self.assertDictEqual(iptables.chain_present("salt"), ret) with patch.dict(iptables.__opts__, {"test": True}): ret.update( { "comment": ( "iptables salt chain in filter" " table needs to be set for ipv4" ), "result": None, } ) self.assertDictEqual(iptables.chain_present("salt"), ret) with patch.dict(iptables.__opts__, {"test": False}): mock = MagicMock(side_effect=[True, ""]) with patch.dict(iptables.__salt__, {"iptables.new_chain": mock}): ret.update( { "result": True, "comment": ( "iptables salt chain in filter" " table create success for ipv4" ), "changes": {"locale": "salt"}, } ) self.assertDictEqual(iptables.chain_present("salt"), ret) ret.update( { "changes": {}, "result": False, "comment": ( "Failed to create salt chain in filter table: for ipv4" ), } ) self.assertDictEqual(iptables.chain_present("salt"), ret)
def test_chain_present(self): ''' Test to verify the chain is exist. ''' ret = {'name': 'salt', 'changes': {}, 'result': True, 'comment': ''} mock = MagicMock(side_effect=[True, False, False, False]) with patch.dict(iptables.__salt__, {'iptables.check_chain': mock}): ret.update({ 'comment': 'iptables salt chain is already' ' exist in filter table for ipv4' }) self.assertDictEqual(iptables.chain_present("salt"), ret) with patch.dict(iptables.__opts__, {'test': True}): ret.update({ 'comment': 'iptables salt chain in filter' ' table needs to be set for ipv4', 'result': None }) self.assertDictEqual(iptables.chain_present("salt"), ret) with patch.dict(iptables.__opts__, {'test': False}): mock = MagicMock(side_effect=[True, '']) with patch.dict(iptables.__salt__, {'iptables.new_chain': mock}): ret.update({ 'result': True, 'comment': 'iptables salt chain in filter' ' table create success for ipv4', 'changes': { 'locale': 'salt' } }) self.assertDictEqual(iptables.chain_present('salt'), ret) ret.update({ 'changes': {}, 'result': False, 'comment': 'Failed to create salt chain' ' in filter table: for ipv4' }) self.assertDictEqual(iptables.chain_present('salt'), ret)
def test_chain_present(self): ''' Test to verify the chain is exist. ''' ret = {'name': 'salt', 'changes': {}, 'result': True, 'comment': '' } mock = MagicMock(side_effect=[True, False, False, False]) with patch.dict(iptables.__salt__, {'iptables.check_chain': mock}): ret.update({'comment': 'iptables salt chain is already' ' exist in filter table for ipv4'}) self.assertDictEqual(iptables.chain_present("salt"), ret) with patch.dict(iptables.__opts__, {'test': True}): ret.update({'comment': 'iptables salt chain in filter' ' table needs to be set for ipv4', 'result': None}) self.assertDictEqual(iptables.chain_present("salt"), ret) with patch.dict(iptables.__opts__, {'test': False}): mock = MagicMock(side_effect=[True, '']) with patch.dict(iptables.__salt__, {'iptables.new_chain': mock}): ret.update({'result': True, 'comment': 'iptables salt chain in filter' ' table create success for ipv4', 'changes': {'locale': 'salt'}}) self.assertDictEqual(iptables.chain_present('salt'), ret) ret.update({'changes': {}, 'result': False, 'comment': 'Failed to create salt chain' ' in filter table: for ipv4'}) self.assertDictEqual(iptables.chain_present('salt'), ret)