Exemple #1
0
    def test_check(self):
        '''
        Test for Check that an entry exists in the specified set.
        '''
        self.assertEqual(ipset.check(), 'Error: Set needs to be specified')

        self.assertEqual(ipset.check('s'),
                         'Error: Entry needs to be specified')

        with patch.object(ipset, '_find_set_type', return_value=None):
            self.assertEqual(ipset.check('set', 'entry'),
                             'Error: Set set does not exist')

        with patch.object(ipset, '_find_set_type', return_value=True):
            with patch.object(ipset, '_find_set_members',
                              side_effect=['entry', '']):
                self.assertTrue(ipset.check('set', 'entry'))
                self.assertFalse(ipset.check('set', 'entry'))
Exemple #2
0
    def test_check(self):
        """
        Test for Check that an entry exists in the specified set.
        """
        self.assertEqual(ipset.check(), "Error: Set needs to be specified")

        self.assertEqual(ipset.check("s"),
                         "Error: Entry needs to be specified")

        with patch.object(ipset, "_find_set_type", return_value=None):
            self.assertEqual(ipset.check("set", "entry"),
                             "Error: Set set does not exist")

        with patch.object(ipset, "_find_set_type", return_value="hash:ip"):
            with patch.object(
                    ipset,
                    "_find_set_members",
                    side_effect=[
                        "entry",
                        "",
                        ["192.168.0.4", "192.168.0.5"],
                        ["192.168.0.3"],
                        ["192.168.0.6"],
                        ["192.168.0.4", "192.168.0.5"],
                        ["192.168.0.3"],
                        ["192.168.0.6"],
                    ],
            ):
                self.assertTrue(ipset.check("set", "entry"))
                self.assertFalse(ipset.check("set", "entry"))
                self.assertTrue(ipset.check("set", "192.168.0.4/31"))
                self.assertFalse(ipset.check("set", "192.168.0.4/31"))
                self.assertFalse(ipset.check("set", "192.168.0.4/31"))
                self.assertTrue(ipset.check("set", "192.168.0.4-192.168.0.5"))
                self.assertFalse(ipset.check("set", "192.168.0.4-192.168.0.5"))
                self.assertFalse(ipset.check("set", "192.168.0.4-192.168.0.5"))

        with patch.object(ipset, "_find_set_type", return_value="hash:net"):
            with patch.object(
                    ipset,
                    "_find_set_members",
                    side_effect=[
                        "entry",
                        "",
                        "192.168.0.4/31",
                        "192.168.0.4/30",
                        "192.168.0.4/31",
                        "192.168.0.4/30",
                    ],
            ):
                self.assertTrue(ipset.check("set", "entry"))
                self.assertFalse(ipset.check("set", "entry"))
                self.assertTrue(ipset.check("set", "192.168.0.4/31"))
                self.assertFalse(ipset.check("set", "192.168.0.4/31"))
                self.assertTrue(ipset.check("set", "192.168.0.4-192.168.0.5"))
                self.assertFalse(ipset.check("set", "192.168.0.4-192.168.0.5"))
Exemple #3
0
    def test_check(self):
        '''
        Test for Check that an entry exists in the specified set.
        '''
        self.assertEqual(ipset.check(), 'Error: Set needs to be specified')

        self.assertEqual(ipset.check('s'),
                         'Error: Entry needs to be specified')

        with patch.object(ipset, '_find_set_type', return_value=None):
            self.assertEqual(ipset.check('set', 'entry'),
                             'Error: Set set does not exist')

        with patch.object(ipset, '_find_set_type', return_value='hash:ip'):
            with patch.object(ipset, '_find_set_members',
                              side_effect=['entry', '',
                                           ['192.168.0.4', '192.168.0.5'],
                                           ['192.168.0.3'], ['192.168.0.6'],
                                           ['192.168.0.4', '192.168.0.5'],
                                           ['192.168.0.3'], ['192.168.0.6'],
                                           ]):
                self.assertTrue(ipset.check('set', 'entry'))
                self.assertFalse(ipset.check('set', 'entry'))
                self.assertTrue(ipset.check('set', '192.168.0.4/31'))
                self.assertFalse(ipset.check('set', '192.168.0.4/31'))
                self.assertFalse(ipset.check('set', '192.168.0.4/31'))
                self.assertTrue(ipset.check('set', '192.168.0.4-192.168.0.5'))
                self.assertFalse(ipset.check('set', '192.168.0.4-192.168.0.5'))
                self.assertFalse(ipset.check('set', '192.168.0.4-192.168.0.5'))

        with patch.object(ipset, '_find_set_type', return_value='hash:net'):
            with patch.object(ipset, '_find_set_members',
                              side_effect=['entry', '',
                                           '192.168.0.4/31', '192.168.0.4/30',
                                           '192.168.0.4/31', '192.168.0.4/30',
                                           ]):
                self.assertTrue(ipset.check('set', 'entry'))
                self.assertFalse(ipset.check('set', 'entry'))
                self.assertTrue(ipset.check('set', '192.168.0.4/31'))
                self.assertFalse(ipset.check('set', '192.168.0.4/31'))
                self.assertTrue(ipset.check('set', '192.168.0.4-192.168.0.5'))
                self.assertFalse(ipset.check('set', '192.168.0.4-192.168.0.5'))
Exemple #4
0
    def test_check(self):
        '''
        Test for Check that an entry exists in the specified set.
        '''
        self.assertEqual(ipset.check(), 'Error: Set needs to be specified')

        self.assertEqual(ipset.check('s'),
                         'Error: Entry needs to be specified')

        with patch.object(ipset, '_find_set_type', return_value=None):
            self.assertEqual(ipset.check('set', 'entry'),
                             'Error: Set set does not exist')

        with patch.object(ipset, '_find_set_type', return_value=True):
            with patch.object(ipset,
                              '_find_set_members',
                              side_effect=[
                                  'entry',
                                  '',
                                  ["192.168.0.4", "192.168.0.5"],
                                  ["192.168.0.3"],
                                  ["192.168.0.6"],
                                  ["192.168.0.4", "192.168.0.5"],
                                  ["192.168.0.3"],
                                  ["192.168.0.6"],
                              ]):
                self.assertTrue(ipset.check('set', 'entry'))
                self.assertFalse(ipset.check('set', 'entry'))
                self.assertTrue(ipset.check('set', '192.168.0.4/31'))
                self.assertFalse(ipset.check('set', '192.168.0.4/31'))
                self.assertFalse(ipset.check('set', '192.168.0.4/31'))
                self.assertTrue(ipset.check('set', '192.168.0.4-192.168.0.5'))
                self.assertFalse(ipset.check('set', '192.168.0.4-192.168.0.5'))
                self.assertFalse(ipset.check('set', '192.168.0.4-192.168.0.5'))