chrome/common/extensions/docs/server2/directory_zipper_test.py

   1 #!/usr/bin/env python
   2 # Copyright 2013 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import unittest
   7 from cStringIO import StringIO
   8 from zipfile import ZipFile
   9
  10 from compiled_file_system import CompiledFileSystem
  11 from directory_zipper import DirectoryZipper
  12 from file_system import FileNotFoundError
  13 from test_file_system import TestFileSystem
  14 from object_store_creator import ObjectStoreCreator
  15
  16
  17 _TEST_DATA = {
  18   'top': {
  19     'one.txt': 'one.txt contents',
  20     'two': {
  21       'three.txt': 'three.txt contents',
  22       'four.txt': 'four.txt contents',
  23     },
  24   }
  25 }
  26
  27
  28 class DirectoryZipperTest(unittest.TestCase):
  29   def setUp(self):
  30     self._directory_zipper = DirectoryZipper(
  31         CompiledFileSystem.Factory(ObjectStoreCreator.ForTest()),
  32         TestFileSystem(_TEST_DATA))
  33
  34   def testTopZip(self):
  35     top_zip = ZipFile(StringIO(self._directory_zipper.Zip('top/').Get()))
  36     self.assertEqual(['top/one.txt', 'top/two/four.txt', 'top/two/three.txt'],
  37                      sorted(top_zip.namelist()))
  38     self.assertEqual('one.txt contents', top_zip.read('top/one.txt'))
  39     self.assertEqual('three.txt contents', top_zip.read('top/two/three.txt'))
  40     self.assertEqual('four.txt contents', top_zip.read('top/two/four.txt'))
  41
  42   def testTwoZip(self):
  43     two_zip = ZipFile(StringIO(self._directory_zipper.Zip('top/two/').Get()))
  44     self.assertEqual(['two/four.txt', 'two/three.txt'],
  45                      sorted(two_zip.namelist()))
  46     self.assertEqual('three.txt contents', two_zip.read('two/three.txt'))
  47     self.assertEqual('four.txt contents', two_zip.read('two/four.txt'))
  48
  49   def testNotFound(self):
  50     self.assertRaises(FileNotFoundError,
  51                       self._directory_zipper.Zip('notfound/').Get)
  52
  53
  54 if __name__ == '__main__':
  55   unittest.main()