Extensions Docs Server: Samples zip files

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

Index: chrome/common/extensions/docs/server2/local_fetcher_test.py
diff --git a/chrome/common/extensions/docs/server2/local_fetcher_test.py b/chrome/common/extensions/docs/server2/local_fetcher_test.py
index 479d7f90b582b586b14efaf3ab38bfe9cf2549c9..0990dac2d8071582964b96491ace0b177737a4bd 100755
--- a/chrome/common/extensions/docs/server2/local_fetcher_test.py
+++ b/chrome/common/extensions/docs/server2/local_fetcher_test.py
@@ -9,15 +9,29 @@ import unittest
 from local_fetcher import LocalFetcher
 
 class LocalFetcherTest(unittest.TestCase):
+  def setUp(self):
+    self._base_path = os.path.join('test_data', 'local_fetcher')
+    self._fetcher = LocalFetcher(self._base_path)
+
   def testFetchResource(self):
-    fetcher = LocalFetcher('')
-    local_path = os.path.join('test_data', 'local_fetcher')
-    self.assertEquals('test1\n', fetcher.FetchResource(
-        os.path.join(local_path, 'test1.txt')).content)
-    self.assertEquals('test2\n', fetcher.FetchResource(
-        os.path.join(local_path, 'test2.txt')).content)
-    self.assertEquals('test3\n', fetcher.FetchResource(
-        os.path.join(local_path, 'test3.txt')).content)
+    self.assertEquals('test1\n',
+                      self._fetcher.FetchResource('test1.txt').content)
+    self.assertEquals('test2\n',
+                      self._fetcher.FetchResource('test2.txt').content)
+    self.assertEquals('test3\n',
+                      self._fetcher.FetchResource('test3.txt').content)
+
+  def testListDirectory(self):
+    expected = ['file%d.html' % i for i in range(7)]
+    self.assertEquals(
+        expected,
+        sorted(self._fetcher.ListDirectory('recursive_list/list').content))
+    expected2 = ['recursive_list/' + x for x in expected]
+    expected2.extend(['recursive_list/list/' + x for x in expected])
+    self.assertEquals(
+        expected2,
+        sorted(self._fetcher.ListDirectory('recursive_list',
+                                    recursive=True).content))
 
 if __name__ == '__main__':
   unittest.main()