Docserver: Cache the samples.

chrome/common/extensions/docs/server2/jsc_view_test.py

 #!/usr/bin/env python
 # Copyright 2014 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 unittest
 
 from jsc_view import GetEventByNameFromEvents
 from api_schema_graph import APISchemaGraph
 from availability_finder import AvailabilityFinder, AvailabilityInfo
 from branch_utility import BranchUtility, ChannelInfo
 from compiled_file_system import CompiledFileSystem
 from extensions_paths import CHROME_EXTENSIONS
 from fake_host_file_system_provider import FakeHostFileSystemProvider
 from fake_url_fetcher import FakeUrlFetcher
 from features_bundle import FeaturesBundle
 from future import Future
 from host_file_system_iterator import HostFileSystemIterator
-from jsc_view import JSCView, _FormatValue
+from jsc_view import CreateJSCView, _JSCViewBuilder, _FormatValue
 from object_store_creator import ObjectStoreCreator
 from schema_processor import SchemaProcessorFactoryForTest
+from servlet import Request
 from server_instance import ServerInstance
 from test_data.api_data_source.canned_master_fs import CANNED_MASTER_FS_DATA
 from test_data.canned_data import CANNED_API_FILE_SYSTEM_DATA
 from test_data.object_level_availability.tabs import TABS_SCHEMA_BRANCHES
 from test_file_system import TestFileSystem
 from test_util import Server2Path
 
 
 class _FakeTemplateCache(object):
 
@@ skipping 77 matching lines @@
     events['types'][0]['functions'].append(add_rules)
     self.assertEqual(add_rules,
                      GetEventByNameFromEvents(events)['addRules'])
 
     events['types'][0]['functions'].append(add_rules)
     # Duplicates are an error.
     self.assertRaises(AssertionError, GetEventByNameFromEvents, events)
 
   def testCreateId(self):
     fake_avail_finder = _FakeAvailabilityFinder(self._fake_availability)
-    dict_ = JSCView(self._api_models.GetContentScriptAPIs().Get(),
-                    self._api_models.GetModel('tester').Get(),
-                    fake_avail_finder,
-                    self._json_cache,
-                    _FakeTemplateCache(),
-                    self._features_bundle,
-                    None,
-                    'extensions').ToDict()
+    dict_ = CreateJSCView(
+        self._api_models.GetContentScriptAPIs().Get(),
+        self._api_models.GetModel('tester').Get(),
+        fake_avail_finder,
+        self._json_cache,
+        _FakeTemplateCache(),
+        self._features_bundle,
+        None,
+        'extensions',
+        [],
+        Request.ForTest(''))
     self.assertEquals('type-TypeA', dict_['types'][0]['id'])
     self.assertEquals('property-TypeA-b',
                       dict_['types'][0]['properties'][0]['id'])
     self.assertEquals('method-get', dict_['functions'][0]['id'])
     self.assertEquals('event-EventA', dict_['events'][0]['id'])
 
   # TODO(kalman): re-enable this when we have a rebase option.
   def DISABLED_testToDict(self):
     fake_avail_finder = _FakeAvailabilityFinder(self._fake_availability)
     expected_json = self._LoadJSON('expected_tester.json')
-    dict_ = JSCView(self._api_models.GetContentScriptAPIs().Get(),
-                    self._api_models.GetModel('tester').Get(),
-                    fake_avail_finder,
-                    self._json_cache,
-                    _FakeTemplateCache(),
-                    self._features_bundle,
-                    None,
-                    'extensions').ToDict()
+    dict_ = CreateJSCView(
+        self._api_models.GetContentScriptAPIs().Get(),
+        self._api_models.GetModel('tester').Get(),
+        fake_avail_finder,
+        self._json_cache,
+        _FakeTemplateCache(),
+        self._features_bundle,
+        None,
+        'extensions',
+        [],
+        Request.ForTest(''))
     self.assertEquals(expected_json, dict_)
 
   def testAddRules(self):
     fake_avail_finder = _FakeAvailabilityFinder(self._fake_availability)
-    dict_ = JSCView(self._api_models.GetContentScriptAPIs().Get(),
-                    self._api_models.GetModel('add_rules_tester').Get(),
-                    fake_avail_finder,
-                    self._json_cache,
-                    _FakeTemplateCache(),
-                    self._features_bundle,
-                    self._FakeLoadAddRulesSchema(),
-                    'extensions').ToDict()
+    dict_ = CreateJSCView(
+        self._api_models.GetContentScriptAPIs().Get(),
+        self._api_models.GetModel('add_rules_tester').Get(),
+        fake_avail_finder,
+        self._json_cache,
+        _FakeTemplateCache(),
+        self._features_bundle,
+        self._FakeLoadAddRulesSchema(),
+        'extensions',
+        [],
+        Request.ForTest(''))
 
     # 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'][
                           'parameters'][0]['name'])
 
     # Check that the second event has addListener defined.
     self.assertEquals('noRules', dict_['events'][1]['name'])
     self.assertEquals('add_rules_tester', dict_['name'])
     self.assertEquals('noRules', dict_['events'][1]['name'])
     self.assertEquals('callback',
                       dict_['events'][0]['byName']['addListener'][
                           'parameters'][0]['name'])
 
   def testGetIntroList(self):
     fake_avail_finder = _FakeAvailabilityFinder(self._fake_availability)
