Exemplo n.º 1
0
    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])
        with patch.dict(nftables.__salt__, {"nftables.check_chain": mock}):
            ret.update({'comment': 'nftables salt chain is already'
                        ' exist in filter table for ipv4'})
            self.assertDictEqual(nftables.chain_present('salt'), ret)

            mock = MagicMock(side_effect=[True, ''])
            with patch.dict(nftables.__salt__, {"nftables.new_chain": mock}):
                ret.update({'changes': {'locale': 'salt'},
                            'comment': 'nftables salt chain in filter'
                            ' table create success for ipv4'})
                self.assertDictEqual(nftables.chain_present('salt'), ret)

                ret.update({'changes': {},
                            'comment': 'Failed to create salt chain'
                            ' in filter table:  for ipv4',
                            'result': False})
                self.assertDictEqual(nftables.chain_present('salt'), ret)
Exemplo n.º 2
0
    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])
        with patch.dict(nftables.__salt__, {"nftables.check_chain": mock}):
            ret.update({'comment': 'nftables salt chain is already'
                        ' exist in filter table for ipv4'})
            self.assertDictEqual(nftables.chain_present('salt'), ret)

            mock = MagicMock(side_effect=[True, ''])
            with patch.dict(nftables.__salt__, {"nftables.new_chain": mock}):
                ret.update({'changes': {'locale': 'salt'},
                            'comment': 'nftables salt chain in filter'
                            ' table create success for ipv4'})
                self.assertDictEqual(nftables.chain_present('salt'), ret)

                ret.update({'changes': {},
                            'comment': 'Failed to create salt chain'
                            ' in filter table:  for ipv4',
                            'result': False})
                self.assertDictEqual(nftables.chain_present('salt'), ret)
Exemplo n.º 3
0
    def test_chain_present(self):
        """
            Test to verify the chain is exist.
        """
        ret = {"name": "salt", "changes": {}, "result": True, "comment": ""}
        mock = MagicMock(side_effect=[
            {
                "result": True,
                "comment": ""
            },
            {
                "result": False,
                "comment": ""
            },
            {
                "result": False,
                "comment": ""
            },
        ])
        with patch.dict(nftables.__salt__, {"nftables.check_chain": mock}):
            ret.update({
                "comment":
                "nftables salt chain is already"
                " exist in filter table for ipv4"
            })
            self.assertDictEqual(nftables.chain_present("salt"), ret)

            mock = MagicMock(side_effect=[
                {
                    "result": True,
                    "comment": ""
                },
                {
                    "result": False,
                    "comment": ""
                },
            ])
            with patch.dict(nftables.__salt__, {"nftables.new_chain": mock}):
                with patch.dict(nftables.__opts__, {"test": False}):
                    ret.update({
                        "changes": {
                            "locale": "salt"
                        },
                        "comment":
                        "nftables salt chain in filter"
                        " table create success for ipv4",
                    })
                    self.assertDictEqual(nftables.chain_present("salt"), ret)

                    ret.update({
                        "changes": {},
                        "comment": "Failed to create salt chain"
                        " in filter table:  for ipv4",
                        "result": False,
                    })
                    self.assertDictEqual(nftables.chain_present("salt"), ret)