Fix context hook unit test error.

Change-Id: I8559b62c2b485d385626253cfcca4ecda2ecaa7f
Closes-Bug: 1808900
This commit is contained in:
zxgao 2018-12-18 11:01:05 +08:00
parent 1c9b721b5c
commit c801a3f8ea
1 changed files with 8 additions and 2 deletions

View File

@ -19,6 +19,7 @@ from oslo_config import cfg
from oslo_context import context
from mogan.api import hooks
from mogan.common import policy
from mogan.tests import base
@ -92,10 +93,12 @@ def fake_headers(admin=False):
class TestContextHook(base.TestCase):
@mock.patch.object(context, 'RequestContext')
def test_context_hook(self, mock_ctx):
@mock.patch.object(policy, 'check')
def test_context_hook(self, mock_policy, mock_ctx, is_admin=True):
headers = fake_headers(admin=True)
reqstate = FakeRequestState(headers=headers)
context_hook = hooks.ContextHook(None)
mock_policy.return_value = is_admin
context_hook.before(reqstate)
mock_ctx.assert_called_with(
auth_token=headers['X-Auth-Token'],
@ -109,11 +112,14 @@ class TestContextHook(base.TestCase):
roles=headers['X-Roles'].split(','))
@mock.patch.object(context, 'RequestContext')
def test_context_hook_public_api(self, mock_ctx):
@mock.patch.object(policy, 'check')
def test_context_hook_public_api(self, mock_policy, mock_ctx,
is_admin=True):
headers = fake_headers(admin=True)
env = {'is_public_api': True}
reqstate = FakeRequestState(headers=headers, environ=env)
context_hook = hooks.ContextHook(None)
mock_policy.return_value = is_admin
context_hook.before(reqstate)
mock_ctx.assert_called_with(
auth_token=headers['X-Auth-Token'],