diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_compression.py | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/tests/test_compression.py b/tests/test_compression.py index 97b319a..13c7835 100644 --- a/tests/test_compression.py +++ b/tests/test_compression.py @@ -93,6 +93,20 @@ L0stUijJSFXISayqVEjJTwcAlGd4GBcAAAA= with gemato.compression.open_compressed_file('gz', rf, 'rb') as gz: self.assertEqual(gz.read(), UTF16_TEST_STRING) + def test_open_potentially_compressed_path_fileno_passthrough(self): + with tempfile.NamedTemporaryFile(suffix='.gz') as rf: + fs1 = gemato.compression.open_potentially_compressed_path( + rf.name, 'w', encoding='utf_16_be') + with fs1 as cf: + self.assertListEqual([f.fileno() for f in fs1.files], + [cf.fileno() for f in fs1.files]) + + fs2 = gemato.compression.open_potentially_compressed_path( + rf.name, 'r', encoding='utf_16_be') + with fs2 as cf: + self.assertListEqual([f.fileno() for f in fs2.files], + [cf.fileno() for f in fs2.files]) + class Bzip2CompressionTest(unittest.TestCase): BASE64 = b''' @@ -196,6 +210,20 @@ OxleaA== except gemato.exceptions.UnsupportedCompression: raise unittest.SkipTest('bz2 compression unsupported') + def test_open_potentially_compressed_path_fileno_passthrough(self): + with tempfile.NamedTemporaryFile(suffix='.gz') as rf: + fs1 = gemato.compression.open_potentially_compressed_path( + rf.name, 'w', encoding='utf_16_be') + with fs1 as cf: + self.assertListEqual([f.fileno() for f in fs1.files], + [cf.fileno() for f in fs1.files]) + + fs2 = gemato.compression.open_potentially_compressed_path( + rf.name, 'r', encoding='utf_16_be') + with fs2 as cf: + self.assertListEqual([f.fileno() for f in fs2.files], + [cf.fileno() for f in fs2.files]) + class LZMALegacyCompressionTest(unittest.TestCase): BASE64 = b''' @@ -310,6 +338,20 @@ ADUdSd6zBOkOpekGFH46zix9wE9VT65OVeV479//7uUAAA== except gemato.exceptions.UnsupportedCompression: raise unittest.SkipTest('lzma compression unsupported') + def test_open_potentially_compressed_path_fileno_passthrough(self): + with tempfile.NamedTemporaryFile(suffix='.gz') as rf: + fs1 = gemato.compression.open_potentially_compressed_path( + rf.name, 'w', encoding='utf_16_be') + with fs1 as cf: + self.assertListEqual([f.fileno() for f in fs1.files], + [cf.fileno() for f in fs1.files]) + + fs2 = gemato.compression.open_potentially_compressed_path( + rf.name, 'r', encoding='utf_16_be') + with fs2 as cf: + self.assertListEqual([f.fileno() for f in fs2.files], + [cf.fileno() for f in fs2.files]) + class XZCompressionTest(unittest.TestCase): BASE64 = b''' @@ -400,6 +442,20 @@ dGhlIGxhenkgZG9nAADjZCTmHjHqggABLxeBCEmxH7bzfQEAAAAABFla except gemato.exceptions.UnsupportedCompression: raise unittest.SkipTest('xz compression unsupported') + def test_open_potentially_compressed_path_fileno_passthrough(self): + with tempfile.NamedTemporaryFile(suffix='.gz') as rf: + fs1 = gemato.compression.open_potentially_compressed_path( + rf.name, 'w', encoding='utf_16_be') + with fs1 as cf: + self.assertListEqual([f.fileno() for f in fs1.files], + [cf.fileno() for f in fs1.files]) + + fs2 = gemato.compression.open_potentially_compressed_path( + rf.name, 'r', encoding='utf_16_be') + with fs2 as cf: + self.assertListEqual([f.fileno() for f in fs2.files], + [cf.fileno() for f in fs2.files]) + class NoCompressionTest(unittest.TestCase): """ |