Skip to content

Commit d5db2e4

Browse files
committed
optimize tests
1 parent 53b6305 commit d5db2e4

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

tests/http_server.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,8 @@ async def my_response_middleware(**server):
101101
response.set_status(503, b'Under Maintenance')
102102
response.set_content_type(b'text/plain')
103103

104+
return b'Under Maintenance'
105+
104106

105107
@app.route('/getheaderline')
106108
async def get_headerline(**server):

tests/test_http_client.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,10 +63,11 @@ def test_get_ok_10(self):
6363

6464
self.assertEqual(
6565
header[:header.find(b'\r\n')],
66-
b'HTTP/1.0 503 Under Maintenance'
66+
b'HTTP/1.0 503 Service Unavailable'
6767
)
6868
self.assertTrue(b'\r\nContent-Type: text/plain' in header)
6969
self.assertFalse(chunked_detected(header))
70+
self.assertEqual(body, b'Under Maintenance')
7071

7172
# these values are set by the request and response middleware
7273
self.assertTrue(b'\r\nX-Foo: baz' in header and
@@ -174,7 +175,7 @@ def test_head_10(self):
174175
version='1.0')
175176

176177
self.assertEqual(header[:header.find(b'\r\n')],
177-
b'HTTP/1.0 503 Under Maintenance')
178+
b'HTTP/1.0 503 Service Unavailable')
178179
self.assertTrue(b'\r\nContent-Length: ' in header)
179180
self.assertTrue(b'\r\nContent-Type: text/plain' in header)
180181
self.assertFalse(b'\r\nTransfer-Encoding: chunked\r\n' in header)

0 commit comments

Comments
 (0)