changeset:   91618:486c1a81ee32
user:        Berker Peksag <berker.peksag@gmail.com>
date:        Wed Jul 09 03:12:23 2014 +0300
files:       Lib/test/test_asynchat.py
description:
Issue #6916: Use assertWarns in test_asynchat.


diff -r 3b2af26f4638 -r 486c1a81ee32 Lib/test/test_asynchat.py
--- a/Lib/test/test_asynchat.py	Tue Jul 08 16:22:48 2014 -0700
+++ b/Lib/test/test_asynchat.py	Wed Jul 09 03:12:23 2014 +0300
@@ -282,10 +282,10 @@
 
 class TestFifo(unittest.TestCase):
     def test_basic(self):
-        with warnings.catch_warnings(record=True) as w:
+        with self.assertWarns(DeprecationWarning) as cm:
             f = asynchat.fifo()
-            if w:
-                assert issubclass(w[0].category, DeprecationWarning)
+        self.assertEqual(str(cm.warning),
+                         "fifo class will be removed in Python 3.6")
         f.push(7)
         f.push(b'a')
         self.assertEqual(len(f), 2)
@@ -300,10 +300,10 @@
         self.assertEqual(f.pop(), (0, None))
 
     def test_given_list(self):
-        with warnings.catch_warnings(record=True) as w:
+        with self.assertWarns(DeprecationWarning) as cm:
             f = asynchat.fifo([b'x', 17, 3])
-            if w:
-                assert issubclass(w[0].category, DeprecationWarning)
+        self.assertEqual(str(cm.warning),
+                         "fifo class will be removed in Python 3.6")
         self.assertEqual(len(f), 3)
         self.assertEqual(f.pop(), (1, b'x'))
         self.assertEqual(f.pop(), (1, 17))