Issue #1625: Add stream ordering test to test_bz2.

diff --git a/Lib/test/test_bz2.py b/Lib/test/test_bz2.py
index 350d2ded101bf5fbb677478010f23a5c5074f221..e2c34d5fbff7eb9f356835d9cc55e577031c8d61 100644 (file)
--- a/Lib/test/test_bz2.py
+++ b/Lib/test/test_bz2.py
@@ -426,6 +426,17 @@ class BZ2FileTest(BaseTest):
                 next(bz2f)
             self.assertEqual(bz2f.readlines(), [])
 
+    def testMultiStreamOrdering(self):
+        # Test the ordering of streams when reading a multi-stream archive.
+        data1 = b"foo" * 1000
+        data2 = b"bar" * 1000
+        with BZ2File(self.filename, "w") as bz2f:
+            bz2f.write(data1)
+        with BZ2File(self.filename, "a") as bz2f:
+            bz2f.write(data2)
+        with BZ2File(self.filename) as bz2f:
+            self.assertEqual(bz2f.read(), data1 + data2)
+
     # Tests for a BZ2File wrapping another file object:
 
     def testReadBytesIO(self):