Index: chrome/common/extensions/docs/server2/file_system_test.py |
diff --git a/chrome/common/extensions/docs/server2/file_system_test.py b/chrome/common/extensions/docs/server2/file_system_test.py |
index 116f12dcab87f05105d4491c065b0b8079dd2958..158f9a49ec6d524c1b6ff6aacf820a6b4e7c376f 100755 |
--- a/chrome/common/extensions/docs/server2/file_system_test.py |
+++ b/chrome/common/extensions/docs/server2/file_system_test.py |
@@ -59,6 +59,31 @@ class FileSystemTest(unittest.TestCase): |
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, []) |
not at google - send to devlin
2014/08/21 21:39:58
For all of these, the expectation should go first.
|
+ |
+ for root, dirs, files in file_system.Walk('', depth=1): |
+ all_dirs.extend(dirs) |
+ all_files.extend(files) |
+ self.assertEqual(all_dirs, ['templates/']) |
+ self.assertEqual(all_files, ['file.txt']) |
+ |
+ 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(all_dirs), |
+ sorted(['templates/', 'public/', 'json/'])) |
+ self.assertEqual(sorted(all_files), sorted(['file.txt', 'README'])) |
+ |
+ |
def testSubWalk(self): |
expected_files = set([ |
'/redirects.json', |