Dynamically generate a heading for Extension Docs API pages

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

Index: chrome/common/extensions/docs/server2/availability_finder_test.py
diff --git a/chrome/common/extensions/docs/server2/availability_finder_test.py b/chrome/common/extensions/docs/server2/availability_finder_test.py
new file mode 100755
index 0000000000000000000000000000000000000000..1fdc9abae167075846c0682af4bb4e1384b114c9
--- /dev/null
+++ b/chrome/common/extensions/docs/server2/availability_finder_test.py
@@ -0,0 +1,172 @@
+#!/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_finder import AvailabilityFinder
+from compiled_file_system import CompiledFileSystem
+from fake_fetchers import ConfigureFakeFetchers
+from object_store_creator import ObjectStoreCreator
+from test_file_system import TestFileSystem
+
+def _LoadJSON(name, mode='rb'):
+  return json.load(
+      open(os.path.join('test_data', 'availability_finder', name), mode))
+
+def _CreateSvnFileSystem(branch):
+  return TestFileSystem(_LoadJSON('test_avail_filesystem.json')[branch])
+
+class AvailabilityInfo(object):
+  def __init__(self, channel, version):
+    self.channel = channel
+    self.version = version
+
+class _FakeChromeVersionUtility(object):
+  def GetLatestVersionNumber(self):
+    return 28
+
+  def GetBranchNumberForVersion(self, version):
+    return _LoadJSON('test_avail_branch_list.json')[str(version)]
+
+class AvailabilityFinderTest(unittest.TestCase):
+  def setUp(self):
+    # ChromeVersionUtility attempts to fetch data omaha proxy
+    ConfigureFakeFetchers()
+    self._avail_ds_factory = AvailabilityFinder.Factory(
+        _FakeChromeVersionUtility(),
+        ObjectStoreCreator.ForTest(),
+        CompiledFileSystem.Factory(
+            TestFileSystem(_LoadJSON('test_avail_filesystem.json')['trunk']),
+            ObjectStoreCreator.ForTest()),
+        _CreateSvnFileSystem)
+    self._avail_ds = self._avail_ds_factory.Create()
+    self._avail_ds._permission_apis = ['alarms', 'bookmarks', 'cookies']
+    self._avail_ds._manifest_apis = ['storage', 'runtime']
+    self._avail_ds._orphan_apis = ['idle', 'menus', 'tabs', 'windows']
+  '''
+  def testCheckFileSystemExistence(self):
+    self.assertEquals(True,
+        self._avail_ds._CheckFileSystemExistence('tabs', '20'))
+    self.assertEquals(True,
+        self._avail_ds._CheckFileSystemExistence('tabs', '19'))
+    self.assertEquals(True,
+        self._avail_ds._CheckFileSystemExistence('tabs', '18'))
+    self.assertEquals(False,
+        self._avail_ds._CheckFileSystemExistence('windows', '22'))
+    self.assertEquals(True,
+        self._avail_ds._CheckFileSystemExistence('windows', '24'))
+
+  def testCheckManifestExistence(self):
+    self.assertEquals(True,
+        self._avail_ds._CheckManifestExistence('storage', '27'))
+    self.assertEquals(False,
+        self._avail_ds._CheckManifestExistence('storage', '26'))
+    self.assertEquals(True,
+        self._avail_ds._CheckManifestExistence('runtime', '21'))
+    self.assertEquals(False,
+        self._avail_ds._CheckManifestExistence('runtime', '20'))
+    self.assertEquals(False,
+        self._avail_ds._CheckManifestExistence('cookies', '27'))
+
+  def testCheckPermissionExistence(self):
+    self.assertEquals(False,
+        self._avail_ds._CheckStablePermissionExistence('cookies', '27'))
+    self.assertEquals(True,
+        self._avail_ds._CheckStablePermissionExistence('cookies', '28'))
+    self.assertEquals(True,
+        self._avail_ds._CheckStablePermissionExistence('cookies', 'trunk'))
+    self.assertEquals(True,
+        self._avail_ds._CheckStablePermissionExistence('alarms', '24'))
+    self.assertEquals(False,
+        self._avail_ds._CheckStablePermissionExistence('alarms', '23'))
+    self.assertEquals(True,
+        self._avail_ds._CheckStablePermissionExistence('bookmarks', '21'))
+    self.assertEquals(False,
+        self._avail_ds._CheckStablePermissionExistence('bookmarks', '20'))
+
+  def testCheckExtensionAPIExistence(self):
+    self.assertEquals(True,
+        self._avail_ds._CheckExtensionAPIExistence('idle', '17'))
+    self.assertEquals(True,
+        self._avail_ds._CheckExtensionAPIExistence('idle', '12'))
+    self.assertEquals(True,
+        self._avail_ds._CheckExtensionAPIExistence('idle', '5'))
+    self.assertEquals(True,
+        self._avail_ds._CheckExtensionAPIExistence('menus', '6'))
+    self.assertEquals(False,
+        self._avail_ds._CheckExtensionAPIExistence('menus', '5'))
+
+  def testFindEarliestStableAvailability(self):
+    self.assertEquals('18',
+        self._avail_ds._FindEarliestStableAvailability('tabs', 27))
+    self.assertEquals('24',
+        self._avail_ds._FindEarliestStableAvailability('alarms', 27))
+    self.assertEquals('23',
+        self._avail_ds._FindEarliestStableAvailability('windows', 27))
+    self.assertEquals('27',
+        self._avail_ds._FindEarliestStableAvailability('storage', 27))
+    self.assertEquals('6',
+        self._avail_ds._FindEarliestStableAvailability('menus', 27))
+    self.assertEquals('5',
+        self._avail_ds._FindEarliestStableAvailability('idle', 27))
+  '''
+  def testGetApiAvailability(self):
+    self.assertEqual('stable',
+        self._avail_ds.GetApiAvailability('earlyAPI1').channel)
+    self.assertEqual('10',
+        self._avail_ds.GetApiAvailability('earlyAPI1').version)
+    self.assertEqual('trunk',
+        self._avail_ds.GetApiAvailability('earlyAPI2').channel)
+    self.assertEqual('trunk',
+        self._avail_ds.GetApiAvailability('earlyAPI2').version)
+    self.assertEquals('stable',
+        self._avail_ds.GetApiAvailability('tabs').channel)
+    self.assertEquals(18,
+        self._avail_ds.GetApiAvailability('tabs').version)
+    self.assertEquals('stable',
+        self._avail_ds.GetApiAvailability('bookmarks').channel)
+    self.assertEquals(21,
+        self._avail_ds.GetApiAvailability('bookmarks').version)
+    self.assertEquals('stable',
+        self._avail_ds.GetApiAvailability('windows').channel)
+    self.assertEquals(23,
+        self._avail_ds.GetApiAvailability('windows').version)
+    self.assertEquals('beta',
+        self._avail_ds.GetApiAvailability('storage').channel)
+    self.assertEquals(27,
+        self._avail_ds.GetApiAvailability('storage').version)
+    self.assertEquals('stable',
+        self._avail_ds.GetApiAvailability('alarms').channel)
+    self.assertEquals(24,
+        self._avail_ds.GetApiAvailability('alarms').version)
+    self.assertEquals('dev',
+        self._avail_ds.GetApiAvailability('sync').channel)
+    self.assertEquals(28,
+        self._avail_ds.GetApiAvailability('sync').version)
+    self.assertEquals('dev',
+        self._avail_ds.GetApiAvailability('cookies').channel)
+    self.assertEquals(28,
+        self._avail_ds.GetApiAvailability('cookies').version)
+    self.assertEquals('trunk',
+        self._avail_ds.GetApiAvailability('notReallyBetaAPI').channel)
+    self.assertEquals('trunk',
+        self._avail_ds.GetApiAvailability('notReallyBetaAPI').version)
+    self.assertEquals('trunk',
+        self._avail_ds.GetApiAvailability('trunkAPI').channel)
+    self.assertEquals('trunk',
+        self._avail_ds.GetApiAvailability('trunkAPI').version)
+    self.assertEquals('stable',
+        self._avail_ds.GetApiAvailability('menus').channel)
+    self.assertEquals(6,
+        self._avail_ds.GetApiAvailability('menus').version)
+    self.assertEquals('stable',
+        self._avail_ds.GetApiAvailability('idle').channel)
+    self.assertEquals(5,
+        self._avail_ds.GetApiAvailability('idle').version)
+
+if __name__ == '__main__':
+  unittest.main()