33
44import httpx
55import pytest
6- from pytest_httpx import to_response
7- from pytest_httpx ._httpx_internals import Response
6+ from httpx import Response
87
98from firebolt .common .settings import Settings
109from firebolt .common .urls import (
@@ -83,7 +82,7 @@ def do_mock(
8382 ** kwargs ,
8483 ) -> Response :
8584 assert request .url == provider_url
86- return to_response (
85+ return Response (
8786 status_code = httpx .codes .OK ,
8887 json = list_to_paginated_response (mock_providers ),
8988 )
@@ -103,7 +102,7 @@ def do_mock(
103102 ** kwargs ,
104103 ) -> Response :
105104 assert request .url == region_url
106- return to_response (
105+ return Response (
107106 status_code = httpx .codes .OK ,
108107 json = list_to_paginated_response (mock_regions ),
109108 )
@@ -123,7 +122,7 @@ def do_mock(
123122 ** kwargs ,
124123 ) -> Response :
125124 assert request .url == instance_type_url
126- return to_response (
125+ return Response (
127126 status_code = httpx .codes .OK ,
128127 json = list_to_paginated_response (mock_instance_types ),
129128 )
@@ -143,7 +142,7 @@ def do_mock(
143142 ** kwargs ,
144143 ) -> Response :
145144 assert request .url == engine_url
146- return to_response (
145+ return Response (
147146 status_code = httpx .codes .OK ,
148147 json = {"engine" : mock_engine .dict ()},
149148 )
@@ -165,7 +164,7 @@ def do_mock(
165164 ** kwargs ,
166165 ) -> Response :
167166 assert request .url == account_engine_url
168- return to_response (
167+ return Response (
169168 status_code = httpx .codes .OK ,
170169 json = {"engine" : mock_engine .dict ()},
171170 )
@@ -205,7 +204,7 @@ def do_mock(
205204 mock_database .description = database_properties ["description" ]
206205
207206 assert request .url == databases_url
208- return to_response (
207+ return Response (
209208 status_code = httpx .codes .OK ,
210209 json = {"database" : mock_database .dict ()},
211210 )
@@ -218,7 +217,7 @@ def databases_get_callback(databases_url: str, mock_database) -> Callable:
218217 def get_databases_callback_inner (
219218 request : httpx .Request = None , ** kwargs
220219 ) -> Response :
221- return to_response (
220+ return Response (
222221 status_code = httpx .codes .OK , json = {"edges" : [{"node" : mock_database .dict ()}]}
223222 )
224223
@@ -239,7 +238,7 @@ def do_mock(
239238 ** kwargs ,
240239 ) -> Response :
241240 assert request .url == database_url
242- return to_response (
241+ return Response (
243242 status_code = httpx .codes .OK ,
244243 json = {"database" : mock_database .dict ()},
245244 )
@@ -254,7 +253,7 @@ def do_mock(
254253 ** kwargs ,
255254 ) -> Response :
256255 assert request .url == database_url
257- return to_response (
256+ return Response (
258257 status_code = httpx .codes .OK ,
259258 json = {},
260259 )
@@ -276,7 +275,7 @@ def do_mock(
276275 ** kwargs ,
277276 ) -> Response :
278277 assert request .url == database_get_by_name_url
279- return to_response (
278+ return Response (
280279 status_code = httpx .codes .OK ,
281280 json = {"database_id" : {"database_id" : mock_database .database_id }},
282281 )
@@ -300,7 +299,7 @@ def do_mock(
300299 ** kwargs ,
301300 ) -> Response :
302301 assert request .url == database_get_url
303- return to_response (
302+ return Response (
304303 status_code = httpx .codes .OK ,
305304 json = {"database" : mock_database .dict ()},
306305 )
@@ -335,7 +334,7 @@ def do_mock(
335334 ** kwargs ,
336335 ) -> Response :
337336 assert request .url == bindings_url
338- return to_response (
337+ return Response (
339338 status_code = httpx .codes .OK ,
340339 json = list_to_paginated_response ([binding ]),
341340 )
@@ -350,7 +349,7 @@ def do_mock(
350349 ** kwargs ,
351350 ) -> Response :
352351 assert request .url == bindings_url
353- return to_response (
352+ return Response (
354353 status_code = httpx .codes .OK ,
355354 json = list_to_paginated_response ([]),
356355 )
@@ -374,7 +373,7 @@ def do_mock(
374373 ** kwargs ,
375374 ) -> Response :
376375 assert request .url == create_binding_url
377- return to_response (
376+ return Response (
378377 status_code = httpx .codes .OK ,
379378 json = {"binding" : binding .dict ()},
380379 )
0 commit comments