Docserver: Make MockFileSystem not iterate over the entire file system as part

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

 #!/usr/bin/env python
 # Copyright 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import functools
 import os
 
 from appengine_wrappers import GetAppVersion
 from compiled_file_system import CompiledFileSystem
@@ skipping 65 matching lines @@
         compiled_fs_creator(f, CompiledFileSystemTest, category='foo'))
 
   def testPopulateFromFile(self):
     def Sleepy(key, val):
       return '%s%s' % ('Z' * len(key), 'z' * len(val))
     compiled_fs = _GetTestCompiledFsCreator()(Sleepy, CompiledFileSystemTest)
     self.assertEqual('ZZZZZZZZzzzzzzzzzzzzzzzzz',
                      compiled_fs.GetFromFile('404.html').Get())
     self.assertEqual('ZZZZZZZZZZZZZZzzzzzzzzzzzzzzzzzz',
                      compiled_fs.GetFromFile('apps/a11y.html').Get())
-    self.assertEqual('ZZZZZZZZZZZZZZZZZZZZZZZzzzzzzzzzzzzzzzzzz',
-                     compiled_fs.GetFromFile('/apps/fakedir/file.html').Get())
+    self.assertEqual('ZZZZZZZZZZZZZZZZZZZZZZzzzzzzzzzzzzzzzzzz',
+                     compiled_fs.GetFromFile('apps/fakedir/file.html').Get())
 
   def testPopulateFromFileListing(self):
     def strip_ext(path, files):
       return [os.path.splitext(f)[0] for f in files]
     compiled_fs = _GetTestCompiledFsCreator()(strip_ext, CompiledFileSystemTest)
     expected_top_listing = [
       '404',
       'apps/a11y',
       'apps/about_apps',
       'apps/deepdir/deeper/deepest',
       'apps/deepdir/deepfile',
       'apps/fakedir/file',
       'extensions/activeTab',
       'extensions/alarms'
     ]
     self.assertEqual(expected_top_listing,
-                     sorted(compiled_fs.GetFromFileListing('/').Get()))
-    self.assertEqual(expected_top_listing,
                      sorted(compiled_fs.GetFromFileListing('').Get()))
     expected_apps_listing = [
       'a11y',
       'about_apps',
       'deepdir/deeper/deepest',
       'deepdir/deepfile',
       'fakedir/file',
     ]
     self.assertEqual(expected_apps_listing,
-                     sorted(compiled_fs.GetFromFileListing('/apps/').Get()))
-    self.assertEqual(expected_apps_listing,
                      sorted(compiled_fs.GetFromFileListing('apps/').Get()))
     self.assertEqual(['file',],
-                     compiled_fs.GetFromFileListing('/apps/fakedir/').Get())
-    self.assertEqual(['file',],
                      compiled_fs.GetFromFileListing('apps/fakedir/').Get())
     self.assertEqual(['deeper/deepest', 'deepfile'],
                      sorted(compiled_fs.GetFromFileListing(
-                        '/apps/deepdir/').Get()))
-    self.assertEqual(['deeper/deepest', 'deepfile'],
-                     sorted(compiled_fs.GetFromFileListing(
                          'apps/deepdir/').Get()))
     self.assertEqual(['deepest'],
                      compiled_fs.GetFromFileListing(
-                         '/apps/deepdir/deeper/').Get())
-    self.assertEqual(['deepest'],
-                     compiled_fs.GetFromFileListing(
                          'apps/deepdir/deeper/').Get())
 
   def testCaching(self):
     compiled_fs = _GetTestCompiledFsCreator()(identity, CompiledFileSystemTest)
     self.assertEqual('404.html contents',
                      compiled_fs.GetFromFile('404.html').Get())
     self.assertEqual(set(('file.html',)),
                      set(compiled_fs.GetFromFileListing('apps/fakedir').Get()))
 
-    compiled_fs._file_system._obj['404.html'] = 'boom'
-    compiled_fs._file_system._obj['apps']['fakedir']['boom.html'] = 'blam'
+    compiled_fs._file_system._path_values['404.html'] = 'boom'
+    compiled_fs._file_system._path_values['apps/fakedir/'] = [
+        'file.html', 'boom.html']
     self.assertEqual('404.html contents',
                      compiled_fs.GetFromFile('404.html').Get())
     self.assertEqual(set(('file.html',)),
                      set(compiled_fs.GetFromFileListing('apps/fakedir').Get()))
 
     compiled_fs._file_system.IncrementStat()
     self.assertEqual('boom', compiled_fs.GetFromFile('404.html').Get())
     self.assertEqual(set(('file.html', 'boom.html')),
                      set(compiled_fs.GetFromFileListing('apps/fakedir').Get()))
 
@@ skipping 43 matching lines @@
     # determining that ahead of time (though perhaps the API could give some
     # kind of clue, if we really cared).
     future = compiled_fs.GetFromFileListing('extensions/')
     self.assertTrue(*mock_fs.CheckAndReset(stat_count=1,
                                            read_count=1,
                                            read_resolve_count=1))
     future.Get()
     self.assertTrue(*mock_fs.CheckAndReset())
 
     # Similar configuration to the 'apps/' case but deeper.
-    future = compiled_fs.GetFromFileListing('/')
+    future = compiled_fs.GetFromFileListing('')
     self.assertTrue(*mock_fs.CheckAndReset(stat_count=1,
                                            read_count=2,
                                            read_resolve_count=1))
     future.Get()
     self.assertTrue(*mock_fs.CheckAndReset(read_count=2, read_resolve_count=3))
 
 
 
 if __name__ == '__main__':
   unittest.main()