| 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(
|
|
|