mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-09-24 16:30:27 +08:00
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
import shutil
|
|
import tempfile
|
|
import unittest
|
|
|
|
import six
|
|
|
|
from galaxy.util.compression_utils import (
|
|
CompressedFile,
|
|
get_fileobj_raw
|
|
)
|
|
|
|
|
|
class CompressionUtilTestCase(unittest.TestCase):
|
|
|
|
def test_compression_safety(self):
|
|
self.assert_safety("test-data/unsafe.tar", False)
|
|
self.assert_safety("test-data/unsafe_relative_symlink.tar", False)
|
|
self.assert_safety("test-data/unsafe.zip", False)
|
|
self.assert_safety("test-data/4.bed.zip", True)
|
|
self.assert_safety("test-data/testdir.tar", True)
|
|
self.assert_safety("test-data/safetar_with_symlink.tar", True)
|
|
self.assert_safety("test-data/safe_relative_symlink.tar", True)
|
|
|
|
def test_get_fileobj_raw(self):
|
|
self.assert_format_detected("test-data/4.bed.zip", "zip")
|
|
self.assert_format_detected(
|
|
"test-data/4.bed.zip", None, ["bz2", "gzip"]
|
|
)
|
|
self.assert_format_detected("test-data/4.bed.gz", "gzip")
|
|
self.assert_format_detected(
|
|
"test-data/4.bed.gz", None, ["bz2", "zip"]
|
|
)
|
|
self.assert_format_detected("test-data/4.bed.bz2", "bz2")
|
|
self.assert_format_detected(
|
|
"test-data/4.bed.bz2", None, ["gzip", "zip"]
|
|
)
|
|
|
|
def assert_safety(self, path, expected_to_be_safe):
|
|
temp_dir = tempfile.mkdtemp()
|
|
try:
|
|
if expected_to_be_safe:
|
|
CompressedFile(path).extract(temp_dir)
|
|
else:
|
|
with self.assertRaises(Exception):
|
|
CompressedFile(path).extract(temp_dir)
|
|
finally:
|
|
shutil.rmtree(temp_dir, ignore_errors=True)
|
|
|
|
def assert_format_detected(self, path, expected_fmt, allowed_fmts=None):
|
|
for mode in ['r', 'rb', 'rt', 'U']:
|
|
if 'b' in mode:
|
|
expected_type = six.binary_type
|
|
else:
|
|
expected_type = six.text_type
|
|
fmt, fh = get_fileobj_raw(path, mode, allowed_fmts)
|
|
assert fmt == expected_fmt
|
|
assert isinstance(fh.read(0), expected_type)
|