Skip to content

Commit 62e5140

Browse files
authored
Newcodegen (#4)
* upgrade rest * read response before returning
1 parent 8d3dad8 commit 62e5140

File tree

5 files changed

+664
-773
lines changed

5 files changed

+664
-773
lines changed

sdk/signalr/azure-messaging-webpubsubservice/azure/messaging/webpubsubservice/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -214,4 +214,5 @@ def send_request(self, http_request, **kwargs):
214214
request=request_copy,
215215
_internal_response=pipeline_response.http_response
216216
)
217+
response.read()
217218
return response

sdk/signalr/azure-messaging-webpubsubservice/azure/messaging/webpubsubservice/aio.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -99,8 +99,10 @@ async def send_request(self, http_request: corerest.HttpRequest, **kwargs: Any)
9999
# request=request_copy,
100100
# )
101101
pipeline_response = await self._pipeline.run(request_copy._internal_request, **kwargs)
102-
return corerest.AsyncHttpResponse(
102+
response = corerest.AsyncHttpResponse(
103103
status_code=pipeline_response.http_response.status_code,
104104
request=request_copy,
105105
_internal_response=pipeline_response.http_response
106106
)
107+
await response.read()
108+
return response

sdk/signalr/azure-messaging-webpubsubservice/azure/messaging/webpubsubservice/core/rest/__init__.py

+25-7
Original file line numberDiff line numberDiff line change
@@ -24,24 +24,42 @@
2424
#
2525
# --------------------------------------------------------------------------
2626
try:
27-
from ._rest_py3 import HttpRequest
28-
from ._rest_py3 import HttpResponse
29-
from ._rest_py3 import AsyncHttpResponse
30-
from ._rest_py3 import _StreamContextManager
31-
from ._rest_py3 import _AsyncStreamContextManager
27+
from ._rest_py3 import (
28+
HttpRequest,
29+
HttpResponse,
30+
AsyncHttpResponse,
31+
_StreamContextManager,
32+
_AsyncStreamContextManager,
33+
StreamConsumedError,
34+
ResponseNotReadError,
35+
ResponseClosedError,
36+
)
3237

3338
__all__ = [
3439
"HttpRequest",
3540
"HttpResponse",
3641
"AsyncHttpResponse",
3742
"_StreamContextManager",
3843
"_AsyncStreamContextManager",
44+
"StreamConsumedError",
45+
"ResponseNotReadError",
46+
"ResponseClosedError",
3947
]
4048
except (SyntaxError, ImportError):
41-
from ._rest import HttpRequest
42-
from ._rest import HttpResponse
49+
from ._rest import (
50+
HttpRequest,
51+
HttpResponse,
52+
_StreamContextManager,
53+
StreamConsumedError,
54+
ResponseNotReadError,
55+
ResponseClosedError,
56+
)
4357

4458
__all__ = [
4559
"HttpRequest",
4660
"HttpResponse",
61+
"_StreamContextManager",
62+
"StreamConsumedError",
63+
"ResponseNotReadError",
64+
"ResponseClosedError",
4765
]

0 commit comments

Comments
 (0)