def test_make_host_acl(self, make_acl_mock):
"""Test host acl."""
zk = zksasl.SASLZkClient()
zk.make_host_acl('foo', 'rdwca')
make_acl_mock.assert_called_once_with(
scheme='sasl', credential='host/foo@my-realm', read=True,
write=True, delete=True, create=True, admin=True
)
def test_make_role_acl(self, make_acl_mock):
"""Test constructing role acl for valid role."""
zk = zksasl.SASLZkClient()
zk.make_role_acl('servers', 'ra')
make_acl_mock.assert_called_once_with(
scheme='sasl', credential='file:///treadmill/roles/servers',
read=True, write=False, delete=False, create=False, admin=True
)
def test_make_user_acl(self, make_acl_mock):
"""Test constucting user acl."""
zk = zksasl.SASLZkClient()
zk.make_user_acl('foo', 'rw')
make_acl_mock.assert_called_once_with(
scheme='sasl', credential='foo', read=True,
write=True, create=False, delete=False, admin=False
)
def test_connect_without_connargs(self, zkclient_mock):
"""Test connect.
"""
zk = zksasl.SASLZkClient(hosts='zkservice@123:21')
zkclient_mock.assert_called_once_with(hosts='123:21',
sasl_data={
'service': 'zkservice',
'mechanisms': ['GSSAPI']
})
def test_connect_with_connargs(self, zkclient_mock):
"""Test connect with args.
"""
zk = zksasl.SASLZkClient(hosts='lala@foobar:123',
sasl_data={
'service': 'foo',
'mechanisms': ['bar']
})
zkclient_mock.assert_called_once_with(hosts='foobar:123',
sasl_data={
'service': 'lala',
'mechanisms': ['bar']
})
def test_make_role_acl(self, make_acl_mock):
"""Test constructing role acl for valid role."""
zk = zksasl.SASLZkClient()
zk.make_role_acl('servers', 'ra')
# TODO: re-enable when role support is added.
#
# make_acl_mock.assert_called_once_with(
# scheme='sasl', credential='role/servers', read=True,
# write=False, delete=False, create=False, admin=True
# )
make_acl_mock.assert_called_once_with('world',
'anyone',
read=True,
write=False,
delete=False,
create=False,
admin=True)
