salt/tests/unit/acl/client_test.py
Dmitry Kuzmenko 27017e34ce Regex support in publisher ACL rules.
Users in publisher ACL and users and functions in the publisher ACL
blacklist now have matched in the following order:
1. Equality
2. Glob
3. Regex

Users in publisher ACL still not supported in systems without pwd python
module (Windows, any else?)
2016-07-18 19:15:28 +03:00

76 lines
2.7 KiB
Python

# -*- coding: utf-8 -*-
# Import python libs
from __future__ import absolute_import
# Import Salt Libs
from salt import acl
# Import Salt Testing Libs
from salttesting import TestCase, skipIf
from salttesting.mock import (
NO_MOCK,
NO_MOCK_REASON,
)
from salttesting.helpers import ensure_in_syspath
ensure_in_syspath('../../')
@skipIf(NO_MOCK, NO_MOCK_REASON)
class ClientACLTestCase(TestCase):
'''
Unit tests for salt.acl.ClientACL
'''
def setUp(self):
self.blacklist = {
'users': ['joker', 'penguin', '*bad_*', 'blocked_.*', '^Homer$'],
'modules': ['cmd.run', 'test.fib', 'rm-rf.*'],
}
def test_user_is_blacklisted(self):
'''
test user_is_blacklisted
'''
client_acl = acl.PublisherACL(self.blacklist)
self.assertTrue(client_acl.user_is_blacklisted('joker'))
self.assertTrue(client_acl.user_is_blacklisted('penguin'))
self.assertTrue(client_acl.user_is_blacklisted('bad_'))
self.assertTrue(client_acl.user_is_blacklisted('bad_user'))
self.assertTrue(client_acl.user_is_blacklisted('bad_*'))
self.assertTrue(client_acl.user_is_blacklisted('user_bad_'))
self.assertTrue(client_acl.user_is_blacklisted('blocked_'))
self.assertTrue(client_acl.user_is_blacklisted('blocked_user'))
self.assertTrue(client_acl.user_is_blacklisted('blocked_.*'))
self.assertTrue(client_acl.user_is_blacklisted('Homer'))
self.assertFalse(client_acl.user_is_blacklisted('batman'))
self.assertFalse(client_acl.user_is_blacklisted('robin'))
self.assertFalse(client_acl.user_is_blacklisted('bad'))
self.assertFalse(client_acl.user_is_blacklisted('blocked'))
self.assertFalse(client_acl.user_is_blacklisted('NotHomer'))
self.assertFalse(client_acl.user_is_blacklisted('HomerSimpson'))
def test_cmd_is_blacklisted(self):
'''
test cmd_is_blacklisted
'''
client_acl = acl.PublisherACL(self.blacklist)
self.assertTrue(client_acl.cmd_is_blacklisted('cmd.run'))
self.assertTrue(client_acl.cmd_is_blacklisted('test.fib'))
self.assertTrue(client_acl.cmd_is_blacklisted('rm-rf.root'))
self.assertFalse(client_acl.cmd_is_blacklisted('cmd.shell'))
self.assertFalse(client_acl.cmd_is_blacklisted('test.versions'))
self.assertFalse(client_acl.cmd_is_blacklisted('arm-rf.root'))
self.assertTrue(client_acl.cmd_is_blacklisted(['cmd.run', 'state.sls']))
self.assertFalse(client_acl.cmd_is_blacklisted(['state.highstate', 'state.sls']))
if __name__ == '__main__':
from integration import run_tests
run_tests(ClientACLTestCase, needs_daemon=False)