bpo-44022: Improve the regression test. (GH-26503)

gpshead · web-flow · commit e60ab843cbb0 · 2021-06-02T20:43:38.000-07:00
It wasn't actually detecting the regression due to the
assertion being too lenient.
diff --git a/Lib/test/test_httplib.py b/Lib/test/test_httplib.py
@@ -1186,7 +1186,12 @@ def test_overflowing_header_limit_after_100(self):
             'r\n' * 32768
         )
         resp = client.HTTPResponse(FakeSocket(body))
-        self.assertRaises(client.HTTPException, resp.begin)
+        with self.assertRaises(client.HTTPException) as cm:
+            resp.begin()
+        # We must assert more because other reasonable errors that we
+        # do not want can also be HTTPException derived.
+        self.assertIn('got more than ', str(cm.exception))
+        self.assertIn('headers', str(cm.exception))
 
     def test_overflowing_chunked_line(self):
         body = (
