Docserver: Generate a table of extension/app API owners

chrome/common/extensions/docs/server2/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 unittest
 
 from test_file_system import TestFileSystem
 
 file_system = TestFileSystem({
@@ skipping 41 matching lines @@
     all_files = []
     all_dirs = []
     for root, dirs, files in file_system.Walk(''):
       if not root: root = '^'
       all_files += [root + '/' + name for name in files]
       all_dirs += [root + '/' + name for name in dirs]
 
     self.assertEqual(sorted(expected_files), sorted(all_files))
     self.assertEqual(sorted(expected_dirs), sorted(all_dirs))
 
+  def testWalkDepth(self):
+    all_dirs = []
+    all_files = []
+    for root, dirs, files in file_system.Walk('', depth=0):
+      all_dirs.extend(dirs)
+      all_files.extend(files)
+    self.assertEqual([], all_dirs)
+    self.assertEqual([], all_files)
+
+    for root, dirs, files in file_system.Walk('', depth=1):
+      all_dirs.extend(dirs)
+      all_files.extend(files)
+    self.assertEqual(['templates/'], all_dirs)
+    self.assertEqual(['file.txt'], all_files)
+
+    all_dirs = []
+    all_files = []
+    for root, dirs, files in file_system.Walk('', depth=2):
+      all_dirs.extend(dirs)
+      all_files.extend(files)
+    self.assertEqual(sorted(['templates/', 'public/', 'json/']),
+                     sorted(all_dirs))
+    self.assertEqual(sorted(['file.txt', 'README']), sorted(all_files))
+
+
   def testSubWalk(self):
     expected_files = set([
       '/redirects.json',
       'apps/404.html',
       'apps/a11y.html',
       'extensions/404.html',
       'extensions/cookies.html'
     ])
 
     all_files = set()
@@ skipping 23 matching lines @@
     self.assertTrue(exists('templates/README'))
     self.assertFalse(exists('templates/README/'))
     self.assertTrue(exists('templates/public/redirects.json'))
     self.assertFalse(exists('templates/public/redirects.json/'))
     self.assertTrue(exists('templates/public/apps/a11y.html'))
     self.assertFalse(exists('templates/public/apps/a11y.html/'))
 
 
 if __name__ == '__main__':
   unittest.main()