Docserver: Cleanup to remove the APIListDataSource dependency from

chrome/common/extensions/docs/server2/reference_resolver_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 json
 import os
 import sys
 import unittest
 
 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
 
   def get(self, key, disable_refs=False):
     assert disable_refs, 'ReferenceResolve should be disabling refs'
     if key not in self._json:
       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):
     self._base_path = os.path.join(sys.path[0], 'test_data', 'test_json')
 
   def _ReadLocalFile(self, filename):
     with open(os.path.join(self._base_path, filename), 'r') as f:
       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',
       'text': 'foo',
       'name': 'foo'
     }, resolver.GetLink('foo', namespace='baz'))
     self.assertEqual({
       'href': 'foo.html#type-foo_t1',
       'text': 'foo.foo_t1',
       'name': 'foo_t1'
@@ skipping 108 matching lines @@
                                  namespace='baz'))
 
     # Allow bare "$ref:foo.bar." at the end of a string.
     self.assertEqual(
         '<a href="bar.html#type-bon">bar.bon</a>.',
         resolver.ResolveAllLinks('$ref:bar.bon.',
                                  namespace='baz'))
 
 if __name__ == '__main__':
   unittest.main()