Add integration tests for the graph client

Author: samwelkanda

tests/integration/test_graph_client.py

@@ -0,0 +1,64 @@
+import pytest
+from requests import Session
+
+from msgraphcore.enums import APIVersion
+from msgraphcore.graph_client import GraphClient
+from msgraphcore.middleware.authorization import AuthorizationHandler
+
+
+class _CustomTokenCredential:
+    def get_token(self, scopes):
+        return ['{token:https://graph.microsoft.com/}']
+
+
+def test_graph_client_with_default_middleware():
+    """
+    Test that a graph client uses default middleware if none are provided
+    """
+    credential = _CustomTokenCredential()
+    client = GraphClient(credential=credential)
+    response = client.get(
+        'https://proxy.apisandbox.msdn.microsoft.com/svc?url=https://graph.microsoft.com/v1.0/me'
+    )
+    assert response.status_code == 200
+
+
+def test_graph_client_with_user_provided_session():
+    """
+    Test that the graph client works with a user provided session object
+    """
+
+    session = Session()
+    credential = _CustomTokenCredential()
+    client = GraphClient(session=session, credential=credential)
+    response = client.get(
+        'https://proxy.apisandbox.msdn.microsoft.com/svc?url=https://graph.microsoft.com/v1.0/me'
+    )
+    assert response.status_code == 200
+
+
+def test_graph_client_with_custom_settings():
+    """
+    Test that the graph client works with user provided configuration
+    """
+    credential = _CustomTokenCredential()
+    client = GraphClient(api_version=APIVersion.beta, credential=credential)
+    response = client.get(
+        'https://proxy.apisandbox.msdn.microsoft.com/svc?url=https://graph.microsoft.com/v1.0/me'
+    )
+    assert response.status_code == 200
+
+
+def test_client_factory_with_custom_middleware():
+    """
+    Test client factory works with user provided middleware
+    """
+    credential = _CustomTokenCredential()
+    middleware = [
+        AuthorizationHandler(credential),
+    ]
+    client = GraphClient(middleware=middleware)
+    response = client.get(
+        'https://proxy.apisandbox.msdn.microsoft.com/svc?url=https://graph.microsoft.com/v1.0/me'
+    )
+    assert response.status_code == 200