Add missing type annotations.

Author: SzymonFig

tests/client_generators/dependencies/test_async_base_client.py

@@ -19,7 +19,7 @@
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_post_to_correct_url_with_correct_payload(httpx_mock):
+async def test_execute_sends_post_to_correct_url_with_correct_payload(httpx_mock) -> None:
     httpx_mock.add_response()
 
     client = AsyncBaseClient(url="http://base_url/endpoint")
@@ -44,7 +44,7 @@ async def test_execute_sends_post_to_correct_url_with_correct_payload(httpx_mock
 
 
 @pytest.mark.asyncio
-async def test_execute_parses_pydantic_variables_before_sending(httpx_mock):
+async def test_execute_parses_pydantic_variables_before_sending(httpx_mock) -> None:
     class TestModel1(BaseModel):
         a: int
 
@@ -110,7 +110,7 @@ class TestModel1(BaseModel):
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_payload_without_unset_arguments(httpx_mock):
+async def test_execute_sends_payload_without_unset_arguments(httpx_mock) -> None:
     httpx_mock.add_response()
     client = AsyncBaseClient(url="http://base_url")
     query_str = """
@@ -185,7 +185,7 @@ class TestInputA(BaseModel):
 @pytest.mark.asyncio
 async def test_execute_sends_payload_with_serialized_datetime_without_exception(
     httpx_mock,
-):
+) -> None:
     httpx_mock.add_response()
     client = AsyncBaseClient(url="http://base_url")
     query_str = "query Abc($arg: DATETIME) { abc }"
@@ -212,7 +212,7 @@ async def test_execute_sends_request_with_correct_content_type(httpx_mock):
 @pytest.mark.asyncio
 async def test_execute_sends_request_with_extra_headers_and_correct_content_type(
     httpx_mock,
-):
+) -> None:
     httpx_mock.add_response()
     client = AsyncBaseClient(url="http://base_url", headers={"h_key": "h_value"})
 
@@ -224,7 +224,7 @@ async def test_execute_sends_request_with_extra_headers_and_correct_content_type
 
 
 @pytest.mark.asyncio
-async def test_execute_passes_kwargs_to_json_post(mocker):
+async def test_execute_passes_kwargs_to_json_post(mocker) -> None:
     http_client = mocker.AsyncMock()
 
     await AsyncBaseClient(url="http://base_url", http_client=http_client).execute(
@@ -236,7 +236,7 @@ async def test_execute_passes_kwargs_to_json_post(mocker):
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_json_request_with_headers_from_passed_kwargs(httpx_mock):
+async def test_execute_sends_json_request_with_headers_from_passed_kwargs(httpx_mock) -> None:
     httpx_mock.add_response()
     client = AsyncBaseClient(
         url="http://base_url",
@@ -263,7 +263,7 @@ async def test_execute_sends_json_request_with_headers_from_passed_kwargs(httpx_
 @pytest.mark.asyncio
 async def test_execute_sends_file_with_multipart_form_data_content_type(
     httpx_mock, txt_file
-):
+) -> None:
     httpx_mock.add_response()
 
     client = AsyncBaseClient(url="http://base_url")
@@ -276,7 +276,7 @@ async def test_execute_sends_file_with_multipart_form_data_content_type(
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_file_as_multipart_request(httpx_mock, txt_file):
+async def test_execute_sends_file_as_multipart_request(httpx_mock, txt_file) -> None:
     httpx_mock.add_response()
     query_str = "query Abc($file: Upload!) { abc(file: $file) }"
 
@@ -307,7 +307,7 @@ async def test_execute_sends_file_as_multipart_request(httpx_mock, txt_file):
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_file_from_memory(httpx_mock, in_memory_txt_file):
+async def test_execute_sends_file_from_memory(httpx_mock, in_memory_txt_file) -> None:
     httpx_mock.add_response()
     query_str = "query Abc($file: Upload!) { abc(file: $file) }"
 
@@ -338,7 +338,7 @@ async def test_execute_sends_file_from_memory(httpx_mock, in_memory_txt_file):
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_multiple_files(httpx_mock, txt_file, png_file):
+async def test_execute_sends_multiple_files(httpx_mock, txt_file, png_file) -> None:
     httpx_mock.add_response()
     query_str = "query Abc($files: [Upload!]!) { abc(files: $files) }"
 
@@ -374,7 +374,7 @@ async def test_execute_sends_multiple_files(httpx_mock, txt_file, png_file):
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_nested_file(httpx_mock, txt_file):
+async def test_execute_sends_nested_file(httpx_mock, txt_file) -> None:
     class InputType(BaseModel):
         file_: Any
 
@@ -408,7 +408,7 @@ class InputType(BaseModel):
 
 
 @pytest.mark.asyncio
-async def test_execute_sends_each_file_only_once(httpx_mock, txt_file):
+async def test_execute_sends_each_file_only_once(httpx_mock, txt_file) -> None:
     httpx_mock.add_response()
     query_str = "query Abc($files: [Upload!]!) { abc(files: $files) }"
 
@@ -439,7 +439,7 @@ async def test_execute_sends_each_file_only_once(httpx_mock, txt_file):
 
 
 @pytest.mark.asyncio
-async def test_execute_passes_kwargs_to_multipart_post(mocker, txt_file):
+async def test_execute_passes_kwargs_to_multipart_post(mocker, txt_file) -> None:
     http_client = mocker.AsyncMock()
 
     await AsyncBaseClient(url="http://base_url", http_client=http_client).execute(
@@ -457,7 +457,7 @@ async def test_execute_passes_kwargs_to_multipart_post(mocker, txt_file):
 @pytest.mark.asyncio
 async def test_execute_sends_multipart_request_with_headers_from_passed_kwargs(
     httpx_mock, txt_file
-):
+) -> None:
     httpx_mock.add_response()
     client = AsyncBaseClient(
         url="http://base_url",
@@ -494,7 +494,7 @@ async def test_execute_sends_multipart_request_with_headers_from_passed_kwargs(
 )
 def test_get_data_raises_graphql_client_http_error(
     mocker, status_code, response_content
-):
+) -> None:
     client = AsyncBaseClient(url="base_url", http_client=mocker.MagicMock())
     response = httpx.Response(
         status_code=status_code, content=json.dumps(response_content)
@@ -509,7 +509,7 @@ def test_get_data_raises_graphql_client_http_error(
 @pytest.mark.parametrize("response_content", ["invalid_json", {"not_data": ""}, ""])
 def test_get_data_raises_graphql_client_invalid_response_error(
     mocker, response_content
-):
+) -> None:
     client = AsyncBaseClient(url="base_url", http_client=mocker.MagicMock())
     response = httpx.Response(status_code=200, content=json.dumps(response_content))
 
@@ -553,7 +553,7 @@ def test_get_data_raises_graphql_client_invalid_response_error(
         },
     ],
 )
-def test_get_data_raises_graphql_client_graphql_multi_error(mocker, response_content):
+def test_get_data_raises_graphql_client_graphql_multi_error(mocker, response_content) -> None:
     client = AsyncBaseClient(url="base_url", http_client=mocker.MagicMock())
 
     with pytest.raises(GraphQLClientGraphQLMultiError):
@@ -566,7 +566,7 @@ def test_get_data_raises_graphql_client_graphql_multi_error(mocker, response_con
     "response_content",
     [{"errors": [], "data": {}}, {"errors": None, "data": {}}, {"data": {}}],
 )
-def test_get_data_doesnt_raise_exception(mocker, response_content):
+def test_get_data_doesnt_raise_exception(mocker, response_content) -> None:
     client = AsyncBaseClient(url="base_url", http_client=mocker.MagicMock())
 
     data = client.get_data(
@@ -577,7 +577,7 @@ def test_get_data_doesnt_raise_exception(mocker, response_content):
 
 
 @pytest.mark.asyncio
-async def test_base_client_used_as_context_manager_closes_http_client(mocker):
+async def test_base_client_used_as_context_manager_closes_http_client(mocker) -> None:
     fake_client = mocker.AsyncMock()
     async with AsyncBaseClient(url="base_url", http_client=fake_client) as base_client:
         await base_client.execute("")

tests/client_generators/dependencies/test_exceptions.py

@@ -4,7 +4,7 @@
 )
 
 
-def test_from_dict_returns_graphql_error_with_parsed_data():
+def test_from_dict_returns_graphql_error_with_parsed_data() -> None:
     error_dict = {
         "message": "Error message",
         "locations": [{"line": 6, "column": 7}],
@@ -25,7 +25,7 @@ def test_from_dict_returns_graphql_error_with_parsed_data():
     assert result.original == error_dict
 
 
-def test_from_errors_dicts_returns_graphql_multi_error_with_parsed_data():
+def test_from_errors_dicts_returns_graphql_multi_error_with_parsed_data() -> None:
     errors_dicts = [
         {
             "message": "Error message",

tests/main/clients/client_forward_refs/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/client_forward_refs_shorter_results/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/custom_config_file/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/custom_files_names/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/custom_query_builder/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager 
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/custom_scalars/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/example/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/extended_models/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/fragments_on_abstract_types/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/inline_fragments/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/interface_as_fragment/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/multiple_fragments/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/only_used_inputs_and_enums/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/operations/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/remote_schema/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable
 

tests/main/clients/shorter_results/expected_client/async_base_client.py

@@ -28,8 +28,8 @@
 except ImportError:
     from contextlib import asynccontextmanager
 
-    @asynccontextmanager  # type: ignore
-    async def ws_connect(*args, **kwargs):  # pylint: disable=unused-argument
+    @asynccontextmanager
+    async def ws_connect(*args: Any, **kwargs: Any) -> AsyncIterator[None]:  # pylint: disable=unused-argument
         raise NotImplementedError("Subscriptions require 'websockets' package.")
         yield  # pylint: disable=unreachable