Fix typing in custom template e2e test

dbanty · dbanty · commit d58d0fa45a57 · 2020-11-07T14:41:52.000-07:00
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/defaults_tests_defaults_post.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_booleans.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_floats.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_integers.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_basic_list_of_strings.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/get_user_list.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/int_enum_tests_int_enum_post.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/json_body_tests_json_body_post.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/no_response_tests_no_response_get.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/octet_stream_tests_octet_stream_get.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/optional_value_tests_optional_query_param.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/test_inline_objects.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/unsupported_content_tests_unsupported_content_get.py b/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,
diff --git a/end_to_end_tests/golden-record-custom/custom_e2e/api/tests/upload_file_tests_upload_post.py b/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,
diff --git a/end_to_end_tests/test_custom_templates/endpoint_module.pyi b/end_to_end_tests/test_custom_templates/endpoint_module.pyi
@@ -2,6 +2,8 @@ from typing import Optional
 
 import httpx
 
+from ...types import Response
+
 Client = httpx.Client
 
 {% for relative in endpoint.relative_imports %}
@@ -31,8 +33,8 @@ def _parse_response(*, response: httpx.Response) -> Optional[{{ return_string }}
 
 
 
-def _build_response(*, response: httpx.Response) -> httpx.Response[{{ return_string }}]:
-    return httpx.Response(
+def _build_response(*, response: httpx.Response) -> Response[{{ return_string }}]:
+    return Response(
         status_code=response.status_code,
         content=response.content,
         headers=response.headers,