-    model = JSCView(self._api_models.GetContentScriptAPIs().Get(),
-                    self._api_models.GetModel('tester').Get(),
-                    fake_avail_finder,
-                    self._json_cache,
-                    _FakeTemplateCache(),
-                    self._features_bundle,
-                    None,
-                    'extensions')
+    model = _JSCViewBuilder(
+        self._api_models.GetContentScriptAPIs().Get(),
+        self._api_models.GetModel('tester').Get(),
+        fake_avail_finder,
+        self._json_cache,
+        _FakeTemplateCache(),
+        self._features_bundle,
+        None,
+        'extensions',
+        [])
     expected_list = [
       { 'title': 'Description',
         'content': [
           { 'text': 'a test api' }
         ]
       },
       { 'title': 'Availability',
         'content': [
           { 'partial': 'motemplate chrome/common/extensions/docs/' +
                        'templates/private/intro_tables/stable_message.html',
@@ skipping 35 matching lines @@
             'text': 'Welcome!'
           }
         ]
       }
     ]
     self.assertEquals(model._GetIntroTableList(), expected_list)
 
     # Tests the same data with a scheduled availability.
     fake_avail_finder = _FakeAvailabilityFinder(
         AvailabilityInfo(ChannelInfo('beta', '1453', 27), scheduled=28))
-    model = JSCView(self._api_models.GetContentScriptAPIs().Get(),
-                    self._api_models.GetModel('tester').Get(),
-                    fake_avail_finder,
-                    self._json_cache,
-                    _FakeTemplateCache(),
-                    self._features_bundle,
-                    None,
-                    'extensions')
+    model = _JSCViewBuilder(
+        self._api_models.GetContentScriptAPIs().Get(),
+        self._api_models.GetModel('tester').Get(),
+        fake_avail_finder,
+        self._json_cache,
+        _FakeTemplateCache(),
+        self._features_bundle,
+        None,
+        'extensions',
+        [])
     expected_list[1] = {
       'title': 'Availability',
       'content': [
         { 'partial': 'motemplate chrome/common/extensions/docs/' +
                      'templates/private/intro_tables/beta_message.html',
           'version': 27,
           'scheduled': 28
         }
       ]
     }
@@ skipping 16 matching lines @@
   def testGetAPIAvailability(self):
     api_availabilities = {
       'bluetooth': 31,
       'contextMenus': 'master',
       'jsonStableAPI': 20,
       'idle': 5,
       'input.ime': 18,
       'tabs': 18
     }
     for api_name, availability in api_availabilities.iteritems():
-      model_dict = JSCView(
+      model_dict = CreateJSCView(
           self._api_models.GetContentScriptAPIs().Get(),
           self._api_models.GetModel(api_name).Get(),
           self._avail_finder,
           self._json_cache,
           _FakeTemplateCache(),
           _FakeFeaturesBundle(),
           None,
-          'extensions').ToDict()
+          'extensions',
+          [],
+          Request.ForTest(''))
       self.assertEquals(availability,
                         model_dict['introList'][1]['content'][0]['version'])
 
 
 class JSCViewWithNodeAvailabilityTest(unittest.TestCase):
   def setUp(self):
     tabs_unmodified_versions = (16, 20, 23, 24)
     self._branch_utility = BranchUtility(
         os.path.join('branch_utility', 'first.json'),
         os.path.join('branch_utility', 'second.json'),
@@ skipping 47 matching lines @@
           'tabs.fakeTabsProperty1': None,
           'tabs.get': None,
           'tabs.onUpdated': None,
           'tabs.InjectDetails': 25,
           'tabs.fakeTabsProperty2': 15,
           'tabs.getCurrent': 19,
           'tabs.onActivated': 30
       }
       self.assertEquals(node_availabilities[node], actual)
 
-    model_dict = JSCView(
+    model_dict = CreateJSCView(
         self._api_models.GetContentScriptAPIs().Get(),
         self._api_models.GetModel('tabs').Get(),
         self._avail_finder,
         self._json_cache,
         _FakeTemplateCache(),
         _FakeFeaturesBundle(),
         None,
-        'extensions').ToDict()
+        'extensions',
+        [],
+        Request.ForTest(''))
 
     # Test nodes that have the same availability as their parent.
 
     # Test type.
     assertEquals('tabs.Tab', model_dict['types'][0]['availability'])
     # Test property.
     assertEquals('tabs.fakeTabsProperty1',
                  model_dict['properties'][0]['availability'])
     # Test function.
     assertEquals('tabs.get', model_dict['functions'][1]['availability'])
@@ skipping 18 matching lines @@
     # Test a node that became deprecated.
     self.assertEquals({
       'scheduled': None,
       'version': 26,
       'partial': 'motemplate chrome/common/extensions/docs/templates/' +
           'private/intro_tables/deprecated_message.html'
       }, model_dict['types'][2]['availability'])
 
 if __name__ == '__main__':
   unittest.main()