]> git.ipfire.org Git - thirdparty/Python/cpython.git/commitdiff
Backport test_nntplib fixes from default.
authorAntoine Pitrou <solipsis@pitrou.net>
Mon, 25 Jun 2012 16:08:54 +0000 (18:08 +0200)
committerAntoine Pitrou <solipsis@pitrou.net>
Mon, 25 Jun 2012 16:08:54 +0000 (18:08 +0200)
Lib/test/test_nntplib.py

index 86827f37f5be7909ab9978cc5b9d599ed8bc6f0a..7335b23b2d440eda0f29e10b63861a867b6c181c 100644 (file)
@@ -177,9 +177,12 @@ class NetworkedNNTPTestsMixin:
         self.assertTrue(resp.startswith("220 "), resp)
         self.check_article_resp(resp, article, art_num)
         # Tolerate running the tests from behind a NNTP virus checker
+        blacklist = lambda line: line.startswith(b'X-Antivirus')
+        filtered_head_lines = [line for line in head.lines
+                               if not blacklist(line)]
         filtered_lines = [line for line in article.lines
-                                  if not line.startswith(b'X-Antivirus')]
-        self.assertEqual(filtered_lines, head.lines + [b''] + body.lines)
+                          if not blacklist(line)]
+        self.assertEqual(filtered_lines, filtered_head_lines + [b''] + body.lines)
 
     def test_capabilities(self):
         # The server under test implements NNTP version 2 and has a