From d5db2e422ea4a2502eb91e381124c36c01a7d086 Mon Sep 17 00:00:00 2001
From: nggit <12218311+nggit@users.noreply.github.com>
Date: Sun, 26 Nov 2023 06:34:40 +0700
Subject: [PATCH] optimize tests

---
 tests/http_server.py      | 2 ++
 tests/test_http_client.py | 5 +++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/http_server.py b/tests/http_server.py
index ce691bd..d790bf9 100644
--- 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
index 8ca6de3..9467c4c 100644
--- 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)