Docserver: Add more support for object level availability in templates

chrome/common/extensions/docs/server2/api_data_source_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
 
@@ skipping 20 matching lines @@
 from test_util import Server2Path
 from third_party.json_schema_compiler.memoize import memoize
 
 
 class _FakeTemplateCache(object):
 
   def GetFromFile(self, key):
     return Future(value='handlebar %s' % key)
 
 
-class _FakeAvailabilityFinder(object):
-  def __init__(self, fake_availability):
-    self._fake_availability = fake_availability
-
-  def GetAPIAvailability(self, api_name):
-    return self._fake_availability
-
-  def GetAPINodeAvailability(self, api_name):
-    '''The tests that use this fake class don't
-    use the node availability, so just return a
-    dummy graph.
-    '''
-    return APISchemaGraph(_graph={'dummy': 'graph'})
-
-
 class _FakeFeaturesBundle(object):
   def GetAPIFeatures(self):
     return Future(value={
       'bluetooth': {'value': True},
       'contextMenus': {'value': True},
       'jsonStableAPI': {'value': True},
       'idle': {'value': True},
       'input.ime': {'value': True},
       'tabs': {'value': True}
     })
 
 
 class _FakeAvailabilityFinder(object):
   def __init__(self, fake_availability):
     self._fake_availability = fake_availability
 
   def GetAPIAvailability(self, api_name):
     return self._fake_availability
 
   def GetAPINodeAvailability(self, api_name):
-    '''The tests that use this fake class don't
-    use the node availability, so just return a
-    dummy graph.
-    '''
-    return APISchemaGraph(_graph={'dummy': 'graph'})
+    schema_graph = APISchemaGraph()
+    api_graph = APISchemaGraph(json.loads(
+        CANNED_TRUNK_FS_DATA['api'][api_name + '.json']))
+    # Give the graph fake ChannelInfo; it's not used in tests.
+    schema_graph.Update(api_graph, annotation=ChannelInfo('stable', '28', 28))
+    return schema_graph
 
 
 class APIDataSourceTest(unittest.TestCase):
   def setUp(self):
     self._base_path = Server2Path('test_data', 'test_json')
 
     server_instance = ServerInstance.ForTest(
         TestFileSystem(CANNED_TRUNK_FS_DATA, relative_to=CHROME_EXTENSIONS))
     file_system = server_instance.host_file_system_provider.GetTrunk()
     self._json_cache = server_instance.compiled_fs_factory.ForJson(file_system)
@@ skipping 64 matching lines @@
                       self._json_cache,
                       _FakeTemplateCache(),
                       self._features_bundle,
                       None).ToDict()
     self.assertEquals(expected_json, dict_)
 
   def testAddRules(self):
     fake_avail_finder = _FakeAvailabilityFinder(self._fake_availability)
     dict_ = _JSCModel(self._api_models.GetModel('add_rules_tester').Get(),
                       fake_avail_finder,
-                      #self._fake_availability,
                       self._json_cache,
                       _FakeTemplateCache(),
                       self._features_bundle,
                       self._FakeLoadAddRulesSchema()).ToDict()
 
     # Check that the first event has the addRulesFunction defined.
     self.assertEquals('add_rules_tester', dict_['name'])
     self.assertEquals('rules', dict_['events'][0]['name'])
     self.assertEquals('notable_name_to_check_for',
                       dict_['events'][0]['byName']['addRules'][
@@ skipping 177 matching lines @@
 
     # Test nodes that have the same availability as their parent.
     assertEquals('tabs.get', model_dict['functions'][1]['availability'])
     # Test nodes with varying availabilities.
     assertEquals('tabs.getCurrent',
                  model_dict['functions'][0]['availability']['version'])
 
 
 if __name__ == '__main__':
   unittest.main()