Add missing responses

Colin-b · Sep 29, 2024 · 5298043 · 5298043
1 parent 63079b8
commit 5298043
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 0 deletions.
diff --git a/tests/oauth2/implicit/test_oauth2_implicit_async.py b/tests/oauth2/implicit/test_oauth2_implicit_async.py
@@ -184,6 +184,13 @@ async def test_oauth2_implicit_flow_token_is_reused_if_only_nonce_differs(
         token_field_name="custom_token",
     )
 
+    httpx_mock.add_response(
+        url="https://authorized_only",
+        method="GET",
+        match_headers={
+            "Authorization": f"Bearer {token}",
+        },
+    )
     async with httpx.AsyncClient() as client:
         await client.get("https://authorized_only", auth=auth2)
 
@@ -639,6 +646,13 @@ async def test_oauth2_implicit_flow_token_is_reused_if_not_expired(
 
     auth2 = httpx_auth.OAuth2Implicit("https://provide_token")
 
+    httpx_mock.add_response(
+        url="https://authorized_only",
+        method="GET",
+        match_headers={
+            "Authorization": f"Bearer {token}",
+        },
+    )
     async with httpx.AsyncClient() as client:
         await client.get("https://authorized_only", auth=auth2)
 

diff --git a/tests/oauth2/implicit/test_oauth2_implicit_sync.py b/tests/oauth2/implicit/test_oauth2_implicit_sync.py
@@ -179,6 +179,13 @@ def test_oauth2_implicit_flow_token_is_reused_if_only_nonce_differs(
         token_field_name="custom_token",
     )
 
+    httpx_mock.add_response(
+        url="https://authorized_only",
+        method="GET",
+        match_headers={
+            "Authorization": f"Bearer {token}",
+        },
+    )
     with httpx.Client() as client:
         client.get("https://authorized_only", auth=auth2)
 
@@ -616,6 +623,13 @@ def test_oauth2_implicit_flow_token_is_reused_if_not_expired(
 
     auth2 = httpx_auth.OAuth2Implicit("https://provide_token")
 
+    httpx_mock.add_response(
+        url="https://authorized_only",
+        method="GET",
+        match_headers={
+            "Authorization": f"Bearer {token}",
+        },
+    )
     with httpx.Client() as client:
         client.get("https://authorized_only", auth=auth2)
 

diff --git a/tests/oauth2/resource_owner_password/test_oauth2_resource_owner_password_async.py b/tests/oauth2/resource_owner_password/test_oauth2_resource_owner_password_async.py
@@ -431,6 +431,18 @@ async def test_oauth2_password_credentials_flow_refresh_token_invalid(
         match_content=b"grant_type=refresh_token&refresh_token=tGzv3JOkF0XG5Qx2TlKWIA",
     )
 
+    httpx_mock.add_response(
+        method="POST",
+        url="https://provide_access_token",
+        json={
+            "access_token": "2YotnFZFEjr1zCsicMWpAA",
+            "token_type": "example",
+            "expires_in": "0",
+            "refresh_token": "tGzv3JOkF0XG5Qx2TlKWIA",
+            "example_parameter": "example_value",
+        },
+        match_content=b"grant_type=password&username=test_user&password=test_pwd",
+    )
     httpx_mock.add_response(
         url="https://authorized_only",
         method="GET",
@@ -473,6 +485,13 @@ async def test_oauth2_password_credentials_flow_refresh_token_access_token_not_e
     async with httpx.AsyncClient() as client:
         await client.get("https://authorized_only", auth=auth)
 
+    httpx_mock.add_response(
+        url="https://authorized_only",
+        method="GET",
+        match_headers={
+            "Authorization": "Bearer 2YotnFZFEjr1zCsicMWpAA",
+        },
+    )
     # expect Bearer token to remain the same
     async with httpx.AsyncClient() as client:
         await client.get("https://authorized_only", auth=auth)