95 lines
4.0 KiB
Python
95 lines
4.0 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.
|
|
|
|
from karbor import context
|
|
from karbor.tests import base
|
|
|
|
|
|
class ContextTestCase(base.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(5, 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="domain",
|
|
user_domain="user-domain",
|
|
project_domain="project-domain")
|
|
self.assertEqual('user tenant domain user-domain project-domain',
|
|
ctx.to_dict()["user_identity"])
|