Fix typing in custom template e2e test

Author: dbanty

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/defaults_tests_defaults_post.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 import datetime
@@ -26,8 +28,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[Union[None, HTTPVal
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[Union[None, HTTPValidationError]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[Union[None, HTTPValidationError]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -47,7 +49,7 @@ def httpx_request(
     list_prop: Union[Unset, List[AnEnum]] = UNSET,
     union_prop: Union[Unset, float, str] = "not a float",
     enum_prop: Union[Unset, AnEnum] = UNSET,
-) -> httpx.Response[Union[None, HTTPValidationError]]:
+) -> Response[Union[None, HTTPValidationError]]:
 
     json_datetime_prop: Union[Unset, str] = UNSET
     if not isinstance(datetime_prop, Unset):

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_booleans.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from typing import List, cast
@@ -15,8 +17,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[List[bool]]:
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[List[bool]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[List[bool]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -27,7 +29,7 @@ def _build_response(*, response: httpx.Response) -> httpx.Response[List[bool]]:
 def httpx_request(
     *,
     client: Client,
-) -> httpx.Response[List[bool]]:
+) -> Response[List[bool]]:
 
     response = client.request(
         "get",

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_floats.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from typing import List, cast
@@ -15,8 +17,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[List[float]]:
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[List[float]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[List[float]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -27,7 +29,7 @@ def _build_response(*, response: httpx.Response) -> httpx.Response[List[float]]:
 def httpx_request(
     *,
     client: Client,
-) -> httpx.Response[List[float]]:
+) -> Response[List[float]]:
 
     response = client.request(
         "get",

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_integers.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from typing import List, cast
@@ -15,8 +17,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[List[int]]:
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[List[int]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[List[int]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -27,7 +29,7 @@ def _build_response(*, response: httpx.Response) -> httpx.Response[List[int]]:
 def httpx_request(
     *,
     client: Client,
-) -> httpx.Response[List[int]]:
+) -> Response[List[int]]:
 
     response = client.request(
         "get",

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_strings.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from typing import List, cast
@@ -15,8 +17,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[List[str]]:
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[List[str]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[List[str]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -27,7 +29,7 @@ def _build_response(*, response: httpx.Response) -> httpx.Response[List[str]]:
 def httpx_request(
     *,
     client: Client,
-) -> httpx.Response[List[str]]:
+) -> Response[List[str]]:
 
     response = client.request(
         "get",

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_user_list.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 import datetime
@@ -28,8 +30,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[Union[List[AModel],
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[Union[List[AModel], HTTPValidationError]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[Union[List[AModel], HTTPValidationError]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -42,7 +44,7 @@ def httpx_request(
     client: Client,
     an_enum_value: List[AnEnum],
     some_date: Union[datetime.date, datetime.datetime],
-) -> httpx.Response[Union[List[AModel], HTTPValidationError]]:
+) -> Response[Union[List[AModel], HTTPValidationError]]:
 
     json_an_enum_value = []
     for an_enum_value_item_data in an_enum_value:

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/int_enum_tests_int_enum_post.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from typing import Dict
@@ -22,8 +24,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[Union[None, HTTPVal
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[Union[None, HTTPValidationError]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[Union[None, HTTPValidationError]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -35,7 +37,7 @@ def httpx_request(
     *,
     client: Client,
     int_enum: AnIntEnum,
-) -> httpx.Response[Union[None, HTTPValidationError]]:
+) -> Response[Union[None, HTTPValidationError]]:
 
     json_int_enum = int_enum.value
 

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/json_body_tests_json_body_post.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from ...models.a_model import AModel
@@ -20,8 +22,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[Union[None, HTTPVal
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[Union[None, HTTPValidationError]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[Union[None, HTTPValidationError]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -33,7 +35,7 @@ def httpx_request(
     *,
     client: Client,
     json_body: AModel,
-) -> httpx.Response[Union[None, HTTPValidationError]]:
+) -> Response[Union[None, HTTPValidationError]]:
 
     json_json_body = json_body.to_dict()
 

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/no_response_tests_no_response_get.py

@@ -1,10 +1,12 @@
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[None]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[None]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -15,7 +17,7 @@ def _build_response(*, response: httpx.Response) -> httpx.Response[None]:
 def httpx_request(
     *,
     client: Client,
-) -> httpx.Response[None]:
+) -> Response[None]:
 
     response = client.request(
         "get",

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/octet_stream_tests_octet_stream_get.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from io import BytesIO
@@ -17,8 +19,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[File]:
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[File]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[File]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -29,7 +31,7 @@ def _build_response(*, response: httpx.Response) -> httpx.Response[File]:
 def httpx_request(
     *,
     client: Client,
-) -> httpx.Response[File]:
+) -> Response[File]:
 
     response = client.request(
         "get",

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/optional_value_tests_optional_query_param.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from typing import Dict, List, Union
@@ -22,8 +24,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[Union[None, HTTPVal
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[Union[None, HTTPValidationError]]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[Union[None, HTTPValidationError]]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -35,7 +37,7 @@ def httpx_request(
     *,
     client: Client,
     query_param: Union[Unset, List[str]] = UNSET,
-) -> httpx.Response[Union[None, HTTPValidationError]]:
+) -> Response[Union[None, HTTPValidationError]]:
 
     json_query_param: Union[Unset, List[Any]] = UNSET
     if not isinstance(query_param, Unset):

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/test_inline_objects.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from ...models.json_body import JsonBody
@@ -16,8 +18,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[Response_200]:
     return None
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[Response_200]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[Response_200]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -29,7 +31,7 @@ def httpx_request(
     *,
     client: Client,
     json_body: JsonBody,
-) -> httpx.Response[Response_200]:
+) -> Response[Response_200]:
 
     json_json_body = json_body.to_dict()
 

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/unsupported_content_tests_unsupported_content_get.py

@@ -1,10 +1,12 @@
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[None]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[None]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -15,7 +17,7 @@ def _build_response(*, response: httpx.Response) -> httpx.Response[None]:
 def httpx_request(
     *,
     client: Client,
-) -> httpx.Response[None]:
+) -> Response[None]:
 
     response = client.request(
         "get",

end_to_end_tests/golden-record-custom/custom_e2e/api/tests/upload_file_tests_upload_post.py

@@ -2,6 +2,8 @@
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 from typing import Dict, Union, cast
@@ -27,11 +29,11 @@ def _parse_response(*, response: httpx.Response) -> Optional[Union[
 
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[Union[
+def _build_response(*, response: httpx.Response) -> Response[Union[
     None,
     HTTPValidationError
 ]]:
-    return httpx.Response(
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
@@ -43,7 +45,7 @@ def httpx_request(*,
     client: Client,
     multipart_data: BodyUploadFileTestsUploadPost,
     keep_alive: Union[Unset, bool] = UNSET,
-) -> httpx.Response[Union[
+) -> Response[Union[
     None,
     HTTPValidationError
 ]]: