0c29ef7ffe
Here we add the context module from Neutron, although it is currently marked as private because it will undergo changes related to the ongoing enginefacade and Keystone V3 work. We also add some of the roles management methods from Neutron's policy module. These are required for the initial DB patches that follow this one. Partially implements: Blueprint neutron-lib Change-Id: I77f8a05d6e3167f3096ad637d124e47ac39a83df
184 lines
7.8 KiB
Python
184 lines
7.8 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 mock
|
|
|
|
from oslo_context import context as oslo_context
|
|
from testtools import matchers
|
|
|
|
from neutron_lib import _context
|
|
from neutron_lib.tests import _base
|
|
|
|
|
|
class TestNeutronContext(_base.BaseTestCase):
|
|
|
|
def test_neutron_context_create(self):
|
|
ctx = _context.Context('user_id', 'tenant_id')
|
|
self.assertEqual('user_id', ctx.user_id)
|
|
self.assertEqual('tenant_id', ctx.project_id)
|
|
self.assertEqual('tenant_id', ctx.tenant_id)
|
|
request_id = ctx.request_id
|
|
if isinstance(request_id, bytes):
|
|
request_id = request_id.decode('utf-8')
|
|
self.assertThat(request_id, matchers.StartsWith('req-'))
|
|
self.assertEqual('user_id', ctx.user)
|
|
self.assertEqual('tenant_id', ctx.tenant)
|
|
self.assertIsNone(ctx.user_name)
|
|
self.assertIsNone(ctx.tenant_name)
|
|
self.assertIsNone(ctx.auth_token)
|
|
|
|
def test_neutron_context_getter_setter(self):
|
|
ctx = _context.Context('Anakin', 'Skywalker')
|
|
self.assertEqual('Anakin', ctx.user_id)
|
|
self.assertEqual('Skywalker', ctx.tenant_id)
|
|
ctx.user_id = 'Darth'
|
|
ctx.tenant_id = 'Vader'
|
|
self.assertEqual('Darth', ctx.user_id)
|
|
self.assertEqual('Vader', ctx.tenant_id)
|
|
|
|
def test_neutron_context_create_with_name(self):
|
|
ctx = _context.Context('user_id', 'tenant_id',
|
|
tenant_name='tenant_name',
|
|
user_name='user_name')
|
|
# Check name is set
|
|
self.assertEqual('user_name', ctx.user_name)
|
|
self.assertEqual('tenant_name', ctx.tenant_name)
|
|
# Check user/tenant contains its ID even if user/tenant_name is passed
|
|
self.assertEqual('user_id', ctx.user)
|
|
self.assertEqual('tenant_id', ctx.tenant)
|
|
|
|
def test_neutron_context_create_with_request_id(self):
|
|
ctx = _context.Context('user_id', 'tenant_id', request_id='req_id_xxx')
|
|
self.assertEqual('req_id_xxx', ctx.request_id)
|
|
|
|
def test_neutron_context_create_with_timestamp(self):
|
|
now = "Right Now!"
|
|
ctx = _context.Context('user_id', 'tenant_id', timestamp=now)
|
|
self.assertEqual(now, ctx.timestamp)
|
|
|
|
def test_neutron_context_create_is_advsvc(self):
|
|
ctx = _context.Context('user_id', 'tenant_id', is_advsvc=True)
|
|
self.assertFalse(ctx.is_admin)
|
|
self.assertTrue(ctx.is_advsvc)
|
|
|
|
def test_neutron_context_create_with_auth_token(self):
|
|
ctx = _context.Context('user_id', 'tenant_id',
|
|
auth_token='auth_token_xxx')
|
|
self.assertEqual('auth_token_xxx', ctx.auth_token)
|
|
|
|
def test_neutron_context_from_dict(self):
|
|
owner = {'user_id': 'Luke', 'tenant_id': 'Skywalker'}
|
|
ctx = _context.Context.from_dict(owner)
|
|
self.assertEqual(owner['user_id'], ctx.user_id)
|
|
self.assertEqual(owner['tenant_id'], ctx.tenant_id)
|
|
|
|
def test_neutron_context_to_dict(self):
|
|
ctx = _context.Context('user_id', 'tenant_id')
|
|
ctx_dict = ctx.to_dict()
|
|
self.assertEqual('user_id', ctx_dict['user_id'])
|
|
self.assertEqual('tenant_id', ctx_dict['project_id'])
|
|
self.assertEqual(ctx.request_id, ctx_dict['request_id'])
|
|
self.assertEqual('user_id', ctx_dict['user'])
|
|
self.assertEqual('tenant_id', ctx_dict['tenant'])
|
|
self.assertIsNone(ctx_dict['user_name'])
|
|
self.assertIsNone(ctx_dict['tenant_name'])
|
|
self.assertIsNone(ctx_dict['project_name'])
|
|
self.assertIsNone(ctx_dict['auth_token'])
|
|
|
|
def test_neutron_context_to_dict_with_name(self):
|
|
ctx = _context.Context('user_id', 'tenant_id',
|
|
tenant_name='tenant_name',
|
|
user_name='user_name')
|
|
ctx_dict = ctx.to_dict()
|
|
self.assertEqual('user_name', ctx_dict['user_name'])
|
|
self.assertEqual('tenant_name', ctx_dict['tenant_name'])
|
|
self.assertEqual('tenant_name', ctx_dict['project_name'])
|
|
|
|
def test_neutron_context_to_dict_with_auth_token(self):
|
|
ctx = _context.Context('user_id', 'tenant_id',
|
|
auth_token='auth_token_xxx')
|
|
ctx_dict = ctx.to_dict()
|
|
self.assertEqual('auth_token_xxx', ctx_dict['auth_token'])
|
|
|
|
def test_neutron_context_admin_to_dict(self):
|
|
ctx = _context.get_admin_context()
|
|
ctx_dict = ctx.to_dict()
|
|
self.assertIsNone(ctx_dict['user_id'])
|
|
self.assertIsNone(ctx_dict['tenant_id'])
|
|
self.assertIsNone(ctx_dict['auth_token'])
|
|
self.assertTrue(ctx_dict['is_admin'])
|
|
self.assertIsNotNone(ctx.session)
|
|
self.assertNotIn('session', ctx_dict)
|
|
|
|
def test_neutron_context_admin_without_session_to_dict(self):
|
|
ctx = _context.get_admin_context_without_session()
|
|
ctx_dict = ctx.to_dict()
|
|
self.assertIsNone(ctx_dict['user_id'])
|
|
self.assertIsNone(ctx_dict['tenant_id'])
|
|
self.assertIsNone(ctx_dict['auth_token'])
|
|
self.assertFalse(hasattr(ctx, 'session'))
|
|
|
|
def test_neutron_context_elevated_retains_request_id(self):
|
|
ctx = _context.Context('user_id', 'tenant_id')
|
|
self.assertFalse(ctx.is_admin)
|
|
req_id_before = ctx.request_id
|
|
|
|
elevated_ctx = ctx.elevated()
|
|
self.assertTrue(elevated_ctx.is_admin)
|
|
self.assertEqual(req_id_before, elevated_ctx.request_id)
|
|
|
|
def test_neutron_context_elevated_idempotent(self):
|
|
ctx = _context.Context('user_id', 'tenant_id')
|
|
self.assertFalse(ctx.is_admin)
|
|
elevated_ctx = ctx.elevated()
|
|
self.assertTrue(elevated_ctx.is_admin)
|
|
elevated2_ctx = elevated_ctx.elevated()
|
|
self.assertTrue(elevated2_ctx.is_admin)
|
|
|
|
def test_neutron_context_overwrite(self):
|
|
ctx1 = _context.Context('user_id', 'tenant_id')
|
|
self.assertEqual(ctx1.request_id,
|
|
oslo_context.get_current().request_id)
|
|
|
|
# If overwrite is not specified, request_id should be updated.
|
|
ctx2 = _context.Context('user_id', 'tenant_id')
|
|
self.assertNotEqual(ctx2.request_id, ctx1.request_id)
|
|
self.assertEqual(ctx2.request_id,
|
|
oslo_context.get_current().request_id)
|
|
|
|
# If overwrite is specified, request_id should be kept.
|
|
ctx3 = _context.Context('user_id', 'tenant_id', overwrite=False)
|
|
self.assertNotEqual(ctx3.request_id, ctx2.request_id)
|
|
self.assertEqual(ctx2.request_id,
|
|
oslo_context.get_current().request_id)
|
|
|
|
def test_neutron_context_get_admin_context_not_update_local_store(self):
|
|
ctx = _context.Context('user_id', 'tenant_id')
|
|
req_id_before = oslo_context.get_current().request_id
|
|
self.assertEqual(ctx.request_id, req_id_before)
|
|
|
|
ctx_admin = _context.get_admin_context()
|
|
self.assertEqual(req_id_before, oslo_context.get_current().request_id)
|
|
self.assertNotEqual(req_id_before, ctx_admin.request_id)
|
|
|
|
@mock.patch.object(_context.ContextBaseWithSession, 'session')
|
|
def test_superclass_session(self, mocked_session):
|
|
ctx = _context.Context('user_id', 'tenant_id')
|
|
# make sure context uses parent class session that is mocked
|
|
self.assertEqual(mocked_session, ctx.session)
|
|
|
|
def test_session_cached(self):
|
|
ctx = _context.Context('user_id', 'tenant_id')
|
|
session1 = ctx.session
|
|
session2 = ctx.session
|
|
self.assertIs(session1, session2)
|