132 lines
5.2 KiB
Python
132 lines
5.2 KiB
Python
|
|
# Copyright 2011 OpenStack Foundation
|
|
#
|
|
# 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.
|
|
|
|
from unittest import mock
|
|
|
|
import ddt
|
|
|
|
from cinder import context
|
|
from cinder import test
|
|
|
|
|
|
@ddt.ddt
|
|
class ContextTestCase(test.TestCase):
|
|
|
|
def test_request_context_sets_is_admin(self):
|
|
ctxt = context.RequestContext('111',
|
|
'222',
|
|
roles=['admin', 'weasel'])
|
|
self.assertTrue(ctxt.is_admin)
|
|
|
|
def test_request_context_sets_is_admin_upcase(self):
|
|
ctxt = context.RequestContext('111',
|
|
'222',
|
|
roles=['Admin', 'weasel'])
|
|
self.assertTrue(ctxt.is_admin)
|
|
|
|
def test_request_context_read_deleted(self):
|
|
ctxt = context.RequestContext('111',
|
|
'222',
|
|
read_deleted='yes')
|
|
self.assertEqual('yes', ctxt.read_deleted)
|
|
|
|
ctxt.read_deleted = 'no'
|
|
self.assertEqual('no', ctxt.read_deleted)
|
|
|
|
def test_request_context_read_deleted_invalid(self):
|
|
self.assertRaises(ValueError,
|
|
context.RequestContext,
|
|
'111',
|
|
'222',
|
|
read_deleted=True)
|
|
|
|
ctxt = context.RequestContext('111', '222')
|
|
self.assertRaises(ValueError,
|
|
setattr,
|
|
ctxt,
|
|
'read_deleted',
|
|
True)
|
|
|
|
def test_request_context_elevated(self):
|
|
user_context = context.RequestContext(
|
|
'fake_user', 'fake_project', is_admin=False)
|
|
self.assertFalse(user_context.is_admin)
|
|
admin_context = user_context.elevated()
|
|
self.assertFalse(user_context.is_admin)
|
|
self.assertTrue(admin_context.is_admin)
|
|
self.assertNotIn('admin', user_context.roles)
|
|
self.assertIn('admin', admin_context.roles)
|
|
|
|
def test_service_catalog_nova_and_swift(self):
|
|
service_catalog = [
|
|
{u'type': u'compute', u'name': u'nova'},
|
|
{u'type': u's3', u'name': u's3'},
|
|
{u'type': u'image', u'name': u'glance'},
|
|
{u'type': u'volume', u'name': u'cinder'},
|
|
{u'type': u'ec2', u'name': u'ec2'},
|
|
{u'type': u'object-store', u'name': u'swift'},
|
|
{u'type': u'identity', u'name': u'keystone'},
|
|
{u'type': None, u'name': u'S_withtypeNone'},
|
|
{u'type': u'co', u'name': u'S_partofcompute'}]
|
|
|
|
compute_catalog = [{u'type': u'compute', u'name': u'nova'}]
|
|
object_catalog = [{u'name': u'swift', u'type': u'object-store'}]
|
|
ctxt = context.RequestContext('111', '222',
|
|
service_catalog=service_catalog)
|
|
self.assertEqual(4, len(ctxt.service_catalog))
|
|
return_compute = [v for v in ctxt.service_catalog if
|
|
v['type'] == u'compute']
|
|
return_object = [v for v in ctxt.service_catalog if
|
|
v['type'] == u'object-store']
|
|
self.assertEqual(compute_catalog, return_compute)
|
|
self.assertEqual(object_catalog, return_object)
|
|
|
|
def test_user_identity(self):
|
|
ctx = context.RequestContext("user", "tenant",
|
|
domain_id="domain",
|
|
user_domain_id="user-domain",
|
|
project_domain_id="project-domain")
|
|
self.assertEqual('user tenant domain user-domain project-domain',
|
|
ctx.to_dict()["user_identity"])
|
|
|
|
@ddt.data(('ec729e9946bc43c39ece6dfa7de70eea',
|
|
'c466a48309794261b64a4f02cfcc3d64'),
|
|
('ec729e9946bc43c39ece6dfa7de70eea', None),
|
|
(None, 'c466a48309794261b64a4f02cfcc3d64'),
|
|
(None, None))
|
|
@ddt.unpack
|
|
@mock.patch('cinder.context.CONF')
|
|
def test_cinder_internal_context(self, project_id, user_id, mock_conf):
|
|
mock_conf.cinder_internal_tenant_project_id = project_id
|
|
mock_conf.cinder_internal_tenant_user_id = user_id
|
|
ctx = context.get_internal_tenant_context()
|
|
if project_id is None or user_id is None:
|
|
self.assertIsNone(ctx)
|
|
else:
|
|
self.assertEqual(user_id, ctx.user_id)
|
|
self.assertEqual(project_id, ctx.project_id)
|
|
|
|
def test_request_context_no_roles(self):
|
|
ctxt = context.RequestContext('111',
|
|
'222')
|
|
self.assertEqual([], ctxt.roles)
|
|
|
|
def test_request_context_with_roles(self):
|
|
roles = ['alpha', 'beta']
|
|
ctxt = context.RequestContext('111',
|
|
'222',
|
|
roles=roles)
|
|
self.assertEqual(roles, ctxt.roles)
|