Home | History | Annotate | Download | only in system 
      1 # Copyright (C) 2010 Google Inc. All rights reserved.
      2 #
      3 # Redistribution and use in source and binary forms, with or without
      4 # modification, are permitted provided that the following conditions
      5 # are met:
      6 #
      7 # 1.  Redistributions of source code must retain the above copyright
      8 #     notice, this list of conditions and the following disclaimer.
      9 # 2.  Redistributions in binary form must reproduce the above copyright
     10 #     notice, this list of conditions and the following disclaimer in the
     11 #     documentation and/or other materials provided with the distribution.
     12 #
     13 # THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
     14 # EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     15 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     16 # DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
     17 # DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
     18 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     19 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
     20 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     21 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     22 # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     23 
     24 import shutil
     25 import tempfile
     26 import webkitpy.thirdparty.unittest2 as unittest
     27 import zipfile
     28 
     29 from webkitpy.common.system.filesystem_mock import MockFileSystem
     30 from webkitpy.common.system.zipfileset import ZipFileSet
     31 
     32 
     33 class FakeZip(object):
     34     def __init__(self, filesystem):
     35         self._filesystem = filesystem
     36         self._files = {}
     37 
     38     def add_file(self, filename, contents):
     39         self._files[filename] = contents
     40 
     41     def open(self, filename):
     42         return FileSetFileHandle(self, filename, self._filesystem)
     43 
     44     def namelist(self):
     45         return self._files.keys()
     46 
     47     def read(self, filename):
     48         return self._files[filename]
     49 
     50     def extract(self, filename, path):
     51         self._filesystem.write_text_file(self._filesystem.join(path, filename), self.read(filename))
     52 
     53     def delete(self, filename):
     54         raise Exception("Can't delete from a ZipFileSet.")
     55 
     56 
     57 class ZipFileSetTest(unittest.TestCase):
     58     def setUp(self):
     59         self._filesystem = MockFileSystem()
     60         self._zip = ZipFileSet('blah', self._filesystem, self.make_fake_zip)
     61 
     62     def make_fake_zip(self, zip_url):
     63         result = FakeZip(self._filesystem)
     64         result.add_file('some-file', 'contents')
     65         result.add_file('a/b/some-other-file', 'other contents')
     66         return (None, result)
     67 
     68     def test_open(self):
     69         file = self._zip.open('a/b/some-other-file')
     70         self.assertEqual('a/b/some-other-file', file.name())
     71         self.assertEqual('other contents', file.contents())
     72 
     73     def test_close(self):
     74         zipfileset = ZipFileSet('blah', self._filesystem, self.make_fake_zip)
     75         zipfileset.close()
     76 
     77     def test_read(self):
     78         self.assertEqual('contents', self._zip.read('some-file'))
     79 
     80     def test_extract(self):
     81         self._filesystem.maybe_make_directory('/some-dir')
     82         self._zip.extract('some-file', '/some-dir')
     83         self.assertTrue(self._filesystem.isfile('/some-dir/some-file'))
     84 
     85     def test_deep_extract(self):
     86         self._filesystem.maybe_make_directory('/some-dir')
     87         self._zip.extract('a/b/some-other-file', '/some-dir')
     88         self.assertTrue(self._filesystem.isfile('/some-dir/a/b/some-other-file'))
     89 
     90     def test_cant_delete(self):
     91         self.assertRaises(Exception, self._zip.delete, 'some-file')
     92 
     93     def test_namelist(self):
     94         self.assertTrue('some-file' in self._zip.namelist())
     95