Dynamically generate a heading for Extension Docs API pages

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

+#!/usr/bin/env python
+# Copyright (c) 2013 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import json
+import os
+import unittest
+
+from api_data_source import APIDataSource
+from availability_data_source import AvailabilityDataSource
+from compiled_file_system import CompiledFileSystem
+from in_memory_object_store import InMemoryObjectStore
+from memcache_file_system import MemcacheFileSystem
+from test_file_system import TestFileSystem
+
+_TEST_DIR = 'test_data/test_avail/'
+
+def _LoadJson(name, mode='rb'):
+  with open(os.path.join(_TEST_DIR, name), mode) as f:
+    return json.load(f)
+
+# These methods will prevent an api_data_source from actually loading an api,
+# since that behavior isn't of interest here; we're only interested in whether
+# or not the api is available to be loaded.
+def _LoadJsonAPI(self, api, disable_refs):
+  return {}
+
+def _LoadIdlAPI(self, api, disable_refs):
+  return {}
+
+class _FakeSamplesDataSource(object):
+  def Create(self, request):
+    return {}
+
+class _FakeRefResolver(object):
+  def Create(self, request):
+    return {}
+
+class _FakeChromeVersionDataSource(object):
+  def _CreateMemcacheForBranch(self, branch_number):
+    file_system_json = _LoadJson('test_avail_file_system.json')
+    branch_memcache = InMemoryObjectStore(branch_number)
+    file_system = MemcacheFileSystem(
+        TestFileSystem(file_system_json[branch_number]),
+        branch_memcache)
+    cache_factory = CompiledFileSystem.Factory(file_system, branch_memcache)
+    api_data_source_factory = APIDataSource.Factory(cache_factory, 'api')
+    api_data_source_factory.SetReferenceResolverFactory(_FakeRefResolver())
+    api_data_source_factory.SetSamplesDataSourceFactory(
+        _FakeSamplesDataSource())
+    return api_data_source_factory.Create(None)
+
+  def GetDataSourceForVersion(self, version_number):
+    branch_dict_json = _LoadJson('test_avail_branch_list.json')
+    branch_number = branch_dict_json[version_number]
+    branch_api_data_source = self._CreateMemcacheForBranch(branch_number)
+    return branch_api_data_source
+
+class AvailabilityDataSourceTest(unittest.TestCase):
+  def setUp(self):
+    self._availability_ds = AvailabilityDataSource(
+        _FakeChromeVersionDataSource(),
+        '26',
+        InMemoryObjectStore(''))
+    APIDataSource.Factory._LoadJsonAPI = _LoadJsonAPI
+    APIDataSource.Factory._LoadIdlAPI = _LoadIdlAPI
+
+  def testCheckExistence(self):
+    self.assertEquals(False,
+        self._availability_ds._CheckExistence('gobbledigook', '23'))
+    self.assertEquals(True,
+        self._availability_ds._CheckExistence('alarms', '22'))
+    self.assertEquals(True,
+        self._availability_ds._CheckExistence('alarms', '21'))
+    self.assertEquals(False,
+        self._availability_ds._CheckExistence('alarms', '20'))
+    self.assertEquals(True,
+        self._availability_ds._CheckExistence('tabs', '18'))
+    self.assertEquals(False,
+        self._availability_ds._CheckExistence('tabs', '17'))
+
+  def testFindEarliestAvailability(self):
+    self.assertEquals(None,
+        self._availability_ds.FindEarliestAvailability('gobblediegook'))
+    self.assertEquals('21',
+        self._availability_ds.FindEarliestAvailability('alarms'))
+    self.assertEquals('18',
+        self._availability_ds.FindEarliestAvailability('bookmarks'))
+    self.assertEquals('18',
+        self._availability_ds.FindEarliestAvailability('browserAction'))
+    self.assertEquals('18',
+        self._availability_ds.FindEarliestAvailability('contextMenus'))
+    self.assertEquals('18',
+        self._availability_ds.FindEarliestAvailability('cookies'))
+    self.assertEquals('21',
+        self._availability_ds.FindEarliestAvailability('fileSystem'))
+    self.assertEquals('24',
+        self._availability_ds.FindEarliestAvailability('pushMessaging'))
+    self.assertEquals('21',
+        self._availability_ds.FindEarliestAvailability('runtime'))
+    self.assertEquals('19',
+        self._availability_ds.FindEarliestAvailability('storage'))
+    self.assertEquals('18',
+        self._availability_ds.FindEarliestAvailability('tabs'))
+
+if __name__ == '__main__':
+  unittest.main()