1 files changed, 7 insertions, 6 deletions
diff --git a/chrome/common/extensions/docs/server2/template_data_source_test.py b/chrome/common/extensions/docs/server2/template_data_source_test.py
index 4765e74..813f2fd 100755
--- a/chrome/common/extensions/docs/server2/template_data_source_test.py
+++ b/chrome/common/extensions/docs/server2/template_data_source_test.py
@@ -58,7 +58,7 @@ class TemplateDataSourceTest(unittest.TestCase):
     reference_resolver_factory = ReferenceResolver.Factory(
         api_data_factory,
         self._fake_api_list_data_source_factory,
-        ObjectStoreCreator.Factory())
+        ObjectStoreCreator.TestFactory())
     return (TemplateDataSource.Factory('fake_channel',
                                        api_data_factory,
                                        self._fake_api_list_data_source_factory,
@@ -76,9 +76,10 @@ class TemplateDataSourceTest(unittest.TestCase):
     fetcher = LocalFileSystem(self._base_path)
     compiled_fs_factory = CompiledFileSystem.Factory(
         fetcher,
-        ObjectStoreCreator.Factory())
-    t_data_source = self._CreateTemplateDataSource(compiled_fs_factory,
-                                                   ObjectStoreCreator.Factory())
+        ObjectStoreCreator.TestFactory())
+    t_data_source = self._CreateTemplateDataSource(
+        compiled_fs_factory,
+        ObjectStoreCreator.TestFactory())
     template_a1 = Handlebar(self._ReadLocalFile('test1.html'))
     self.assertEqual(template_a1.render({}, {'templates': {}}).text,
         t_data_source.get('test1').render({}, {'templates': {}}).text)
@@ -94,7 +95,7 @@ class TemplateDataSourceTest(unittest.TestCase):
     fetcher = LocalFileSystem(self._base_path)
     compiled_fs_factory = CompiledFileSystem.Factory(
         fetcher,
-        ObjectStoreCreator.Factory())
+        ObjectStoreCreator.TestFactory())
     t_data_source = self._CreateTemplateDataSource(compiled_fs_factory)
     self.assertEqual(
         self._ReadLocalFile('test_expected.html'),
@@ -107,7 +108,7 @@ class TemplateDataSourceTest(unittest.TestCase):
     context = json.loads(self._ReadLocalFile('test1.json'))
     compiled_fs_factory = CompiledFileSystem.Factory(
         fetcher,
-        ObjectStoreCreator.Factory())
+        ObjectStoreCreator.TestFactory())
     self._RenderTest(
         'test1',
         self._CreateTemplateDataSource(