Inlining references to internal APIs.

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

 #!/usr/bin/env python
 # Copyright 2013 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 unittest
 from copy import deepcopy
 
-from schema_util import RemoveNoDocs, DetectInlineableTypes, InlineDocs
+from process_schema import ProcessSchema
+from future import Future
+from object_store_creator import ObjectStoreCreator
+from host_file_system_provider import HostFileSystemProvider
+from compiled_file_system import CompiledFileSystem
 
+class _FakeReferenceResolver():
+  def GetRefModel(self, ref, api_list):
+    return None, None
+
+class _FakeAPIModels():
+  def GetNames(self):
+    return []
+
+class _FakeFeaturesBundle():
+  def GetAPIFeatures(self):
+    return Future(value={})
 
 class SchemaUtilTest(unittest.TestCase):
-
   def testRemoveNoDocs(self):
     expected_nodoc = [
       {
         'name': 'B',
         'list': [
           {
             'name': 'B2'
           }
         ]
       },
@@ skipping 75 matching lines @@
             'name': 'E2'
           },
           {
             'name': 'E3',
             'nodoc': True
           }
         ]
       }
     ]
 
-    RemoveNoDocs(nodoc_data)
+    object_store_creator = ObjectStoreCreator(start_empty=False)
+    host_file_system_provider = HostFileSystemProvider(object_store_creator)
+    process_schema = ProcessSchema(_FakeReferenceResolver(),
+                                   _FakeAPIModels(),
+                                   _FakeFeaturesBundle(),
+                                   CompiledFileSystem.Factory(
+                                       object_store_creator),
+                                   host_file_system_provider.GetTrunk(),
+                                   True)
+    process_schema._RemoveNoDocs(nodoc_data)
     self.assertEquals(expected_nodoc, nodoc_data)
 
   def testInlineDocs(self):
     schema = {
       'namespace': 'storage',
       'properties': {
         'key2': {
           'description': 'second key',
           '$ref': 'Key'
         },
@@ skipping 42 matching lines @@
             'marker': True,
             'type': 'string'
           },
           'type': 'array',
           'id': 'KeyList',
           'description': 'A list of keys'
         }
       ]
     }
 
+    object_store_creator = ObjectStoreCreator(start_empty=False)
+    host_file_system_provider = HostFileSystemProvider(object_store_creator)
+    process_schema = ProcessSchema(_FakeReferenceResolver(),
+                                   _FakeAPIModels(),
+                                   _FakeFeaturesBundle(),
+                                   CompiledFileSystem.Factory(
+                                       object_store_creator),
+                                   host_file_system_provider.GetTrunk(),
+                                   False)
     inlined_schema = deepcopy(schema)
-    InlineDocs(inlined_schema, False)
+    process_schema._InlineDocs(inlined_schema)
     self.assertEqual(expected_schema, inlined_schema)
 
   def testDetectInline(self):
     schema = {
       'types': [
         {
           'id': 'Key',
           'items': {
             '$ref': 'Value'
           }
         },
         {
           'id': 'Value',
           'marker': True
         }
       ]
     }
 
     expected_schema = {
       'types': [
         {
           'id': 'Key',
           'items': {
             'marker': True,
           }
         }
       ]
     }
 
-    DetectInlineableTypes(schema)
-    InlineDocs(schema, False)
+    object_store_creator = ObjectStoreCreator(start_empty=False)
+    host_file_system_provider = HostFileSystemProvider(object_store_creator)
+    process_schema = ProcessSchema(_FakeReferenceResolver(),
+                                   _FakeAPIModels(),
+                                   _FakeFeaturesBundle(),
+                                   CompiledFileSystem.Factory(
+                                       object_store_creator),
+                                   host_file_system_provider.GetTrunk(),
+                                   False)
+    process_schema._DetectInlineableTypes(schema)
+    process_schema._InlineDocs(schema)
     self.assertEqual(expected_schema, schema)
 
 
 if __name__ == '__main__':
   unittest.main()