| 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..d747024748a140d3bc46cddbd994b958f1ba2644 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)
|
| +
|
| + 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',
|
|
|