Issue #18919: Check warnings messages in the aifc module tests.

serhiy-storchaka · serhiy-storchaka · commit 1f56a94a9e72 · 2013-10-14T20:06:04.000+03:00
diff --git a/Lib/test/test_aifc.py b/Lib/test/test_aifc.py
@@ -275,26 +275,30 @@ def test_read_wrong_marks(self):
         b += b'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0)
         b += b'SSND' + struct.pack('>L', 8) + b'\x00' * 8
         b += b'MARK' + struct.pack('>LhB', 3, 1, 1)
-        with self.assertWarns(UserWarning):
+        with self.assertWarns(UserWarning) as cm:
             f = aifc.open(io.BytesIO(b))
+        self.assertEqual(str(cm.warning), 'Warning: MARK chunk contains '
+                                          'only 0 markers instead of 1')
         self.assertEqual(f.getmarkers(), None)
 
     def test_read_comm_kludge_compname_even(self):
         b = b'FORM' + struct.pack('>L', 4) + b'AIFC'
         b += b'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0)
         b += b'NONE' + struct.pack('B', 4) + b'even' + b'\x00'
         b += b'SSND' + struct.pack('>L', 8) + b'\x00' * 8
-        with self.assertWarns(UserWarning):
+        with self.assertWarns(UserWarning) as cm:
             f = aifc.open(io.BytesIO(b))
+        self.assertEqual(str(cm.warning), 'Warning: bad COMM chunk size')
         self.assertEqual(f.getcompname(), b'even')
 
     def test_read_comm_kludge_compname_odd(self):
         b = b'FORM' + struct.pack('>L', 4) + b'AIFC'
         b += b'COMM' + struct.pack('>LhlhhLL', 18, 0, 0, 0, 0, 0, 0)
         b += b'NONE' + struct.pack('B', 3) + b'odd'
         b += b'SSND' + struct.pack('>L', 8) + b'\x00' * 8
-        with self.assertWarns(UserWarning):
+        with self.assertWarns(UserWarning) as cm:
             f = aifc.open(io.BytesIO(b))
+        self.assertEqual(str(cm.warning), 'Warning: bad COMM chunk size')
         self.assertEqual(f.getcompname(), b'odd')
 
     def test_write_params_raises(self):
