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

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

 #!/usr/bin/env python
 # Copyright (c) 2012 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
 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):
     if branch == 'trunk':
       return 'subversion_fetcher/trunk/'
     return 'subversion_fetcher/' + branch + '/'
 
 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()