code.vuplus.com Git - vuplus_webkit/blob - Tools/Scripts/webkitpy/common/system/zip_mock.py

   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 from webkitpy.common.system.fileset import FileSetFileHandle
  25 from webkitpy.common.system.filesystem_mock import MockFileSystem
  26
  27
  28 class MockZip(object):
  29     """A mock zip file that can have new files inserted into it."""
  30     def __init__(self, filesystem=None):
  31         self._filesystem = filesystem or MockFileSystem()
  32         self._files = {}
  33
  34     def __str__(self):
  35         return "MockZip"
  36
  37     def insert(self, filename, content):
  38         self._files[filename] = content
  39
  40     def namelist(self):
  41         return self._files.keys()
  42
  43     def open(self, filename):
  44         return FileSetFileHandle(self, filename)
  45
  46     def read(self, filename):
  47         return self._files[filename]
  48
  49     def extract(self, filename, path):
  50         full_path = self._filesystem.join(path, filename)
  51         contents = self.open(filename).contents()
  52         self._filesystem.write_text_file(full_path, contents)
  53
  54     def delete(self, filename):
  55         self._files[filename] = None