Skip to content

Commit ccdab64

Browse files
committed
fix web response repr
1 parent 4be5043 commit ccdab64

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

aiohttp/web_reqrep.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -534,6 +534,7 @@ def __init__(self, *, status=200, reason=None, headers=None):
534534
self._req = None
535535
self._resp_impl = None
536536
self._eof_sent = False
537+
self._body_length = 0
537538

538539
if headers is not None:
539540
# TODO: optimize CIMultiDict extending
@@ -589,7 +590,7 @@ def force_close(self):
589590

590591
@property
591592
def body_length(self):
592-
return self._resp_impl.body_length
593+
return self._body_length
593594

594595
@property
595596
def output_length(self):
@@ -914,11 +915,14 @@ def write_eof(self):
914915

915916
yield from self._resp_impl.write_eof()
916917
self._eof_sent = True
918+
self._body_length = self._resp_impl.body_length
917919
self._req = None
918920
self._resp_impl = None
919921

920922
def __repr__(self):
921-
if self.started:
923+
if self._eof_sent:
924+
info = "eof"
925+
elif self.started:
922926
info = "{} {} ".format(self._req.method, self._req.path)
923927
else:
924928
info = "not started"

tests/test_web_response.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,7 @@ def test_repr_after_eof():
393393
assert not resp.started
394394
assert not resp.prepared
395395
resp_repr = repr(resp)
396-
assert resp_repr == '<StreamResponse OK not started>'
396+
assert resp_repr == '<StreamResponse OK eof>'
397397

398398

399399
@asyncio.coroutine

0 commit comments

Comments
 (0)