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 0f3d43855cf20136f40ac864eb5da939c4e0994c..7a0661f097255cd5217d4ac40976a7231eb4eeef 100755 |
--- a/chrome/common/extensions/docs/server2/api_data_source_test.py |
+++ b/chrome/common/extensions/docs/server2/api_data_source_test.py |
@@ -28,6 +28,13 @@ def _GetType(dict_, name): |
if type_['name'] == name: |
return type_ |
+class FakeAvailabilityDataSourceFactory(object): |
+ def Create(self): |
+ return {} |
+ |
+def _FakeGetAvailability(self): |
+ return 'Not Currently Available' |
+ |
class FakeSamplesDataSource(object): |
def Create(self, request): |
return {} |
@@ -50,6 +57,7 @@ class FakeAPIAndListDataSource(object): |
class APIDataSourceTest(unittest.TestCase): |
def setUp(self): |
self._base_path = os.path.join(sys.path[0], 'test_data', 'test_json') |
+ _JSCModel._GetAvailability = _FakeGetAvailability |
def _ReadLocalFile(self, filename): |
with open(os.path.join(self._base_path, filename), 'r') as f: |
@@ -91,9 +99,11 @@ class APIDataSourceTest(unittest.TestCase): |
def testCreateId(self): |
data_source = FakeAPIAndListDataSource( |
self._LoadJSON('test_file_data_source.json')) |
+ fake_avail_factory = FakeAvailabilityDataSourceFactory() |
dict_ = _JSCModel(self._LoadJSON('test_file.json')[0], |
self._CreateRefResolver('test_file_data_source.json'), |
- False).ToDict() |
+ False, |
+ fake_avail_factory).ToDict() |
self.assertEquals('type-TypeA', dict_['types'][0]['id']) |
self.assertEquals('property-TypeA-b', |
dict_['types'][0]['properties'][0]['id']) |
@@ -117,9 +127,11 @@ class APIDataSourceTest(unittest.TestCase): |
self.assertEquals('234,567', _FormatValue(234567)) |
def testFormatDescription(self): |
+ fake_avail_factory = FakeAvailabilityDataSourceFactory() |
dict_ = _JSCModel(self._LoadJSON('ref_test.json')[0], |
self._CreateRefResolver('ref_test_data_source.json'), |
- False).ToDict() |
+ False, |
+ fake_avail_factory).ToDict() |
self.assertEquals(_MakeLink('ref_test.html#type-type2', 'type2'), |
_GetType(dict_, 'type1')['description']) |
self.assertEquals( |