Index: chrome/common/extensions/docs/server2/availability_finder_test.py |
diff --git a/chrome/common/extensions/docs/server2/availability_finder_test.py b/chrome/common/extensions/docs/server2/availability_finder_test.py |
index 0cfc373bec7e0211ae6ea8f66f0983d5b5821df2..1540f36378ed0d5c171fc069d70a3dc1e3840a96 100755 |
--- a/chrome/common/extensions/docs/server2/availability_finder_test.py |
+++ b/chrome/common/extensions/docs/server2/availability_finder_test.py |
@@ -20,12 +20,15 @@ from test_data.canned_data import (CANNED_API_FILE_SYSTEM_DATA, CANNED_BRANCHES) |
from test_data.object_level_availability.tabs import TABS_SCHEMA_BRANCHES |
-class FakeHostFileSystemCreator(object): |
+class FakeHostFileSystemProvider(object): |
def __init__(self, file_system_data): |
self._file_system_data = file_system_data |
- def Create(self, branch): |
+ def GetTrunk(self): |
+ return self.GetBranch('trunk') |
+ |
+ def GetBranch(self, branch): |
return TestFileSystem(self._file_system_data[str(branch)]) |
@@ -39,15 +42,12 @@ class AvailabilityFinderTest(unittest.TestCase): |
ObjectStoreCreator.ForTest()) |
def create_availability_finder(file_system_data): |
- fake_host_fs_creator = FakeHostFileSystemCreator(file_system_data) |
- fake_host_fs = fake_host_fs_creator.Create('trunk') |
- return AvailabilityFinder(HostFileSystemIterator( |
- fake_host_fs_creator, |
- fake_host_fs, |
- self._branch_utility), |
+ fake_host_fs_creator = FakeHostFileSystemProvider(file_system_data) |
+ return AvailabilityFinder(HostFileSystemIterator(fake_host_fs_creator, |
+ self._branch_utility), |
ObjectStoreCreator.ForTest(), |
self._branch_utility, |
- fake_host_fs) |
+ fake_host_fs_creator.GetTrunk()) |
self._avail_finder = create_availability_finder(CANNED_API_FILE_SYSTEM_DATA) |
self._node_avail_finder = create_availability_finder(TABS_SCHEMA_BRANCHES) |