| OLD | NEW |
| 1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
| 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 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 os | 6 import os |
| 7 import unittest | 7 import unittest |
| 8 | 8 |
| 9 import appengine_memcache as memcache | 9 import appengine_memcache as memcache |
| 10 from in_memory_memcache import InMemoryMemcache | 10 from in_memory_memcache import InMemoryMemcache |
| 11 from local_file_system import LocalFileSystem | 11 from local_file_system import LocalFileSystem |
| 12 from memcache_file_system import MemcacheFileSystem | 12 from memcache_file_system import MemcacheFileSystem |
| 13 | 13 |
| 14 class LocalFileSystemTest(unittest.TestCase): | 14 class LocalFileSystemTest(unittest.TestCase): |
| 15 def setUp(self): | 15 def setUp(self): |
| 16 self._memcache = InMemoryMemcache() | 16 self._memcache = InMemoryMemcache() |
| 17 self._file_system = MemcacheFileSystem( | 17 self._file_system = MemcacheFileSystem( |
| 18 LocalFileSystem(os.path.join('test_data', 'file_system')), | 18 LocalFileSystem(os.path.join('test_data', 'file_system'), False), |
| 19 self._memcache) | 19 self._memcache) |
| 20 | 20 |
| 21 def testReadFiles(self): | 21 def testReadFiles(self): |
| 22 expected = { | 22 expected = { |
| 23 'test1.txt': 'test1\n', | 23 'test1.txt': 'test1\n', |
| 24 'test2.txt': 'test2\n', | 24 'test2.txt': 'test2\n', |
| 25 'test3.txt': 'test3\n', | 25 'test3.txt': 'test3\n', |
| 26 } | 26 } |
| 27 self.assertEqual( | 27 self.assertEqual( |
| 28 expected, | 28 expected, |
| 29 self._file_system.Read(['test1.txt', 'test2.txt', 'test3.txt']).Get()) | 29 self._file_system.Read(['test1.txt', 'test2.txt', 'test3.txt']).Get()) |
| 30 | 30 |
| 31 def testListDir(self): | 31 def testListDir(self): |
| 32 expected = ['dir/'] | 32 expected = ['dir/'] |
| 33 for i in range(7): | 33 for i in range(7): |
| 34 expected.append('file%d.html' % i) | 34 expected.append('file%d.html' % i) |
| 35 self.assertEqual( | 35 self.assertEqual( |
| 36 expected, | 36 expected, |
| 37 sorted(self._file_system.Read(['list/']).Get()['list/'])) | 37 sorted(self._file_system.Read(['list/']).Get()['list/'])) |
| 38 expected.remove('file0.html') | 38 expected.remove('file0.html') |
| 39 self._memcache.Set('list/', | 39 self._memcache.Set('list/', |
| 40 (expected, 0), | 40 (expected, 0), |
| 41 memcache.MEMCACHE_FILE_SYSTEM_READ) | 41 memcache.MEMCACHE_FILE_SYSTEM_READ) |
| 42 self.assertEqual( | 42 self.assertEqual( |
| 43 expected, | 43 expected, |
| 44 sorted(self._file_system.Read(['list/']).Get()['list/'])) | 44 sorted(self._file_system.Read(['list/']).Get()['list/'])) |
| 45 | 45 |
| 46 if __name__ == '__main__': | 46 if __name__ == '__main__': |
| 47 unittest.main() | 47 unittest.main() |
| OLD | NEW |