| OLD | NEW | 
|---|
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python | 
| 2 # Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 3 # Use of this source code is governed by a BSD-style license that can be | 
| 4 # found in the LICENSE file. | 4 # found in the LICENSE file. | 
| 5 | 5 | 
| 6 import unittest | 6 import unittest | 
| 7 | 7 | 
| 8 from chained_compiled_file_system import ChainedCompiledFileSystem | 8 from chained_compiled_file_system import ChainedCompiledFileSystem | 
| 9 from compiled_file_system import CompiledFileSystem | 9 from compiled_file_system import CompiledFileSystem | 
| 10 from object_store_creator import ObjectStoreCreator | 10 from object_store_creator import ObjectStoreCreator | 
| (...skipping 13 matching lines...) Expand all  Loading... | 
| 24     'b.txt': 'new b.txt', | 24     'b.txt': 'new b.txt', | 
| 25     'new.txt': 'new file in dir', | 25     'new.txt': 'new file in dir', | 
| 26   }, | 26   }, | 
| 27 } | 27 } | 
| 28 | 28 | 
| 29 identity = lambda _, x: x | 29 identity = lambda _, x: x | 
| 30 | 30 | 
| 31 class ChainedCompiledFileSystemTest(unittest.TestCase): | 31 class ChainedCompiledFileSystemTest(unittest.TestCase): | 
| 32   def setUp(self): | 32   def setUp(self): | 
| 33     object_store_creator = ObjectStoreCreator(start_empty=False) | 33     object_store_creator = ObjectStoreCreator(start_empty=False) | 
| 34     base_file_system = TestFileSystem(_TEST_DATA_BASE) | 34     base_file_system = TestFileSystem(_TEST_DATA_BASE, identity='base') | 
| 35     self._base_factory = CompiledFileSystem.Factory(base_file_system, | 35     self._base_compiled_fs = CompiledFileSystem.Factory( | 
| 36                                                     object_store_creator) | 36         object_store_creator).Create(base_file_system, | 
| 37     self._file_system = TestFileSystem(_TEST_DATA_NEW) | 37                                      identity, | 
| 38     self._patched_factory = CompiledFileSystem.Factory(self._file_system, | 38                                      ChainedCompiledFileSystemTest) | 
| 39                                                        object_store_creator) | 39     chained_factory = ChainedCompiledFileSystem.Factory([base_file_system], | 
| 40     self._chained_factory = ChainedCompiledFileSystem.Factory( | 40                                                         object_store_creator) | 
| 41         [(self._patched_factory, self._file_system), | 41     self._new_file_system = TestFileSystem(_TEST_DATA_NEW, identity='new') | 
| 42          (self._base_factory, base_file_system)]) | 42     self._chained_compiled_fs = chained_factory.Create( | 
| 43     self._base_compiled_fs = self._base_factory.Create(identity, TestFileSystem) | 43         self._new_file_system, identity, ChainedCompiledFileSystemTest) | 
| 44     self._chained_compiled_fs = self._chained_factory.Create( |  | 
| 45         identity, TestFileSystem) |  | 
| 46 | 44 | 
| 47   def testGetFromFile(self): | 45   def testGetFromFile(self): | 
| 48     self.assertEqual(self._chained_compiled_fs.GetFromFile('a.txt').Get(), | 46     self.assertEqual(self._chained_compiled_fs.GetFromFile('a.txt').Get(), | 
| 49                      self._base_compiled_fs.GetFromFile('a.txt').Get()) | 47                      self._base_compiled_fs.GetFromFile('a.txt').Get()) | 
| 50     self.assertEqual(self._chained_compiled_fs.GetFromFile('new.txt').Get(), | 48     self.assertEqual(self._chained_compiled_fs.GetFromFile('new.txt').Get(), | 
| 51                      'a new file') | 49                      'a new file') | 
| 52     self.assertEqual(self._chained_compiled_fs.GetFromFile('dir/new.txt').Get(), | 50     self.assertEqual(self._chained_compiled_fs.GetFromFile('dir/new.txt').Get(), | 
| 53                      'new file in dir') | 51                      'new file in dir') | 
| 54     self._file_system.IncrementStat('a.txt') | 52     self._new_file_system.IncrementStat('a.txt') | 
| 55     self.assertNotEqual(self._chained_compiled_fs.GetFromFile('a.txt').Get(), | 53     self.assertNotEqual(self._chained_compiled_fs.GetFromFile('a.txt').Get(), | 
| 56                         self._base_compiled_fs.GetFromFile('a.txt').Get()) | 54                         self._base_compiled_fs.GetFromFile('a.txt').Get()) | 
| 57     self.assertEqual(self._chained_compiled_fs.GetFromFile('a.txt').Get(), | 55     self.assertEqual(self._chained_compiled_fs.GetFromFile('a.txt').Get(), | 
| 58                      self._file_system.ReadSingle('a.txt').Get()) | 56                      self._new_file_system.ReadSingle('a.txt').Get()) | 
| 59 | 57 | 
| 60   def testGetFromFileListing(self): | 58   def testGetFromFileListing(self): | 
| 61     self.assertEqual(self._chained_compiled_fs.GetFromFile('dir/').Get(), | 59     self.assertEqual(self._chained_compiled_fs.GetFromFile('dir/').Get(), | 
| 62                      self._base_compiled_fs.GetFromFile('dir/').Get()) | 60                      self._base_compiled_fs.GetFromFile('dir/').Get()) | 
| 63     self._file_system.IncrementStat('dir/') | 61     self._new_file_system.IncrementStat('dir/') | 
| 64     self.assertNotEqual( | 62     self.assertNotEqual( | 
| 65         self._chained_compiled_fs.GetFromFileListing('dir/').Get(), | 63         self._chained_compiled_fs.GetFromFileListing('dir/').Get(), | 
| 66         self._base_compiled_fs.GetFromFileListing('dir/').Get()) | 64         self._base_compiled_fs.GetFromFileListing('dir/').Get()) | 
| 67     self.assertEqual( | 65     self.assertEqual( | 
| 68         self._chained_compiled_fs.GetFromFileListing('dir/').Get(), | 66         self._chained_compiled_fs.GetFromFileListing('dir/').Get(), | 
| 69         self._file_system.ReadSingle('dir/').Get()) | 67         self._new_file_system.ReadSingle('dir/').Get()) | 
| 70 | 68 | 
| 71 if __name__ == '__main__': | 69 if __name__ == '__main__': | 
| 72   unittest.main() | 70   unittest.main() | 
| OLD | NEW | 
|---|