5b61df1f53
The context was previously logging at the 'warn' level when unknown kwargs were being passed to its __init__(). Since the agents were passing tenant=None with each rpc request, this was generating an unreasonable amount of log chatter that would not be useful to an operator. The fix is to log at the debug level instead so that the operators don't see the output by default but developers can still choose to. Change-Id: I5c328f628c597eb949c1fe67b23120d2b5d1c7da Related-Bug: #1254530 Partial-Bug: #1255441
140 lines
5.9 KiB
Python
140 lines
5.9 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2012 Nicira, Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# 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 testtools import matchers
|
|
|
|
from neutron import context
|
|
from neutron.openstack.common import local
|
|
from neutron.tests import base
|
|
|
|
|
|
class TestNeutronContext(base.BaseTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestNeutronContext, self).setUp()
|
|
db_api = 'neutron.db.api.get_session'
|
|
self._db_api_session_patcher = mock.patch(db_api)
|
|
self.db_api_session = self._db_api_session_patcher.start()
|
|
self.addCleanup(self._db_api_session_patcher.stop)
|
|
|
|
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)
|
|
self.assertThat(ctx.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)
|
|
|
|
def test_neutron_context_create_logs_unknown_kwarg(self):
|
|
with mock.patch.object(context.LOG, 'debug') as mock_log:
|
|
context.Context('user_id', 'tenant_id', foo=None)
|
|
self.assertEqual(mock_log.call_count, 1)
|
|
|
|
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_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'])
|
|
|
|
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_admin_to_dict(self):
|
|
self.db_api_session.return_value = 'fakesession'
|
|
ctx = context.get_admin_context()
|
|
ctx_dict = ctx.to_dict()
|
|
self.assertIsNone(ctx_dict['user_id'])
|
|
self.assertIsNone(ctx_dict['tenant_id'])
|
|
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.assertFalse(hasattr(ctx, 'session'))
|
|
|
|
def test_neutron_context_with_load_roles_true(self):
|
|
ctx = context.get_admin_context()
|
|
self.assertIn('admin', ctx.roles)
|
|
|
|
def test_neutron_context_with_load_roles_false(self):
|
|
ctx = context.get_admin_context(load_admin_roles=False)
|
|
self.assertFalse(ctx.roles)
|
|
|
|
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_overwrite(self):
|
|
ctx1 = context.Context('user_id', 'tenant_id')
|
|
self.assertEqual(ctx1.request_id, local.store.context.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, local.store.context.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, local.store.context.request_id)
|
|
|
|
def test_neutron_context_get_admin_context_not_update_local_store(self):
|
|
ctx = context.Context('user_id', 'tenant_id')
|
|
req_id_before = local.store.context.request_id
|
|
self.assertEqual(ctx.request_id, req_id_before)
|
|
|
|
ctx_admin = context.get_admin_context()
|
|
self.assertEqual(req_id_before, local.store.context.request_id)
|
|
self.assertNotEqual(req_id_before, ctx_admin.request_id)
|