Extensions Docs Server: Internal file system

chrome/common/extensions/docs/server2/branch_utility_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 branch_utility
+from fake_url_fetcher import FakeUrlFetcher
 import unittest
-import test_urlfetch
 
 class BranchUtilityTest(unittest.TestCase):
+  def setUp(self):
+    self._fetcher = FakeUrlFetcher('test_data')
+
   def testSplitChannelNameFromPath(self):
     self.assertEquals(('dev', 'hello/stuff.html'),
                       branch_utility.SplitChannelNameFromPath(
                       'dev/hello/stuff.html'))
     self.assertEquals(('beta', 'hello/stuff.html'),
                       branch_utility.SplitChannelNameFromPath(
                       'beta/hello/stuff.html'))
     self.assertEquals(('trunk', 'hello/stuff.html'),
                       branch_utility.SplitChannelNameFromPath(
                       'trunk/hello/stuff.html'))
     self.assertEquals(('stable', 'hello/stuff.html'),
                       branch_utility.SplitChannelNameFromPath(
                       'hello/stuff.html'))
     self.assertEquals(('stable', 'hello/dev/stuff.html'),
                       branch_utility.SplitChannelNameFromPath(
                       'hello/dev/stuff.html'))
 
   def testGetBranchNumberForChannelName(self):
     base_path = 'branch_utility/first.json'
     self.assertEquals('1132',
         branch_utility.GetBranchNumberForChannelName('dev',
-                                                     test_urlfetch,
+                                                     self._fetcher,
                                                      base_path))
     self.assertEquals('1084',
         branch_utility.GetBranchNumberForChannelName('beta',
-                                                     test_urlfetch,
+                                                     self._fetcher,
                                                      base_path))
     self.assertEquals('1234',
         branch_utility.GetBranchNumberForChannelName('stable',
-                                                     test_urlfetch,
+                                                     self._fetcher,
                                                      base_path))
     self.assertEquals('trunk',
         branch_utility.GetBranchNumberForChannelName('trunk',
-                                                     test_urlfetch,
+                                                     self._fetcher,
                                                      base_path))
 
 if __name__ == '__main__':
   unittest.main()