Index: chrome/common/extensions/docs/server2/api_data_source_test.py |
diff --git a/chrome/common/extensions/docs/server2/api_data_source_test.py b/chrome/common/extensions/docs/server2/api_data_source_test.py |
index f3c472f74a0700482da78ebfb4bc30efba9ca8b7..0919b944d085580ad6cebf421d6fc34e9bbfa398 100755 |
--- a/chrome/common/extensions/docs/server2/api_data_source_test.py |
+++ b/chrome/common/extensions/docs/server2/api_data_source_test.py |
@@ -3,7 +3,7 @@ |
# Use of this source code is governed by a BSD-style license that can be |
# found in the LICENSE file. |
-import copy |
+from copy import deepcopy |
import json |
import os |
import sys |
@@ -19,6 +19,7 @@ from file_system import FileNotFoundError |
from local_file_system import LocalFileSystem |
from object_store_creator import ObjectStoreCreator |
from reference_resolver import ReferenceResolver |
+from test_file_system import TestFileSystem |
def _MakeLink(href, text): |
return '<a href="%s">%s</a>' % (href, text) |
@@ -28,6 +29,10 @@ def _GetType(dict_, name): |
if type_['name'] == name: |
return type_ |
+class FakeAvailabilityDataSource(object): |
+ def GetAvailability(self, version): |
+ return None |
+ |
class FakeSamplesDataSource(object): |
def Create(self, request): |
return {} |
@@ -50,6 +55,9 @@ class FakeAPIAndListDataSource(object): |
class APIDataSourceTest(unittest.TestCase): |
def setUp(self): |
self._base_path = os.path.join(sys.path[0], 'test_data', 'test_json') |
+ def test_file_system_creator(branch): |
+ return TestFileSystem(self._LoadJSON('basic_docs_filesystem.json')) |
+ self._test_file_system_creator = test_file_system_creator |
def _ReadLocalFile(self, filename): |
with open(os.path.join(self._base_path, filename), 'r') as f: |
@@ -70,7 +78,9 @@ class APIDataSourceTest(unittest.TestCase): |
self._LoadJSON('test_file_data_source.json')) |
dict_ = _JSCModel(self._LoadJSON('test_file.json')[0], |
self._CreateRefResolver('test_file_data_source.json'), |
- False).ToDict() |
+ False, |
+ FakeAvailabilityDataSource(), |
+ self._test_file_system_creator).ToDict() |
self.assertEquals('type-TypeA', dict_['types'][0]['id']) |
self.assertEquals('property-TypeA-b', |
dict_['types'][0]['properties'][0]['id']) |
@@ -85,7 +95,9 @@ class APIDataSourceTest(unittest.TestCase): |
self._LoadJSON('test_file_data_source.json')) |
dict_ = _JSCModel(self._LoadJSON(filename)[0], |
self._CreateRefResolver('test_file_data_source.json'), |
- False).ToDict() |
+ False, |
+ FakeAvailabilityDataSource(), |
+ self._test_file_system_creator).ToDict() |
self.assertEquals(expected_json, dict_) |
def testFormatValue(self): |
@@ -96,7 +108,9 @@ class APIDataSourceTest(unittest.TestCase): |
def testFormatDescription(self): |
dict_ = _JSCModel(self._LoadJSON('ref_test.json')[0], |
self._CreateRefResolver('ref_test_data_source.json'), |
- False).ToDict() |
+ False, |
+ FakeAvailabilityDataSource(), |
+ self._test_file_system_creator).ToDict() |
self.assertEquals(_MakeLink('ref_test.html#type-type2', 'type2'), |
_GetType(dict_, 'type1')['description']) |
self.assertEquals( |
@@ -111,7 +125,25 @@ class APIDataSourceTest(unittest.TestCase): |
def testRemoveNoDocs(self): |
d = self._LoadJSON('nodoc_test.json') |
_RemoveNoDocs(d) |
- self.assertEqual(self._LoadJSON('expected_nodoc.json'), d) |
+ self.assertEquals(self._LoadJSON('expected_nodoc.json'), d) |
+ |
+ def testGetPermissions(self): |
+ model = _JSCModel(self._LoadJSON('test_file.json')[0], |
+ self._CreateRefResolver('test_file_data_source.json'), |
+ False, |
+ FakeAvailabilityDataSource(), |
+ self._test_file_system_creator) |
+ self.assertEquals(model._GetPermissions(), |
+ self._LoadJSON('api_permissions_expected.json')) |
+ |
+ def testGetMoreLearning(self): |
+ model = _JSCModel(self._LoadJSON('test_file.json')[0], |
+ self._CreateRefResolver('test_file_data_source.json'), |
+ False, |
+ FakeAvailabilityDataSource(), |
+ self._test_file_system_creator) |
+ self.assertEquals(model._GetMoreLearning(), |
+ self._LoadJSON('api_learn_more_expected.json')) |
def testInlineDocs(self): |
schema = { |
@@ -173,7 +205,7 @@ class APIDataSourceTest(unittest.TestCase): |
] |
} |
- inlined_schema = copy.deepcopy(schema) |
+ inlined_schema = deepcopy(schema) |
_InlineDocs(inlined_schema) |
self.assertEqual(expected_schema, inlined_schema) |