diff options
Diffstat (limited to 'tests/tar/is_compressed_test.py')
-rw-r--r-- | tests/tar/is_compressed_test.py | 60 |
1 files changed, 0 insertions, 60 deletions
diff --git a/tests/tar/is_compressed_test.py b/tests/tar/is_compressed_test.py deleted file mode 100644 index 6ed7386..0000000 --- a/tests/tar/is_compressed_test.py +++ /dev/null @@ -1,60 +0,0 @@ -# Copyright 2021 The Bazel Authors. All rights reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. -"""Tests if input files are compressed.""" - -import unittest - -from bazel_tools.tools.python.runfiles import runfiles - - -class IsCompressedTest(unittest.TestCase): - - def setUp(self): - self.data_files = runfiles.Create() - - def get_file_under_test(self, file_name): - """Get the file path to a generated archive in the runfiles.""" - - return self.data_files.Rlocation( - "rules_pkg/tests/tar/" + file_name - ) - - def is_zip_compressed(self, file_name): - """Returns true if file_name is zip compressed.""" - with open(self.get_file_under_test(file_name), 'rb') as inp: - content = inp.read(2) - # A quick web search will show these magic constants are correct. - return content[0] == 0x1f and content[1] == 0x8b - - def is_bz2_compressed(self, file_name): - """Returns true if file_name is bz2 compressed.""" - with open(self.get_file_under_test(file_name), 'rb') as inp: - content = inp.read(7) - # A quick web search will show these magic constants are correct. - # This is probably well beyond what we need in this test, but why not? - return (content[0] == ord('B') and content[1] == ord('Z') - and ord('1') <= content[3] and content[3] <= ord('9') - and content[4] == 0x31 and content[5] == 0x41 and content[6] == 0x59) - - def test_guess_compression_from_extension(self): - self.assertTrue( - self.is_bz2_compressed("test-tar-compression-from-extension-bz2.bz2")) - self.assertTrue( - self.is_zip_compressed("test-tar-compression-from-extension-targz.tar.gz")) - self.assertTrue( - self.is_zip_compressed("test-tar-compression-from-extension-tgz.tgz")) - - -if __name__ == '__main__': - unittest.main() |