| 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 |