Pass headers instead of request object when creating a context object

Author: samwelkanda

msgraphcore/middleware/middleware.py

@@ -26,7 +26,7 @@ def add_middleware(self, middleware):
 
     def send(self, request, **kwargs):
 
-        request.context = RequestContext(request)
+        request.context = RequestContext(request.headers)
 
         if self._middleware_present():
             return self._middleware.send(request, **kwargs)

msgraphcore/middleware/request_context.py

@@ -5,9 +5,9 @@
 
 
 class RequestContext:
-    def __init__(self, request):
+    def __init__(self, headers):
         self.middleware_control = middleware_control
-        self.client_request_id = request.headers.get('client-request-id', str(uuid.uuid4()))
+        self.client_request_id = headers.get('client-request-id', str(uuid.uuid4()))
         self._feature_usage = FeatureUsageFlag.NONE
 
     @property
@@ -16,4 +16,5 @@ def feature_usage(self):
 
     @feature_usage.setter
     def set_feature_usage(self, flag: FeatureUsageFlag):
-        self._feature_usage = self.feature_usage | flag
+        print(type(self._feature_usage))
+        self._feature_usage = self._feature_usage | flag

tests/integration/test_graph_client.py tests/integration/test_graphclient.pytests/integration/test_graph_client.py renamed to tests/integration/test_graphclient.py

@@ -71,7 +71,7 @@ def test_graph_client_builds_graph_urls():
 
     responses.add(responses.GET, graph_url, status=200)
 
-    client.get('/me')
+    client.get('/me', headers={})
     assert graph_url == responses.calls[0].request.url
 
 
@@ -85,7 +85,7 @@ def test_does_not_build_graph_urls_for_full_urls():
 
     credential = _CustomTokenCredential()
     client = GraphClient(credential=credential)
-    client.get(other_url)
+    client.get(other_url, headers={})
     request_url = responses.calls[0].request.url
     assert other_url == request_url