Add tests for get_compression

ascoderu · Dec 31, 2018 · 7184c59 · 7184c59
1 parent a670830
commit 7184c59
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_xtarfile.py b/tests/test_xtarfile.py
@@ -6,6 +6,7 @@
 from tempfile import mkstemp
 from unittest import TestCase
 
+from xtarfile.xtarfile import get_compression
 from xtarfile.xtarfile import xtarfile_open
 from xtarfile.xtarfile import HANDLERS
 
@@ -41,6 +42,16 @@ def mode(self, mode):
         return '{}|{}'.format(mode, self.compressor)
 
 
+class GetCompressionTests(TestCase):
+    def test_prefers_explicit_open_mode(self):
+        compression = get_compression('foo.tar.gz', 'r:bz2')
+        self.assertEqual(compression, 'bz2')
+
+    def test_falls_back_to_extension(self):
+        compression = get_compression('foo.tar.gz', 'r')
+        self.assertEqual(compression, 'gz')
+
+
 class OpenTests(TestCase):
     def test_roundtrip(self):
         plugins = [key for (key, value) in HANDLERS.items() if value]