Index: chrome/common/extensions/docs/server2/reference_resolver_test.py |
diff --git a/chrome/common/extensions/docs/server2/reference_resolver_test.py b/chrome/common/extensions/docs/server2/reference_resolver_test.py |
index 8a478577a6ced0b09a47678d83c0ec9666c48a08..3cbcea32f06bf71ea988c9c7c6e648b3e4340aa0 100755 |
--- a/chrome/common/extensions/docs/server2/reference_resolver_test.py |
+++ b/chrome/common/extensions/docs/server2/reference_resolver_test.py |
@@ -12,6 +12,7 @@ from file_system import FileNotFoundError |
from reference_resolver import ReferenceResolver |
from test_object_store import TestObjectStore |
+ |
class FakeAPIDataSource(object): |
def __init__(self, json_data): |
self._json = json_data |
@@ -22,8 +23,14 @@ class FakeAPIDataSource(object): |
raise FileNotFoundError(key) |
return self._json[key] |
- def GetAllNames(self): |
- return self._json.keys() |
+ |
+class FakeAPIModels(object): |
+ def __init__(self, names): |
+ self._names = names |
+ |
+ def GetNames(self): |
+ return self._names |
+ |
class APIDataSourceTest(unittest.TestCase): |
def setUp(self): |
@@ -34,10 +41,9 @@ class APIDataSourceTest(unittest.TestCase): |
return f.read() |
def testGetLink(self): |
- data_source = FakeAPIDataSource( |
- json.loads(self._ReadLocalFile('fake_data_source.json'))) |
- resolver = ReferenceResolver(data_source, |
- data_source, |
+ test_data = json.loads(self._ReadLocalFile('fake_data_source.json')) |
+ resolver = ReferenceResolver(FakeAPIDataSource(test_data), |
+ FakeAPIModels(test_data.keys()), |
TestObjectStore('test')) |
self.assertEqual({ |
'href': 'foo.html', |