1617 lines
62 KiB
Python
1617 lines
62 KiB
Python
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import json
|
|
import mox
|
|
import six
|
|
import uuid
|
|
|
|
from keystoneclient.auth.identity import v3 as ks_auth_v3
|
|
import keystoneclient.exceptions as kc_exception
|
|
from keystoneclient import session as ks_session
|
|
from keystoneclient.v3 import client as kc_v3
|
|
from keystoneclient.v3 import domains as kc_v3_domains
|
|
from oslo.config import cfg
|
|
|
|
from heat.common import exception
|
|
from heat.common import heat_keystoneclient
|
|
from heat.tests import common
|
|
from heat.tests import utils
|
|
|
|
cfg.CONF.import_opt('region_name_for_services', 'heat.common.config')
|
|
cfg.CONF.import_group('keystone_authtoken',
|
|
'keystonemiddleware.auth_token')
|
|
|
|
|
|
class KeystoneClientTest(common.HeatTestCase):
|
|
"""Test cases for heat.common.heat_keystoneclient."""
|
|
|
|
def setUp(self):
|
|
super(KeystoneClientTest, self).setUp()
|
|
|
|
self.mock_admin_client = self.m.CreateMock(kc_v3.Client)
|
|
self.mock_ks_v3_client = self.m.CreateMock(kc_v3.Client)
|
|
self.mock_ks_v3_client_domain_mngr = self.m.CreateMock(
|
|
kc_v3_domains.DomainManager)
|
|
self.m.StubOutWithMock(kc_v3, "Client")
|
|
|
|
dummy_url = 'http://server.test:5000/v2.0'
|
|
cfg.CONF.set_override('auth_uri', dummy_url,
|
|
group='keystone_authtoken')
|
|
cfg.CONF.set_override('admin_user', 'heat',
|
|
group='keystone_authtoken')
|
|
cfg.CONF.set_override('admin_password', 'verybadpass',
|
|
group='keystone_authtoken')
|
|
cfg.CONF.set_override('admin_tenant_name', 'service',
|
|
group='keystone_authtoken')
|
|
cfg.CONF.set_override('stack_user_domain_id', 'adomain123')
|
|
cfg.CONF.set_override('stack_domain_admin', 'adminuser123')
|
|
cfg.CONF.set_override('stack_domain_admin_password', 'adminsecret')
|
|
self.addCleanup(self.m.VerifyAll)
|
|
|
|
def _clear_domain_override(self):
|
|
cfg.CONF.clear_override('stack_user_domain_id')
|
|
|
|
def _stub_admin_client(self, auth_ok=True):
|
|
kc_v3.Client(
|
|
auth_url='http://server.test:5000/v3',
|
|
cacert=None,
|
|
cert=None,
|
|
endpoint='http://server.test:5000/v3',
|
|
insecure=False,
|
|
key=None,
|
|
password='verybadpass',
|
|
project_name='service',
|
|
username='heat').AndReturn(self.mock_admin_client)
|
|
self.mock_admin_client.domains = self.mock_ks_v3_client_domain_mngr
|
|
self.mock_admin_client.authenticate().AndReturn(auth_ok)
|
|
if auth_ok:
|
|
self.mock_admin_client.auth_ref = self.m.CreateMockAnything()
|
|
self.mock_admin_client.auth_ref.user_id = '1234'
|
|
|
|
def _stub_domain_admin_client(self, auth_ok=True):
|
|
kc_v3.Client(
|
|
auth_url='http://server.test:5000/v3',
|
|
cacert=None,
|
|
cert=None,
|
|
endpoint='http://server.test:5000/v3',
|
|
insecure=False,
|
|
key=None,
|
|
password='adminsecret',
|
|
user_domain_id='adomain123',
|
|
username='adminuser123').AndReturn(self.mock_admin_client)
|
|
self.mock_admin_client.domains = self.mock_ks_v3_client_domain_mngr
|
|
self.mock_admin_client.authenticate(
|
|
domain_id='adomain123').AndReturn(auth_ok)
|
|
if auth_ok:
|
|
self.mock_admin_client.auth_ref = self.m.CreateMockAnything()
|
|
self.mock_admin_client.auth_ref.user_id = '1234'
|
|
|
|
def _stubs_v3(self, method='token', auth_ok=True, trust_scoped=True,
|
|
user_id='trustor_user_id', auth_ref=None):
|
|
|
|
if method == 'token':
|
|
kc_v3.Client(
|
|
token='abcd1234', project_id='test_tenant_id',
|
|
auth_url='http://server.test:5000/v3',
|
|
endpoint='http://server.test:5000/v3',
|
|
cacert=None,
|
|
cert=None,
|
|
insecure=False,
|
|
key=None).AndReturn(self.mock_ks_v3_client)
|
|
elif method == 'auth_ref':
|
|
kc_v3.Client(
|
|
auth_ref=auth_ref,
|
|
auth_url='http://server.test:5000/v3',
|
|
endpoint='http://server.test:5000/v3',
|
|
cacert=None,
|
|
cert=None,
|
|
insecure=False,
|
|
key=None).AndReturn(self.mock_ks_v3_client)
|
|
elif method == 'password':
|
|
kc_v3.Client(
|
|
username='test_username',
|
|
password='password',
|
|
project_id='test_tenant_id',
|
|
auth_url='http://server.test:5000/v3',
|
|
endpoint='http://server.test:5000/v3',
|
|
cacert=None,
|
|
cert=None,
|
|
insecure=False,
|
|
key=None).AndReturn(self.mock_ks_v3_client)
|
|
elif method == 'trust':
|
|
kc_v3.Client(
|
|
username='heat',
|
|
password='verybadpass',
|
|
auth_url='http://server.test:5000/v3',
|
|
endpoint='http://server.test:5000/v3',
|
|
trust_id='atrust123',
|
|
cacert=None,
|
|
cert=None,
|
|
insecure=False,
|
|
key=None).AndReturn(self.mock_ks_v3_client)
|
|
self.mock_ks_v3_client.auth_ref = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.auth_ref.user_id = user_id
|
|
self.mock_ks_v3_client.auth_ref.trust_scoped = trust_scoped
|
|
self.mock_ks_v3_client.auth_ref.auth_token = 'atrusttoken'
|
|
|
|
if method != 'auth_ref':
|
|
self.mock_ks_v3_client.authenticate().AndReturn(auth_ok)
|
|
|
|
def test_username_length(self):
|
|
"""Test that user names >64 characters are properly truncated."""
|
|
|
|
self._stubs_v3()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# a >64 character user name and the expected version
|
|
long_user_name = 'U' * 64 + 'S'
|
|
good_user_name = long_user_name[-64:]
|
|
# mock keystone client user functions
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
mock_user = self.m.CreateMockAnything()
|
|
mock_user.id = 'auser123'
|
|
# when keystone is called, the name should have been truncated
|
|
# to the last 64 characters of the long name
|
|
self.mock_ks_v3_client.users.create(name=good_user_name,
|
|
password='password',
|
|
default_project=ctx.tenant_id
|
|
).AndReturn(mock_user)
|
|
|
|
self.mock_ks_v3_client.roles = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.roles.list(
|
|
name='heat_stack_user').AndReturn(self._mock_roles_list())
|
|
self.mock_ks_v3_client.roles.grant(project=ctx.tenant_id,
|
|
role='4546',
|
|
user='auser123').AndReturn(None)
|
|
self.m.ReplayAll()
|
|
# call create_stack_user with a long user name.
|
|
# the cleanup VerifyAll should verify that though we passed
|
|
# long_user_name, keystone was actually called with a truncated
|
|
# user name
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.create_stack_user(long_user_name, password='password')
|
|
|
|
def test_create_stack_user_error_norole(self):
|
|
"""Test error path when no role is found."""
|
|
|
|
self._stubs_v3()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
self.mock_ks_v3_client.roles = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.roles.list(
|
|
name='heat_stack_user').AndReturn([])
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
err = self.assertRaises(exception.Error,
|
|
heat_ks_client.create_stack_user,
|
|
'auser', password='password')
|
|
self.assertIn('Can\'t find role heat_stack_user', six.text_type(err))
|
|
|
|
def _mock_roles_list(self, heat_stack_user='heat_stack_user'):
|
|
mock_roles_list = []
|
|
mock_role = self.m.CreateMockAnything()
|
|
mock_role.id = '4546'
|
|
mock_role.name = heat_stack_user
|
|
mock_roles_list.append(mock_role)
|
|
return mock_roles_list
|
|
|
|
def test_create_stack_domain_user(self):
|
|
"""Test creating a stack domain user."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.users = self.m.CreateMockAnything()
|
|
mock_user = self.m.CreateMockAnything()
|
|
mock_user.id = 'duser123'
|
|
self.mock_admin_client.users.create(name='duser',
|
|
password=None,
|
|
default_project='aproject',
|
|
domain='adomain123'
|
|
).AndReturn(mock_user)
|
|
self.mock_admin_client.roles = self.m.CreateMockAnything()
|
|
self.mock_admin_client.roles.list(
|
|
name='heat_stack_user').AndReturn(self._mock_roles_list())
|
|
self.mock_admin_client.roles.grant(project='aproject',
|
|
role='4546',
|
|
user='duser123').AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.create_stack_domain_user(username='duser',
|
|
project_id='aproject')
|
|
|
|
def test_create_stack_domain_user_legacy_fallback(self):
|
|
"""Test creating a stack domain user, fallback path."""
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stubs_v3()
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
mock_user = self.m.CreateMockAnything()
|
|
mock_user.id = 'auser123'
|
|
self.mock_ks_v3_client.users.create(name='auser',
|
|
password='password',
|
|
default_project=ctx.tenant_id
|
|
).AndReturn(mock_user)
|
|
|
|
self.mock_ks_v3_client.roles = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.roles.list(
|
|
name='heat_stack_user').AndReturn(self._mock_roles_list())
|
|
self.mock_ks_v3_client.roles.grant(project=ctx.tenant_id,
|
|
role='4546',
|
|
user='auser123').AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.create_stack_domain_user(username='auser',
|
|
project_id='aproject',
|
|
password='password')
|
|
|
|
def test_create_stack_domain_user_error_norole(self):
|
|
"""Test creating a stack domain user, no role error path."""
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
self._stub_domain_admin_client()
|
|
|
|
# mock keystone client functions
|
|
self.mock_admin_client.roles = self.m.CreateMockAnything()
|
|
self.mock_admin_client.roles.list(name='heat_stack_user').AndReturn([])
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
err = self.assertRaises(exception.Error,
|
|
heat_ks_client.create_stack_domain_user,
|
|
username='duser', project_id='aproject')
|
|
self.assertIn('Can\'t find role heat_stack_user', six.text_type(err))
|
|
|
|
def test_delete_stack_domain_user(self):
|
|
"""Test deleting a stack domain user."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.users = self.m.CreateMockAnything()
|
|
mock_user = self.m.CreateMockAnything()
|
|
mock_user.id = 'duser123'
|
|
mock_user.domain_id = 'adomain123'
|
|
mock_user.default_project_id = 'aproject'
|
|
self.mock_admin_client.users.get('duser123').AndReturn(mock_user)
|
|
self.mock_admin_client.users.delete('duser123').AndReturn(None)
|
|
self.mock_admin_client.users.get('duser123').AndRaise(
|
|
kc_exception.NotFound)
|
|
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_user(user_id='duser123',
|
|
project_id='aproject')
|
|
# Second delete will raise ignored NotFound
|
|
heat_ks_client.delete_stack_domain_user(user_id='duser123',
|
|
project_id='aproject')
|
|
|
|
def test_delete_stack_domain_user_legacy_fallback(self):
|
|
"""Test deleting a stack domain user, fallback path."""
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stubs_v3()
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.users.delete(user='user123').AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_user(user_id='user123',
|
|
project_id='aproject')
|
|
|
|
def test_delete_stack_domain_user_error_domain(self):
|
|
"""Test deleting a stack domain user, wrong domain."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.users = self.m.CreateMockAnything()
|
|
mock_user = self.m.CreateMockAnything()
|
|
mock_user.id = 'duser123'
|
|
mock_user.domain_id = 'notadomain123'
|
|
mock_user.default_project_id = 'aproject'
|
|
self.mock_admin_client.users.get('duser123').AndReturn(mock_user)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
err = self.assertRaises(ValueError,
|
|
heat_ks_client.delete_stack_domain_user,
|
|
user_id='duser123', project_id='aproject')
|
|
self.assertIn('User delete in invalid domain', err)
|
|
|
|
def test_delete_stack_domain_user_error_project(self):
|
|
"""Test deleting a stack domain user, wrong project."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.users = self.m.CreateMockAnything()
|
|
mock_user = self.m.CreateMockAnything()
|
|
mock_user.id = 'duser123'
|
|
mock_user.domain_id = 'adomain123'
|
|
mock_user.default_project_id = 'notaproject'
|
|
self.mock_admin_client.users.get('duser123').AndReturn(mock_user)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
err = self.assertRaises(ValueError,
|
|
heat_ks_client.delete_stack_domain_user,
|
|
user_id='duser123', project_id='aproject')
|
|
self.assertIn('User delete in invalid project', err)
|
|
|
|
def test_delete_stack_user(self):
|
|
|
|
"""Test deleting a stack user."""
|
|
|
|
self._stubs_v3()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client delete function
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.users.delete(user='atestuser').AndReturn(None)
|
|
self.mock_ks_v3_client.users.delete(user='atestuser').AndRaise(
|
|
kc_exception.NotFound)
|
|
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_user('atestuser')
|
|
# Second delete will raise ignored NotFound
|
|
heat_ks_client.delete_stack_user('atestuser')
|
|
|
|
def test_init_v3_token(self):
|
|
|
|
"""Test creating the client, token auth."""
|
|
|
|
self._stubs_v3()
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.client
|
|
self.assertIsNotNone(heat_ks_client._client)
|
|
|
|
def test_init_v3_token_auth_ref_v2(self):
|
|
|
|
"""Test creating the client, token v2 auth_ref."""
|
|
|
|
expected_auth_ref = {'token': {'id': 'ctx_token', 'expires': '123'},
|
|
'version': 'v2.0'}
|
|
self._stubs_v3(method='auth_ref', auth_ref=expected_auth_ref)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = None
|
|
ctx.auth_token = 'ctx_token'
|
|
ctx.auth_token_info = {'access': {'token': {'expires': '123'}}}
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.client
|
|
self.assertIsNotNone(heat_ks_client._client)
|
|
|
|
def test_init_v3_token_auth_ref_v3(self):
|
|
|
|
"""Test creating the client, token v3 auth_ref."""
|
|
|
|
expected_auth_ref = {'auth_token': 'ctx_token',
|
|
'expires': '456', 'version': 'v3'}
|
|
self._stubs_v3(method='auth_ref', auth_ref=expected_auth_ref)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = None
|
|
ctx.auth_token = 'ctx_token'
|
|
ctx.auth_token_info = {'token': {'expires': '456'}}
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.client
|
|
self.assertIsNotNone(heat_ks_client._client)
|
|
|
|
def test_init_v3_password(self):
|
|
|
|
"""Test creating the client, password auth."""
|
|
|
|
self._stubs_v3(method='password')
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.auth_token = None
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
client = heat_ks_client.client
|
|
self.assertIsNotNone(client)
|
|
self.assertIsNone(ctx.trust_id)
|
|
|
|
def test_init_v3_bad_nocreds(self):
|
|
|
|
"""Test creating the client, no credentials."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.auth_token = None
|
|
ctx.trust_id = None
|
|
ctx.username = None
|
|
ctx.password = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(exception.AuthorizationFailure,
|
|
heat_ks_client._v3_client_init)
|
|
|
|
def test_create_trust_context_trust_id(self):
|
|
|
|
"""Test create_trust_context with existing trust_id."""
|
|
|
|
self._stubs_v3(method='trust')
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = 'atrust123'
|
|
ctx.trustor_user_id = 'trustor_user_id'
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
trust_context = heat_ks_client.create_trust_context()
|
|
self.assertEqual(ctx.to_dict(), trust_context.to_dict())
|
|
|
|
def test_create_trust_context_trust_create_deletegate_subset_roles(self):
|
|
delegate_roles = ['heat_stack_owner']
|
|
self._test_create_trust_context_trust_create(delegate_roles)
|
|
|
|
def test_create_trust_context_trust_create_deletegate_all_roles(self):
|
|
self._test_create_trust_context_trust_create()
|
|
|
|
def _test_create_trust_context_trust_create(self, delegate_roles=None):
|
|
|
|
"""Test create_trust_context when creating a trust."""
|
|
|
|
class MockTrust(object):
|
|
id = 'atrust123'
|
|
|
|
self._stub_admin_client()
|
|
|
|
self._stubs_v3()
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
if delegate_roles:
|
|
cfg.CONF.set_override('trusts_delegated_roles', delegate_roles)
|
|
|
|
trustor_roles = ['heat_stack_owner', 'admin', '__member__']
|
|
trustee_roles = delegate_roles or trustor_roles
|
|
self.mock_ks_v3_client.auth_ref = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.auth_ref.user_id = '5678'
|
|
self.mock_ks_v3_client.auth_ref.project_id = '42'
|
|
self.mock_ks_v3_client.trusts = self.m.CreateMockAnything()
|
|
|
|
self.mock_ks_v3_client.trusts.create(
|
|
trustor_user='5678',
|
|
trustee_user='1234',
|
|
project='42',
|
|
impersonation=True,
|
|
role_names=trustee_roles).AndReturn(MockTrust())
|
|
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context(roles=trustor_roles)
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
trust_context = heat_ks_client.create_trust_context()
|
|
self.assertEqual('atrust123', trust_context.trust_id)
|
|
self.assertEqual('5678', trust_context.trustor_user_id)
|
|
|
|
def test_create_trust_context_trust_create_norole(self):
|
|
|
|
"""Test create_trust_context when creating a trust."""
|
|
|
|
class MockTrust(object):
|
|
id = 'atrust123'
|
|
|
|
self._stub_admin_client()
|
|
|
|
self._stubs_v3()
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
cfg.CONF.set_override('trusts_delegated_roles', ['heat_stack_owner'])
|
|
|
|
self.mock_ks_v3_client.auth_ref = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.auth_ref.user_id = '5678'
|
|
self.mock_ks_v3_client.auth_ref.project_id = '42'
|
|
self.mock_ks_v3_client.trusts = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.trusts.create(
|
|
trustor_user='5678',
|
|
trustee_user='1234',
|
|
project='42',
|
|
impersonation=True,
|
|
role_names=['heat_stack_owner']).AndRaise(kc_exception.NotFound())
|
|
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
exc = self.assertRaises(exception.MissingCredentialError,
|
|
heat_ks_client.create_trust_context)
|
|
expected = 'Missing required credential: roles [\'heat_stack_owner\']'
|
|
self.assertIn(expected, six.text_type(exc))
|
|
|
|
def test_init_domain_cfg_not_set_fallback(self):
|
|
|
|
"""Test error path when config lacks domain config."""
|
|
|
|
self._clear_domain_override()
|
|
cfg.CONF.clear_override('stack_domain_admin')
|
|
cfg.CONF.clear_override('stack_domain_admin_password')
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = None
|
|
self.assertIsNotNone(heat_keystoneclient.KeystoneClient(ctx))
|
|
|
|
def test_init_domain_cfg_not_set_error(self):
|
|
|
|
"""Test error path when config lacks domain config."""
|
|
|
|
cfg.CONF.clear_override('stack_domain_admin')
|
|
cfg.CONF.clear_override('stack_domain_admin_password')
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = None
|
|
err = self.assertRaises(exception.Error,
|
|
heat_keystoneclient.KeystoneClient, ctx)
|
|
exp_msg = ('heat.conf misconfigured, cannot specify '
|
|
'"stack_user_domain_id" or "stack_user_domain_name" '
|
|
'without "stack_domain_admin" and '
|
|
'"stack_domain_admin_password"')
|
|
self.assertIn(exp_msg, six.text_type(err))
|
|
|
|
def test_init_admin_client(self):
|
|
|
|
"""Test the admin_client property."""
|
|
|
|
self._stub_admin_client()
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertEqual(self.mock_admin_client, heat_ks_client.admin_client)
|
|
self.assertEqual(self.mock_admin_client, heat_ks_client._admin_client)
|
|
|
|
def test_init_admin_client_denied(self):
|
|
|
|
"""Test the admin_client property, auth failure path."""
|
|
|
|
self._stub_admin_client(auth_ok=False)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
|
|
# Define wrapper for property or the property raises the exception
|
|
# outside of the assertRaises which fails the test
|
|
def get_admin_client():
|
|
heat_ks_client.admin_client
|
|
|
|
self.assertRaises(exception.AuthorizationFailure,
|
|
get_admin_client)
|
|
|
|
def test_trust_init(self):
|
|
|
|
"""Test consuming a trust when initializing."""
|
|
|
|
self._stubs_v3(method='trust')
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.auth_token = None
|
|
ctx.trust_id = 'atrust123'
|
|
ctx.trustor_user_id = 'trustor_user_id'
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNotNone(heat_ks_client.client)
|
|
self.assertIsNone(ctx.auth_token)
|
|
|
|
def test_trust_init_fail(self):
|
|
|
|
"""Test consuming a trust when initializing, error scoping."""
|
|
|
|
self._stubs_v3(method='trust', trust_scoped=False)
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.auth_token = None
|
|
ctx.trust_id = 'atrust123'
|
|
ctx.trustor_user_id = 'trustor_user_id'
|
|
self.assertRaises(exception.AuthorizationFailure,
|
|
heat_keystoneclient.KeystoneClient, ctx)
|
|
|
|
def test_trust_init_fail_impersonation(self):
|
|
|
|
"""Test consuming a trust when initializing, impersonation error."""
|
|
|
|
self._stubs_v3(method='trust', user_id='wrong_user_id')
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = 'heat'
|
|
ctx.password = None
|
|
ctx.auth_token = None
|
|
ctx.trust_id = 'atrust123'
|
|
ctx.trustor_user_id = 'trustor_user_id'
|
|
self.assertRaises(exception.AuthorizationFailure,
|
|
heat_keystoneclient.KeystoneClient, ctx)
|
|
|
|
def test_trust_init_pw(self):
|
|
|
|
"""Test trust_id is takes precedence username/password specified."""
|
|
|
|
self._stubs_v3(method='trust')
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.auth_token = None
|
|
ctx.trust_id = 'atrust123'
|
|
ctx.trustor_user_id = 'trustor_user_id'
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNotNone(heat_ks_client._client)
|
|
|
|
def test_trust_init_token(self):
|
|
|
|
"""Test trust_id takes precedence when token specified."""
|
|
|
|
self._stubs_v3(method='trust')
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.username = None
|
|
ctx.password = None
|
|
ctx.trust_id = 'atrust123'
|
|
ctx.trustor_user_id = 'trustor_user_id'
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNotNone(heat_ks_client._client)
|
|
|
|
def test_delete_trust(self):
|
|
|
|
"""Test delete_trust when deleting trust."""
|
|
|
|
self._stubs_v3()
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
self.mock_ks_v3_client.trusts = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.trusts.delete('atrust123').AndReturn(None)
|
|
|
|
self.m.ReplayAll()
|
|
ctx = utils.dummy_context()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNone(heat_ks_client.delete_trust(trust_id='atrust123'))
|
|
|
|
def test_delete_trust_not_found(self):
|
|
|
|
"""Test delete_trust when trust already deleted."""
|
|
|
|
self._stubs_v3()
|
|
cfg.CONF.set_override('deferred_auth_method', 'trusts')
|
|
self.mock_ks_v3_client.trusts = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.trusts.delete('atrust123').AndRaise(
|
|
kc_exception.NotFound)
|
|
|
|
self.m.ReplayAll()
|
|
ctx = utils.dummy_context()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNone(heat_ks_client.delete_trust(trust_id='atrust123'))
|
|
|
|
def test_disable_stack_user(self):
|
|
|
|
"""Test disabling a stack user."""
|
|
|
|
self._stubs_v3()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client update function
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.users.update(user='atestuser', enabled=False
|
|
).AndReturn(None)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.disable_stack_user('atestuser')
|
|
|
|
def test_enable_stack_user(self):
|
|
|
|
"""Test enabling a stack user."""
|
|
|
|
self._stubs_v3()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client update function
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.users.update(user='atestuser', enabled=True
|
|
).AndReturn(None)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.enable_stack_user('atestuser')
|
|
|
|
def _stub_admin_user_get(self, user_id, domain_id, project_id):
|
|
self.mock_admin_client.users = self.m.CreateMockAnything()
|
|
mock_user = self.m.CreateMockAnything()
|
|
mock_user.id = user_id
|
|
mock_user.domain_id = domain_id
|
|
mock_user.default_project_id = project_id
|
|
self.mock_admin_client.users.get(user_id).AndReturn(mock_user)
|
|
return mock_user
|
|
|
|
def test_enable_stack_domain_user(self):
|
|
"""Test enabling a stack domain user."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'adomain123', 'aproject')
|
|
self.mock_admin_client.users.update(user='duser123', enabled=True
|
|
).AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.enable_stack_domain_user(user_id='duser123',
|
|
project_id='aproject')
|
|
|
|
def test_enable_stack_domain_user_legacy_fallback(self):
|
|
"""Test enabling a stack domain user, fallback path."""
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stubs_v3()
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.users.update(user='user123', enabled=True
|
|
).AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.enable_stack_domain_user(user_id='user123',
|
|
project_id='aproject')
|
|
|
|
def test_enable_stack_domain_user_error_project(self):
|
|
"""Test enabling a stack domain user, wrong project."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'adomain123', 'notaproject')
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError, heat_ks_client.enable_stack_domain_user,
|
|
user_id='duser123', project_id='aproject')
|
|
|
|
def test_enable_stack_domain_user_error_domain(self):
|
|
"""Test enabling a stack domain user, wrong domain."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'notadomain123', 'aproject')
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError, heat_ks_client.enable_stack_domain_user,
|
|
user_id='duser123', project_id='aproject')
|
|
|
|
def test_disable_stack_domain_user(self):
|
|
"""Test disabling a stack domain user."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'adomain123', 'aproject')
|
|
self.mock_admin_client.users.update(user='duser123', enabled=False
|
|
).AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.disable_stack_domain_user(user_id='duser123',
|
|
project_id='aproject')
|
|
|
|
def test_disable_stack_domain_user_legacy_fallback(self):
|
|
"""Test enabling a stack domain user, fallback path."""
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stubs_v3()
|
|
self.mock_ks_v3_client.users = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.users.update(user='user123', enabled=False
|
|
).AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.disable_stack_domain_user(user_id='user123',
|
|
project_id='aproject')
|
|
|
|
def test_disable_stack_domain_user_error_project(self):
|
|
"""Test disabling a stack domain user, wrong project."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'adomain123', 'notaproject')
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError, heat_ks_client.disable_stack_domain_user,
|
|
user_id='duser123', project_id='aproject')
|
|
|
|
def test_disable_stack_domain_user_error_domain(self):
|
|
"""Test disabling a stack domain user, wrong domain."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'notadomain123', 'aproject')
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError, heat_ks_client.disable_stack_domain_user,
|
|
user_id='duser123', project_id='aproject')
|
|
|
|
def test_delete_stack_domain_user_keypair(self):
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
user = self._stub_admin_user_get('duser123', 'adomain123', 'aproject')
|
|
self.mock_admin_client.credentials = self.m.CreateMockAnything()
|
|
self.mock_admin_client.credentials.delete(
|
|
'acredentialid').AndReturn(None)
|
|
|
|
self.mock_admin_client.users.get('duser123').AndReturn(user)
|
|
self.mock_admin_client.credentials.delete(
|
|
'acredentialid').AndRaise(kc_exception.NotFound)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_user_keypair(
|
|
user_id='duser123', project_id='aproject',
|
|
credential_id='acredentialid')
|
|
# Second delete will raise ignored NotFound
|
|
heat_ks_client.delete_stack_domain_user_keypair(
|
|
user_id='duser123', project_id='aproject',
|
|
credential_id='acredentialid')
|
|
|
|
def test_delete_stack_domain_user_keypair_legacy_fallback(self):
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stubs_v3()
|
|
self.mock_ks_v3_client.credentials = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.credentials.delete(
|
|
'acredentialid').AndReturn(None)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_user_keypair(
|
|
user_id='user123', project_id='aproject',
|
|
credential_id='acredentialid')
|
|
|
|
def test_delete_stack_domain_user_keypair_error_project(self):
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'adomain123', 'notaproject')
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError,
|
|
heat_ks_client.delete_stack_domain_user_keypair,
|
|
user_id='duser123', project_id='aproject',
|
|
credential_id='acredentialid')
|
|
|
|
def test_delete_stack_domain_user_keypair_error_domain(self):
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client functions
|
|
self._stub_domain_admin_client()
|
|
self._stub_admin_user_get('duser123', 'notadomain123', 'aproject')
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError,
|
|
heat_ks_client.delete_stack_domain_user_keypair,
|
|
user_id='duser123', project_id='aproject',
|
|
credential_id='acredentialid')
|
|
|
|
def _stub_uuid(self, values=None):
|
|
# stub UUID.hex to return the values specified
|
|
values = values or []
|
|
self.m.StubOutWithMock(uuid, 'uuid4')
|
|
for v in values:
|
|
mock_uuid = self.m.CreateMockAnything()
|
|
mock_uuid.hex = v
|
|
uuid.uuid4().AndReturn(mock_uuid)
|
|
|
|
def test_create_ec2_keypair(self):
|
|
|
|
"""Test creating ec2 credentials."""
|
|
|
|
self._stubs_v3()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
ex_data = {'access': 'dummy_access',
|
|
'secret': 'dummy_secret'}
|
|
ex_data_json = json.dumps(ex_data)
|
|
|
|
# stub UUID.hex to match ex_data
|
|
self._stub_uuid(['dummy_access', 'dummy_secret'])
|
|
|
|
# mock keystone client credentials functions
|
|
self.mock_ks_v3_client.credentials = self.m.CreateMockAnything()
|
|
mock_credential = self.m.CreateMockAnything()
|
|
mock_credential.id = '123456'
|
|
mock_credential.user_id = 'atestuser'
|
|
mock_credential.blob = ex_data_json
|
|
mock_credential.type = 'ec2'
|
|
|
|
# mock keystone client create function
|
|
self.mock_ks_v3_client.credentials.create(
|
|
user='atestuser', type='ec2', data=ex_data_json,
|
|
project=ctx.tenant_id).AndReturn(mock_credential)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
ec2_cred = heat_ks_client.create_ec2_keypair(user_id='atestuser')
|
|
self.assertEqual('123456', ec2_cred.id)
|
|
self.assertEqual('dummy_access', ec2_cred.access)
|
|
self.assertEqual('dummy_secret', ec2_cred.secret)
|
|
|
|
def test_create_stack_domain_user_keypair(self):
|
|
|
|
"""Test creating ec2 credentials for domain user."""
|
|
|
|
self._stub_domain_admin_client()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
ex_data = {'access': 'dummy_access2',
|
|
'secret': 'dummy_secret2'}
|
|
ex_data_json = json.dumps(ex_data)
|
|
|
|
# stub UUID.hex to match ex_data
|
|
self._stub_uuid(['dummy_access2', 'dummy_secret2'])
|
|
|
|
# mock keystone client credentials functions
|
|
self.mock_admin_client.credentials = self.m.CreateMockAnything()
|
|
mock_credential = self.m.CreateMockAnything()
|
|
mock_credential.id = '1234567'
|
|
mock_credential.user_id = 'atestuser2'
|
|
mock_credential.blob = ex_data_json
|
|
mock_credential.type = 'ec2'
|
|
|
|
# mock keystone client create function
|
|
self.mock_admin_client.credentials.create(
|
|
user='atestuser2', type='ec2', data=ex_data_json,
|
|
project='aproject').AndReturn(mock_credential)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
ec2_cred = heat_ks_client.create_stack_domain_user_keypair(
|
|
user_id='atestuser2', project_id='aproject')
|
|
self.assertEqual('1234567', ec2_cred.id)
|
|
self.assertEqual('dummy_access2', ec2_cred.access)
|
|
self.assertEqual('dummy_secret2', ec2_cred.secret)
|
|
|
|
def test_create_stack_domain_user_keypair_legacy_fallback(self):
|
|
|
|
"""Test creating ec2 credentials for domain user, fallback path."""
|
|
self._clear_domain_override()
|
|
|
|
self._stubs_v3()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
ex_data = {'access': 'dummy_access2',
|
|
'secret': 'dummy_secret2'}
|
|
ex_data_json = json.dumps(ex_data)
|
|
|
|
# stub UUID.hex to match ex_data
|
|
self._stub_uuid(['dummy_access2', 'dummy_secret2'])
|
|
|
|
# mock keystone client credentials functions
|
|
self.mock_ks_v3_client.credentials = self.m.CreateMockAnything()
|
|
mock_credential = self.m.CreateMockAnything()
|
|
mock_credential.id = '1234567'
|
|
mock_credential.user_id = 'atestuser2'
|
|
mock_credential.blob = ex_data_json
|
|
mock_credential.type = 'ec2'
|
|
|
|
# mock keystone client create function
|
|
self.mock_ks_v3_client.credentials.create(
|
|
user='atestuser2', type='ec2', data=ex_data_json,
|
|
project=ctx.tenant_id).AndReturn(mock_credential)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
ec2_cred = heat_ks_client.create_stack_domain_user_keypair(
|
|
user_id='atestuser2', project_id='aproject')
|
|
self.assertEqual('1234567', ec2_cred.id)
|
|
self.assertEqual('dummy_access2', ec2_cred.access)
|
|
self.assertEqual('dummy_secret2', ec2_cred.secret)
|
|
|
|
def test_get_ec2_keypair_id(self):
|
|
|
|
"""Test getting ec2 credential by id."""
|
|
|
|
user_id = 'atestuser'
|
|
self._stubs_v3(user_id=user_id)
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
ex_data = {'access': 'access123',
|
|
'secret': 'secret456'}
|
|
ex_data_json = json.dumps(ex_data)
|
|
|
|
# Create a mock credential response
|
|
credential_id = 'acredential123'
|
|
mock_credential = self.m.CreateMockAnything()
|
|
mock_credential.id = credential_id
|
|
mock_credential.user_id = user_id
|
|
mock_credential.blob = ex_data_json
|
|
mock_credential.type = 'ec2'
|
|
|
|
# mock keystone client get function
|
|
self.mock_ks_v3_client.credentials = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.credentials.get(
|
|
credential_id).AndReturn(mock_credential)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
ec2_cred = heat_ks_client.get_ec2_keypair(credential_id=credential_id)
|
|
self.assertEqual(credential_id, ec2_cred.id)
|
|
self.assertEqual('access123', ec2_cred.access)
|
|
self.assertEqual('secret456', ec2_cred.secret)
|
|
|
|
def _mock_credential_list(self, user_id):
|
|
"""Create a mock credential list response."""
|
|
mock_credential_list = []
|
|
for x in (1, 2, 3):
|
|
mock_credential = self.m.CreateMockAnything()
|
|
mock_credential.id = 'credential_id%s' % x
|
|
mock_credential.user_id = user_id
|
|
mock_credential.blob = json.dumps({'access': 'access%s' % x,
|
|
'secret': 'secret%s' % x})
|
|
mock_credential.type = 'ec2'
|
|
mock_credential_list.append(mock_credential)
|
|
|
|
# mock keystone client list function
|
|
self.mock_ks_v3_client.credentials = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.credentials.list().AndReturn(
|
|
mock_credential_list)
|
|
|
|
def test_get_ec2_keypair_access(self):
|
|
|
|
"""Test getting ec2 credential by access."""
|
|
|
|
user_id = 'atestuser'
|
|
self._stubs_v3(user_id=user_id)
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
self._mock_credential_list(user_id=user_id)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
ec2_cred = heat_ks_client.get_ec2_keypair(access='access2')
|
|
self.assertEqual('credential_id2', ec2_cred.id)
|
|
self.assertEqual('access2', ec2_cred.access)
|
|
self.assertEqual('secret2', ec2_cred.secret)
|
|
|
|
def test_get_ec2_keypair_error(self):
|
|
|
|
"""Test getting ec2 credential error path."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError, heat_ks_client.get_ec2_keypair)
|
|
|
|
def test_delete_ec2_keypair_id(self):
|
|
|
|
"""Test deleting ec2 credential by id."""
|
|
|
|
user_id = 'atestuser'
|
|
self._stubs_v3(user_id=user_id)
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
# mock keystone client credentials functions
|
|
credential_id = 'acredential123'
|
|
self.mock_ks_v3_client.credentials = self.m.CreateMockAnything()
|
|
|
|
# mock keystone client delete function
|
|
self.mock_ks_v3_client.credentials = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.credentials.delete(credential_id)
|
|
self.mock_ks_v3_client.credentials.delete(credential_id).AndRaise(
|
|
kc_exception.NotFound)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNone(heat_ks_client.delete_ec2_keypair(
|
|
credential_id=credential_id))
|
|
# Second delete will raise ignored NotFound
|
|
self.assertIsNone(heat_ks_client.delete_ec2_keypair(
|
|
credential_id=credential_id))
|
|
|
|
def test_delete_ec2_keypair_access(self):
|
|
|
|
"""Test deleting ec2 credential by access."""
|
|
|
|
user_id = 'atestuser'
|
|
self._stubs_v3(user_id=user_id)
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
self._mock_credential_list(user_id=user_id)
|
|
|
|
# mock keystone client delete function
|
|
self.mock_ks_v3_client.credentials.delete(
|
|
'credential_id2').AndReturn(None)
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNone(heat_ks_client.delete_ec2_keypair(access='access2'))
|
|
|
|
def test_deleting_ec2_keypair_error(self):
|
|
|
|
"""Test deleting ec2 credential error path."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
self.m.ReplayAll()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(ValueError, heat_ks_client.delete_ec2_keypair)
|
|
|
|
def test_create_stack_domain_project(self):
|
|
|
|
"""Test the create_stack_domain_project function."""
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
expected_name = '%s-astack' % ctx.tenant_id
|
|
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.projects = self.m.CreateMockAnything()
|
|
dummy = self.m.CreateMockAnything()
|
|
dummy.id = 'aproject123'
|
|
self.mock_admin_client.projects.create(
|
|
name=expected_name,
|
|
domain='adomain123',
|
|
description='Heat stack user project').AndReturn(dummy)
|
|
self.m.ReplayAll()
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertEqual('aproject123',
|
|
heat_ks_client.create_stack_domain_project('astack'))
|
|
|
|
def test_create_stack_domain_project_legacy_fallback(self):
|
|
"""Test the create_stack_domain_project function, fallback path."""
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertEqual(ctx.tenant_id,
|
|
heat_ks_client.create_stack_domain_project('astack'))
|
|
|
|
def test_delete_stack_domain_project(self):
|
|
|
|
"""Test the delete_stack_domain_project function."""
|
|
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.projects = self.m.CreateMockAnything()
|
|
dummy = self.m.CreateMockAnything()
|
|
dummy.id = 'aproject123'
|
|
dummy.domain_id = 'adomain123'
|
|
dummy.delete().AndReturn(None)
|
|
self.mock_admin_client.projects.get(project='aprojectid').AndReturn(
|
|
dummy)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_project(project_id='aprojectid')
|
|
|
|
def test_delete_stack_domain_project_notfound(self):
|
|
|
|
"""Test the delete_stack_domain_project function."""
|
|
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.projects = self.m.CreateMockAnything()
|
|
dummy = self.m.CreateMockAnything()
|
|
dummy.id = 'aproject123'
|
|
dummy.domain_id = 'adomain123'
|
|
dummy.delete().AndRaise(kc_exception.NotFound)
|
|
self.mock_admin_client.projects.get(project='aprojectid').AndReturn(
|
|
dummy)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_project(project_id='aprojectid')
|
|
|
|
def test_delete_stack_domain_project_forbidden(self):
|
|
|
|
"""Test the delete_stack_domain_project function."""
|
|
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.projects = self.m.CreateMockAnything()
|
|
self.mock_admin_client.projects.get(project='aprojectid').AndRaise(
|
|
kc_exception.Forbidden)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_project(project_id='aprojectid')
|
|
|
|
def test_delete_stack_domain_project_wrongdomain(self):
|
|
|
|
"""Test the delete_stack_domain_project function."""
|
|
|
|
self._stub_domain_admin_client()
|
|
self.mock_admin_client.projects = self.m.CreateMockAnything()
|
|
dummy = self.m.CreateMockAnything()
|
|
dummy.id = 'aproject123'
|
|
dummy.domain_id = 'default'
|
|
self.mock_admin_client.projects.get(project='aprojectid').AndReturn(
|
|
dummy)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_project(project_id='aprojectid')
|
|
|
|
def test_delete_stack_domain_project_nodomain(self):
|
|
|
|
"""Test the delete_stack_domain_project function."""
|
|
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
heat_ks_client.delete_stack_domain_project(project_id='aprojectid')
|
|
|
|
def _stub_domain_user_pw_auth(self):
|
|
self.m.StubOutWithMock(ks_auth_v3, 'Password')
|
|
ks_auth_v3.Password(auth_url='http://server.test:5000/v3',
|
|
username='duser',
|
|
password='apassw',
|
|
project_id='aproject',
|
|
user_domain_id='adomain123').AndReturn('dummyauth')
|
|
|
|
def test_stack_domain_user_token(self):
|
|
"""Test stack_domain_user_token function."""
|
|
self._stub_domain_user_pw_auth()
|
|
dummysession = self.m.CreateMockAnything()
|
|
dummyresp = self.m.CreateMockAnything()
|
|
dummyresp.headers = {'X-Subject-Token': 'dummytoken'}
|
|
dummysession.post('http://server.test:5000/v3/auth/tokens?nocatalog',
|
|
authenticated=False,
|
|
headers={'Accept': 'application/json'},
|
|
json=mox.IgnoreArg()).AndReturn(dummyresp)
|
|
self.m.StubOutWithMock(ks_session, 'Session')
|
|
ks_session.Session(auth='dummyauth').AndReturn(dummysession)
|
|
self.m.ReplayAll()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
token = heat_ks_client.stack_domain_user_token(
|
|
username='duser', project_id='aproject', password='apassw')
|
|
self.assertEqual('dummytoken', token)
|
|
|
|
def test_stack_domain_user_token_err_nodomain(self):
|
|
"""Test stack_domain_user_token error path."""
|
|
self._clear_domain_override()
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertRaises(exception.Error,
|
|
heat_ks_client.stack_domain_user_token,
|
|
username='user',
|
|
project_id='aproject',
|
|
password='password')
|
|
|
|
def test_delete_stack_domain_project_legacy_fallback(self):
|
|
"""Test the delete_stack_domain_project function, fallback path."""
|
|
self._clear_domain_override()
|
|
|
|
ctx = utils.dummy_context()
|
|
ctx.trust_id = None
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertIsNone(heat_ks_client.delete_stack_domain_project(
|
|
project_id='aprojectid'))
|
|
|
|
def _test_url_for(self, service_url, expected_kwargs, ctx=None, **kwargs):
|
|
"""
|
|
Helper function for testing url_for depending on different ways to
|
|
pass region name.
|
|
"""
|
|
self._stubs_v3()
|
|
self.mock_ks_v3_client.service_catalog = self.m.CreateMockAnything()
|
|
self.mock_ks_v3_client.service_catalog.url_for(**expected_kwargs)\
|
|
.AndReturn(service_url)
|
|
|
|
self.m.ReplayAll()
|
|
ctx = ctx or utils.dummy_context()
|
|
heat_ks_client = heat_keystoneclient.KeystoneClient(ctx)
|
|
self.assertEqual(service_url, heat_ks_client.url_for(**kwargs))
|
|
self.m.VerifyAll()
|
|
|
|
def test_url_for(self):
|
|
"""
|
|
Test that None value is passed as region name if region name is not
|
|
specified in the config file or as one of the arguments.
|
|
"""
|
|
cfg.CONF.set_override('region_name_for_services', None)
|
|
service_url = 'http://example.com:1234/v1'
|
|
kwargs = {
|
|
'region_name': None
|
|
}
|
|
self._test_url_for(service_url, kwargs)
|
|
|
|
def test_url_for_with_region(self):
|
|
"""
|
|
Test that region name passed as argument is not override by region name
|
|
specified in the config file.
|
|
"""
|
|
cfg.CONF.set_override('region_name_for_services', 'RegionOne')
|
|
service_url = 'http://regiontwo.example.com:1234/v1'
|
|
kwargs = {
|
|
'region_name': 'RegionTwo'
|
|
}
|
|
self._test_url_for(service_url, kwargs, None, **kwargs)
|
|
|
|
def test_url_for_with_region_name_from_config(self):
|
|
"""
|
|
Test that default region name for services from config file is passed
|
|
if region name is not specified in arguments.
|
|
"""
|
|
region_name_for_services = 'RegionOne'
|
|
cfg.CONF.set_override('region_name_for_services',
|
|
region_name_for_services)
|
|
kwargs = {
|
|
'region_name': region_name_for_services
|
|
}
|
|
service_url = 'http://regionone.example.com:1234/v1'
|
|
self._test_url_for(service_url, kwargs)
|
|
|
|
def test_url_for_with_region_name_from_context(self):
|
|
"""
|
|
Test that default region name for services from context is passed
|
|
if region name is not specified in arguments.
|
|
"""
|
|
cfg.CONF.set_override('region_name_for_services', 'RegionOne')
|
|
service_url = 'http://regiontwo.example.com:1234/v1'
|
|
region_name_for_services = 'RegionTwo'
|
|
expected_kwargs = {
|
|
'region_name': region_name_for_services
|
|
}
|
|
ctx = utils.dummy_context('test_username',
|
|
'test_tenant_id',
|
|
'password',
|
|
None,
|
|
None,
|
|
None,
|
|
region_name_for_services)
|
|
self._test_url_for(service_url, expected_kwargs, ctx)
|
|
|
|
|
|
class KeystoneClientTestDomainName(KeystoneClientTest):
|
|
def setUp(self):
|
|
cfg.CONF.set_override('stack_user_domain_name', 'fake_domain_name')
|
|
super(KeystoneClientTestDomainName, self).setUp()
|
|
cfg.CONF.clear_override('stack_user_domain_id')
|
|
|
|
def _clear_domain_override(self):
|
|
cfg.CONF.clear_override('stack_user_domain_name')
|
|
|
|
def _stub_domain_admin_client_domain_get(self):
|
|
dummy_domain = self.m.CreateMockAnything()
|
|
dummy_domain.id = 'adomain123'
|
|
self.mock_ks_v3_client_domain_mngr.list(
|
|
name='fake_domain_name').AndReturn([dummy_domain])
|
|
|
|
def _stub_domain_admin_client(self, auth_ok=True):
|
|
kc_v3.Client(
|
|
auth_url='http://server.test:5000/v3',
|
|
cacert=None,
|
|
cert=None,
|
|
endpoint='http://server.test:5000/v3',
|
|
insecure=False,
|
|
key=None,
|
|
password='adminsecret',
|
|
user_domain_name='fake_domain_name',
|
|
username='adminuser123').AndReturn(self.mock_admin_client)
|
|
self.mock_admin_client.domains = self.mock_ks_v3_client_domain_mngr
|
|
self.mock_admin_client.authenticate(
|
|
domain_name='fake_domain_name').AndReturn(auth_ok)
|
|
if auth_ok:
|
|
self.mock_admin_client.auth_ref = self.m.CreateMockAnything()
|
|
self.mock_admin_client.auth_ref.user_id = '1234'
|
|
|
|
def _stub_domain_user_pw_auth(self):
|
|
self.m.StubOutWithMock(ks_auth_v3, 'Password')
|
|
ks_auth_v3.Password(auth_url='http://server.test:5000/v3',
|
|
username='duser',
|
|
password='apassw',
|
|
project_id='aproject',
|
|
user_domain_name='fake_domain_name'
|
|
).AndReturn('dummyauth')
|
|
|
|
def test_enable_stack_domain_user_error_project(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_enable_stack_domain_user_error_project()
|
|
|
|
def test_delete_stack_domain_user_keypair(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_user_keypair()
|
|
|
|
def test_delete_stack_domain_user_error_project(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_user_error_project()
|
|
|
|
def test_delete_stack_domain_user_keypair_error_project(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_user_keypair_error_project()
|
|
|
|
def test_delete_stack_domain_user(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_user()
|
|
|
|
def test_enable_stack_domain_user(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_enable_stack_domain_user()
|
|
|
|
def test_delete_stack_domain_user_error_domain(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_user_error_domain()
|
|
|
|
def test_disable_stack_domain_user_error_project(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_disable_stack_domain_user_error_project()
|
|
|
|
def test_enable_stack_domain_user_error_domain(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_enable_stack_domain_user_error_domain()
|
|
|
|
def test_delete_stack_domain_user_keypair_error_domain(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_user_keypair_error_domain()
|
|
|
|
def test_disable_stack_domain_user(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_disable_stack_domain_user()
|
|
|
|
def test_disable_stack_domain_user_error_domain(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_disable_stack_domain_user_error_domain()
|
|
|
|
def test_delete_stack_domain_project(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_project()
|
|
|
|
def test_delete_stack_domain_project_notfound(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_project_notfound()
|
|
|
|
def test_delete_stack_domain_project_wrongdomain(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_delete_stack_domain_project_wrongdomain()
|
|
|
|
def test_create_stack_domain_project(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_create_stack_domain_project()
|
|
|
|
def test_create_stack_domain_user(self):
|
|
self._stub_domain_admin_client_domain_get()
|
|
p = super(KeystoneClientTestDomainName, self)
|
|
p.test_create_stack_domain_user()
|