예제 #1
0
파일: ipset_test.py 프로젝트: mjura/salt-1
    def test_absent(self):
        '''
        Test to remove a entry or entries from a chain
        '''
        name = 'salt'
        entry = ['192.168.0.3', '192.168.1.3']

        ret = {'name': name,
               'result': False,
               'comment': '',
               'changes': {}}

        mock = MagicMock(side_effect=[True, False, False])
        with patch.dict(ipset.__salt__, {'ipset.check': mock}):
            comt = ('ipset entry must be specified')
            ret.update({'comment': comt})
            self.assertDictEqual(ipset.absent(name), ret)

            with patch.dict(ipset.__opts__, {'test': True}):
                comt = ('ipset entry 192.168.0.3 needs to removed '
                        'from set salt for family ipv4\n')
                ret.update({'comment': comt, 'result': None})
                self.assertDictEqual(ipset.absent(name, entry, set_name='salt'),
                                     ret)

            comt = ('ipset entry for 192.168.0.3 not present in set (salt) for '
                    'ipv4\nipset entry for 192.168.1.3 not present in set '
                    '(salt) for ipv4\n')
            ret.update({'comment': comt, 'result': True})
            self.assertDictEqual(ipset.absent(name, entry, set_name='salt'),
                                 ret)
예제 #2
0
 def _runner(
     self,
     expected_ret,
     test=False,
     check=False,
     delete=False,
     delete_assertion=False,
 ):
     mock_check = MagicMock(return_value=check)
     mock_delete = MagicMock(return_value=delete)
     with patch.dict(ipset.__opts__, {"test": test}):
         with patch.dict(ipset.__salt__, {
                 "ipset.check": mock_check,
                 "ipset.delete": mock_delete
         }):
             actual_ret = ipset.absent(self.fake_name,
                                       self.fake_entries,
                                       set_name=self.fake_name)
     mock_check.assert_has_calls(
         [call(self.fake_name, e, "ipv4") for e in self.fake_entries],
         any_order=True)
     if delete_assertion:
         expected_calls = [
             call(self.fake_name, e, "ipv4", set_name=self.fake_name)
             for e in self.fake_entries
         ]
         if delete is not True:
             expected_calls = expected_calls[:1]
         mock_delete.assert_has_calls(expected_calls, any_order=True)
     else:
         self.assertTrue(mock_delete.call_count == 0)
     self.assertDictEqual(actual_ret, expected_ret)
예제 #3
0
 def test_absent(self):
     ret = {'name': self.fake_name,
            'result': False,
            'comment': 'ipset entry must be specified',
            'changes': {}}
     mock = MagicMock(return_value=True)
     with patch.dict(ipset.__salt__, {'ipset.check': mock}):
         self.assertDictEqual(ipset.absent(self.fake_name), ret)
예제 #4
0
파일: ipset_test.py 프로젝트: bryson/salt
 def test_absent(self):
     ret = {'name': self.fake_name,
            'result': False,
            'comment': 'ipset entry must be specified',
            'changes': {}}
     mock = MagicMock(return_value=True)
     with patch.dict(ipset.__salt__, {'ipset.check': mock}):
         self.assertDictEqual(ipset.absent(self.fake_name), ret)
예제 #5
0
 def test_absent(self):
     ret = {
         "name": self.fake_name,
         "result": False,
         "comment": "ipset entry must be specified",
         "changes": {},
     }
     mock = MagicMock(return_value=True)
     with patch.dict(ipset.__salt__, {"ipset.check": mock}):
         self.assertDictEqual(ipset.absent(self.fake_name), ret)
예제 #6
0
파일: ipset_test.py 프로젝트: bryson/salt
 def _runner(self, expected_ret, test=False, check=False, delete=False,
             delete_assertion=False):
     mock_check = MagicMock(return_value=check)
     mock_delete = MagicMock(return_value=delete)
     with patch.dict(ipset.__opts__, {'test': test}):
         with patch.dict(ipset.__salt__, {'ipset.check': mock_check,
                                          'ipset.delete': mock_delete}):
             actual_ret = ipset.absent(self.fake_name, self.fake_entries, set_name=self.fake_name)
     mock_check.assert_has_calls([call(self.fake_name, e, 'ipv4') for e in self.fake_entries], any_order=True)
     if delete_assertion:
         expected_calls = [call(self.fake_name, e, 'ipv4', set_name=self.fake_name) for e in self.fake_entries]
         if delete is not True:
             expected_calls = expected_calls[:1]
         mock_delete.assert_has_calls(expected_calls, any_order=True)
     else:
         mock_delete.assert_not_called()
     self.assertDictEqual(actual_ret, expected_ret)