Cleanup unit test for Coriolis api midddleware auth

cloudbase · Nov 1, 2023 · b28780c · b28780c
1 parent 34f54e8
commit b28780c
Showing 1 changed file with 17 additions and 21 deletions.
diff --git a/coriolis/tests/api/middleware/test_auth.py b/coriolis/tests/api/middleware/test_auth.py
@@ -3,22 +3,18 @@
 
 from unittest import mock
 
-from oslo_log import log as logging
 from oslo_middleware import request_id
 from oslo_serialization import jsonutils
 
-import webob
-
+from coriolis.api.middleware.auth import CoriolisKeystoneContext
 from coriolis.api import wsgi
 from coriolis import context
-from coriolis.i18n import _
 from coriolis.tests import test_base
-from coriolis.api.middleware import auth
-from coriolis.api.middleware.auth import CoriolisKeystoneContext
+
 
 class CoriolisKeystoneContextTestCase(test_base.CoriolisBaseTestCase):
     """Test suite for the Coriolis api middleware auth."""
-    
+
     @mock.patch.object(jsonutils, "loads")
     @mock.patch.object(context, "RequestContext")
     def test__call__(self, mock_request_context, mock_jsonutils_loads):
@@ -47,19 +43,21 @@ def test__call__(self, mock_request_context, mock_jsonutils_loads):
         req_mock.environ[request_id.ENV_REQUEST_ID] = mock_req_id
 
         req_mock.headers = {
-            'X_USER' : mock_user,
-            'X_USER_ID' : mock_id,
-            'X_ROLE' : '1,2,3',
-            'X_TENANT_ID' : mock_project_id,
-            'X_TENANT' : 'mock_tenant',
-            'X_TENANT_NAME' : mock_project_name,
-            'X-Project-Domain-Name' : mock_project_domain_name,
-            'X-User-Domain-Name' : mock_user_domain_name,
-            'X_AUTH_TOKEN' : mock_auth_token,
-            'X_SERVICE_CATALOG' : 'mock_catalog',
+            'X_USER': mock_user,
+            'X_USER_ID': mock_id,
+            'X_ROLE': '1,2,3',
+            'X_TENANT_ID': mock_project_id,
+            'X_TENANT': 'mock_tenant',
+            'X_TENANT_NAME': mock_project_name,
+            'X-Project-Domain-Name': mock_project_domain_name,
+            'X-User-Domain-Name': mock_user_domain_name,
+            'X_AUTH_TOKEN': mock_auth_token,
+            'X_SERVICE_CATALOG': 'mock_catalog',
         }
-        mock_roles = [r.strip() for r in req_mock.headers.get('X_ROLE', '').split(',')]
-
+
+        mock_roles = [
+            r.strip() for r in req_mock.headers.get('X_ROLE', '').split(',')
+        ]
 
         result = CoriolisKeystoneContext(wsgi.Middleware)
         result = result(req_mock)
@@ -76,5 +74,3 @@ def test__call__(self, mock_request_context, mock_jsonutils_loads):
             service_catalog=mock_service_catalog,
             request_id=mock_req_id
         )
-
-