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 availability_data_source import AvailabilityDataSource
+from caching_file_system import CachingFileSystem
+from object_store_creator import ObjectStoreCreator
+from test_file_system import TestFileSystem
+
+_TEST_DIR = 'test_data/availability_data_source/'
+
+def _LoadJson(name, mode='rb'):
+  with open(os.path.join(_TEST_DIR, name), mode) as f:
+    return json.load(f)
+
+def _CreateTestFileSystemForBranch(branch):
+  file_system_json = _LoadJson('test_avail_file_system.json')
+  test_file_system = CachingFileSystem(TestFileSystem(file_system_json[branch]),
+                                       ObjectStoreCreator.Factory())
+  return test_file_system
+
+class _FakeChromeVersionDataSource(object):
+  def GetLatestVersionNumber(self):
+    return '27'
+
+  def GetBranchNumberForVersion(self, version_number):
+    branch_dict_json = _LoadJson('test_avail_branch_list.json')
+    branch_number = branch_dict_json[version_number]
+    return branch_number
+
+class AvailabilityDataSourceTest(unittest.TestCase):
+  def setUp(self):
+    self._availability_ds_factory = AvailabilityDataSource.Factory(
+        _FakeChromeVersionDataSource())
+    self._availability_ds_factory._existing_availabilities = {'fake' : 'beta'}
+    self._availability_ds = self._availability_ds_factory.Create()
+    (self._availability_ds.
+        _CreateSvnFileSystemForBranch) = _CreateTestFileSystemForBranch
+
+  def testCheckExistence(self):
+    self.assertEquals(False,
+        self._availability_ds._CheckApiExistence('gobbledigook', '23'))
+    self.assertEquals(True,
+        self._availability_ds._CheckApiExistence('alarms', '22'))
+    self.assertEquals(True,
+        self._availability_ds._CheckApiExistence('alarms', '21'))
+    self.assertEquals(False,
+        self._availability_ds._CheckApiExistence('alarms', '20'))
+    self.assertEquals(True,
+        self._availability_ds._CheckApiExistence('tabs', '18'))
+    self.assertEquals(False,
+        self._availability_ds._CheckApiExistence('tabs', '17'))
+
+  def testFindEarliestAvailability(self):
+    self.assertEquals('beta',
+        self._availability_ds.FindEarliestAvailability('fake'))
+    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()