optimize tests

nggit · Nov 25, 2023 · d5db2e4 · d5db2e4
1 parent 53b6305
commit d5db2e4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tests/http_server.py b/tests/http_server.py
@@ -101,6 +101,8 @@ async def my_response_middleware(**server):
         response.set_status(503, b'Under Maintenance')
         response.set_content_type(b'text/plain')
 
+        return b'Under Maintenance'
+
 
 @app.route('/getheaderline')
 async def get_headerline(**server):

diff --git a/tests/test_http_client.py b/tests/test_http_client.py
@@ -63,10 +63,11 @@ def test_get_ok_10(self):
 
         self.assertEqual(
             header[:header.find(b'\r\n')],
-            b'HTTP/1.0 503 Under Maintenance'
+            b'HTTP/1.0 503 Service Unavailable'
         )
         self.assertTrue(b'\r\nContent-Type: text/plain' in header)
         self.assertFalse(chunked_detected(header))
+        self.assertEqual(body, b'Under Maintenance')
 
         # these values are set by the request and response middleware
         self.assertTrue(b'\r\nX-Foo: baz' in header and
@@ -174,7 +175,7 @@ def test_head_10(self):
                                    version='1.0')
 
         self.assertEqual(header[:header.find(b'\r\n')],
-                         b'HTTP/1.0 503 Under Maintenance')
+                         b'HTTP/1.0 503 Service Unavailable')
         self.assertTrue(b'\r\nContent-Length: ' in header)
         self.assertTrue(b'\r\nContent-Type: text/plain' in header)
         self.assertFalse(b'\r\nTransfer-Encoding: chunked\r\n' in header)