137 lines
5.6 KiB
Python
137 lines
5.6 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.
|
|
|
|
import mock
|
|
|
|
from cinder import context
|
|
from cinder import test
|
|
|
|
|
|
class ContextTestCase(test.TestCase):
|
|
|
|
def test_request_context_sets_is_admin(self):
|
|
ctxt = context.RequestContext('111',
|
|
'222',
|
|
roles=['admin', 'weasel'])
|
|
self.assertEqual(ctxt.is_admin, True)
|
|
|
|
def test_request_context_sets_is_admin_upcase(self):
|
|
ctxt = context.RequestContext('111',
|
|
'222',
|
|
roles=['Admin', 'weasel'])
|
|
self.assertEqual(ctxt.is_admin, True)
|
|
|
|
def test_request_context_read_deleted(self):
|
|
ctxt = context.RequestContext('111',
|
|
'222',
|
|
read_deleted='yes')
|
|
self.assertEqual(ctxt.read_deleted, 'yes')
|
|
|
|
ctxt.read_deleted = 'no'
|
|
self.assertEqual(ctxt.read_deleted, 'no')
|
|
|
|
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', 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.assertFalse('admin' in user_context.roles)
|
|
self.assertTrue('admin' in 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(len(ctxt.service_catalog), 3)
|
|
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(return_compute, compute_catalog)
|
|
self.assertEqual(return_object, object_catalog)
|
|
|
|
def test_user_identity(self):
|
|
ctx = context.RequestContext("user", "tenant",
|
|
domain="domain",
|
|
user_domain="user-domain",
|
|
project_domain="project-domain")
|
|
self.assertEqual('user tenant domain user-domain project-domain',
|
|
ctx.to_dict()["user_identity"])
|
|
|
|
@mock.patch('cinder.context.CONF')
|
|
def test_cinder_internal_context(self, mock_conf):
|
|
project_id = 'ec729e9946bc43c39ece6dfa7de70eea'
|
|
user_id = 'c466a48309794261b64a4f02cfcc3d64'
|
|
mock_conf.cinder_internal_tenant_project_id = project_id
|
|
mock_conf.cinder_internal_tenant_user_id = user_id
|
|
ctx = context.get_internal_tenant_context()
|
|
self.assertEqual(user_id, ctx.user_id)
|
|
self.assertEqual(project_id, ctx.project_id)
|
|
|
|
@mock.patch('cinder.context.CONF')
|
|
def test_cinder_internal_context_missing_user(self, mock_conf):
|
|
project_id = 'ec729e9946bc43c39ece6dfa7de70eea'
|
|
user_id = None
|
|
mock_conf.cinder_internal_tenant_project_id = project_id
|
|
mock_conf.cinder_internal_tenant_user_id = user_id
|
|
ctx = context.get_internal_tenant_context()
|
|
self.assertIsNone(ctx)
|
|
|
|
@mock.patch('cinder.context.CONF')
|
|
def test_cinder_internal_context_missing_project(self, mock_conf):
|
|
project_id = None
|
|
user_id = 'c466a48309794261b64a4f02cfcc3d64'
|
|
mock_conf.cinder_internal_tenant_project_id = project_id
|
|
mock_conf.cinder_internal_tenant_user_id = user_id
|
|
ctx = context.get_internal_tenant_context()
|
|
self.assertIsNone(ctx)
|
|
|
|
@mock.patch('cinder.context.CONF')
|
|
def test_cinder_internal_context_missing_all(self, mock_conf):
|
|
project_id = None
|
|
user_id = None
|
|
mock_conf.cinder_internal_tenant_project_id = project_id
|
|
mock_conf.cinder_internal_tenant_user_id = user_id
|
|
ctx = context.get_internal_tenant_context()
|
|
self.assertIsNone(ctx)
|