fix: remove deprecated server_name argument

mblackgeo · May 31, 2023 · 22b10c8 · 22b10c8
1 parent 9ff7f25
commit 22b10c8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,7 +1,5 @@
 import pytest
 from flask import Flask, make_response
-from jwt import PyJWKSet
-
 from flask_cognito_lib import CognitoAuth
 from flask_cognito_lib.config import Config
 from flask_cognito_lib.decorators import (
@@ -11,6 +9,7 @@
     cognito_logout,
 )
 from flask_cognito_lib.utils import CognitoTokenResponse
+from jwt import PyJWKSet
 
 
 @pytest.fixture(autouse=True)
@@ -161,5 +160,5 @@ def token_response(mocker, access_token, id_token):
 @pytest.fixture
 def client_with_cookie(app, access_token, cfg):
     cl = app.test_client()
-    cl.set_cookie(server_name="localhost", key=cfg.COOKIE_NAME, value=access_token)
+    cl.set_cookie(key=cfg.COOKIE_NAME, value=access_token)
     yield cl
diff --git a/tests/test_decorators.py b/tests/test_decorators.py
@@ -1,6 +1,5 @@
 import pytest
 from flask import session
-
 from flask_cognito_lib.decorators import remove_from_session
 from flask_cognito_lib.exceptions import TokenVerifyError
 
@@ -105,7 +104,7 @@ def test_cognito_logout(client, cfg):
 def test_auth_required_expired_token(client, cfg, app, access_token):
     # 403 if the token verification has failed
     app.config["AWS_COGNITO_EXPIRATION_LEEWAY"] = 0
-    client.set_cookie(server_name="localhost", key=cfg.COOKIE_NAME, value=access_token)
+    client.set_cookie(key=cfg.COOKIE_NAME, value=access_token)
     response = client.get("/private")
     assert response.status_code == 403