Inlining references to internal APIs.

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

Index: chrome/common/extensions/docs/server2/process_schema_test.py
diff --git a/chrome/common/extensions/docs/server2/schema_util_test.py b/chrome/common/extensions/docs/server2/process_schema_test.py
similarity index 62%
rename from chrome/common/extensions/docs/server2/schema_util_test.py
rename to chrome/common/extensions/docs/server2/process_schema_test.py
index 3be428cce412ce5fd4375f5cce66cad6bde53494..eac38fb2bde37227655e7cbdeea7079f060239e1 100755
--- a/chrome/common/extensions/docs/server2/schema_util_test.py
+++ b/chrome/common/extensions/docs/server2/process_schema_test.py
@@ -6,11 +6,25 @@
 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 SchemaUtilTest(unittest.TestCase):
+class _FakeAPIModels():
+  def GetNames(self):
+    return []
+
+class _FakeFeaturesBundle():
+  def GetAPIFeatures(self):
+    return Future(value={})
 
+class SchemaUtilTest(unittest.TestCase):
   def testRemoveNoDocs(self):
     expected_nodoc = [
       {
@@ -106,7 +120,16 @@ class SchemaUtilTest(unittest.TestCase):
       }
     ]
 
-    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):
@@ -169,8 +192,17 @@ class SchemaUtilTest(unittest.TestCase):
       ]
     }
 
+    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):
@@ -200,8 +232,17 @@ class SchemaUtilTest(unittest.TestCase):
       ]
     }
 
-    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)