Die build.py, Die: Part 2 (LocalFetcher, Handlebar support, build script)

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

Index: chrome/common/extensions/docs/server2/resource_fetcher_test.py
diff --git a/chrome/common/extensions/docs/server2/resource_fetcher_test.py b/chrome/common/extensions/docs/server2/resource_fetcher_test.py
index f671bf92e51c93259a65c279ac3faae1d3c45a01..84c782597fd4c9f309e4623ecbf83f539e736a19 100755
--- a/chrome/common/extensions/docs/server2/resource_fetcher_test.py
+++ b/chrome/common/extensions/docs/server2/resource_fetcher_test.py
@@ -6,7 +6,7 @@
 import unittest
 import test_urlfetch
 
-from resource_fetcher import SubversionFetcher
+from resource_fetcher import SubversionFetcher, LocalFetcher

[Aaron Boodman, 2012/06/02 07:50:22]

Imports should be on separate lines:

http://google-styleguide.googlecode.com/svn/trunk/pyguide.html#Imports_format...

[cduvall, 2012/06/02 19:12:05]

Done.

 
 class TestSubversionFetcher(SubversionFetcher):
   def _GetURLFromBranch(self, branch):
@@ -16,13 +16,24 @@ class TestSubversionFetcher(SubversionFetcher):
 
 class SubversionFetcherTest(unittest.TestCase):
   def testFetchResource(self):
-    s_fetcher = TestSubversionFetcher(test_urlfetch)
+    fetcher = TestSubversionFetcher(test_urlfetch)
     self.assertEquals('trunk test\n',
-        s_fetcher.FetchResource('trunk', 'test.json').content)
+        fetcher.FetchResource('trunk', 'test.txt').content)
     self.assertEquals('branch1 test\n',
-        s_fetcher.FetchResource('branch1', 'test.json').content)
+        fetcher.FetchResource('branch1', 'test.txt').content)
     self.assertEquals('branch2 test\n',
-        s_fetcher.FetchResource('branch2', 'test.json').content)
+        fetcher.FetchResource('branch2', 'test.txt').content)
+
+class LocalFetcherTest(unittest.TestCase):
+  def testFetchResource(self):
+    fetcher = LocalFetcher()
+    local_path = 'test_data/local_fetcher/'
+    self.assertEquals('test1\n',
+        fetcher.FetchResource('', local_path + 'test1.txt').content)
+    self.assertEquals('test2\n',
+        fetcher.FetchResource('', local_path + 'test2.txt').content)
+    self.assertEquals('test3\n',
+        fetcher.FetchResource('junk', local_path + 'test3.txt').content)
 
 if __name__ == '__main__':
   unittest.main()