Fix test cases for compatibility with httpx 0.28.0.

This commit is contained in:
依瑪貓 2024-12-02 08:09:17 +08:00
parent e00c14f277
commit 40278eaf06
3 changed files with 16 additions and 10 deletions

View File

@ -43,7 +43,7 @@ dependencies = [
[project.optional-dependencies] [project.optional-dependencies]
devel = [ devel = [
"httpx", "httpx >= 0.20.0",
"OpenCC", "OpenCC",
] ]

View File

@ -67,7 +67,8 @@ class NextUriTestCase(unittest.TestCase):
self.__app.add_url_rule("/test-next", view_func=test_next_uri_view, self.__app.add_url_rule("/test-next", view_func=test_next_uri_view,
methods=["GET", "POST"]) methods=["GET", "POST"])
client: httpx.Client = httpx.Client(app=self.__app, client: httpx.Client = httpx.Client(
transport=httpx.WSGITransport(app=self.__app),
base_url=TEST_SERVER) base_url=TEST_SERVER)
client.headers["Referer"] = TEST_SERVER client.headers["Referer"] = TEST_SERVER
csrf_token: str = get_csrf_token(client) csrf_token: str = get_csrf_token(client)
@ -96,7 +97,8 @@ class NextUriTestCase(unittest.TestCase):
self.__app.add_url_rule("/test-no-next", self.__app.add_url_rule("/test-no-next",
view_func=test_no_next_uri_view, view_func=test_no_next_uri_view,
methods=["GET", "POST"]) methods=["GET", "POST"])
client: httpx.Client = httpx.Client(app=self.__app, client: httpx.Client = httpx.Client(
transport=httpx.WSGITransport(app=self.__app),
base_url=TEST_SERVER) base_url=TEST_SERVER)
client.headers["Referer"] = TEST_SERVER client.headers["Referer"] = TEST_SERVER
csrf_token: str = get_csrf_token(client) csrf_token: str = get_csrf_token(client)
@ -122,7 +124,8 @@ class NextUriTestCase(unittest.TestCase):
self.__app.add_url_rule("/test-invalid-next", self.__app.add_url_rule("/test-invalid-next",
view_func=test_invalid_next_uri_view, view_func=test_invalid_next_uri_view,
methods=["GET", "POST"]) methods=["GET", "POST"])
client: httpx.Client = httpx.Client(app=self.__app, client: httpx.Client = httpx.Client(
transport=httpx.WSGITransport(app=self.__app),
base_url=TEST_SERVER) base_url=TEST_SERVER)
client.headers["Referer"] = TEST_SERVER client.headers["Referer"] = TEST_SERVER
csrf_token: str = get_csrf_token(client) csrf_token: str = get_csrf_token(client)
@ -227,7 +230,8 @@ class PaginationTestCase(unittest.TestCase):
self.assertEqual(pagination.list, self.__params.result) self.assertEqual(pagination.list, self.__params.result)
return "" return ""
self.__client: httpx.Client = httpx.Client(app=self.__app, self.__client: httpx.Client = httpx.Client(
transport=httpx.WSGITransport(app=self.__app),
base_url=TEST_SERVER) base_url=TEST_SERVER)
"""The user client.""" """The user client."""
self.__client.headers["Referer"] = TEST_SERVER self.__client.headers["Referer"] = TEST_SERVER

View File

@ -96,7 +96,9 @@ def get_client(app: Flask, username: str) -> httpx.Client:
:param username: The username. :param username: The username.
:return: The user client. :return: The user client.
""" """
client: httpx.Client = httpx.Client(app=app, base_url=TEST_SERVER) client: httpx.Client = httpx.Client(
transport=httpx.WSGITransport(app=app),
base_url=TEST_SERVER)
client.headers["Referer"] = TEST_SERVER client.headers["Referer"] = TEST_SERVER
csrf_token: str = get_csrf_token(client) csrf_token: str = get_csrf_token(client)
with app.app_context(): with app.app_context():