Index: chrome/common/extensions/docs/server2/patch_servlet_test.py |
diff --git a/chrome/common/extensions/docs/server2/patch_servlet_test.py b/chrome/common/extensions/docs/server2/patch_servlet_test.py |
index df2c74809a2c9d3c4e9000f02e0adbeeae6381bf..731da3aaee9eaff4aae451563b51851a085438ac 100755 |
--- a/chrome/common/extensions/docs/server2/patch_servlet_test.py |
+++ b/chrome/common/extensions/docs/server2/patch_servlet_test.py |
@@ -12,6 +12,7 @@ from patch_servlet import PatchServlet |
from render_servlet import RenderServlet |
from server_instance import ServerInstance |
from servlet import Request |
+from test_branch_utility import TestBranchUtility |
from test_util import DisableLogging |
_ALLOWED_HOST = 'https://chrome-apps-doc.appspot.com' |
@@ -24,7 +25,10 @@ class _PatchServletDelegate(RenderServlet.Delegate): |
def CreateAppSamplesFileSystem(self, object_store_creator): |
return EmptyDirFileSystem() |
- def CreateHostFileSystemForBranch(self, channel): |
+ def CreateBranchUtility(self, object_store_creator): |
+ return TestBranchUtility.CreateWithCannedData() |
+ |
+ def CreateHostFileSystem(self, branch, revision=None): |
return LocalFileSystem.Create() |
class PatchServletTest(unittest.TestCase): |