Dynamically generate a heading for Extension Docs API pages

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

Index: chrome/common/extensions/docs/server2/branch_utility_test.py
diff --git a/chrome/common/extensions/docs/server2/branch_utility_test.py b/chrome/common/extensions/docs/server2/branch_utility_test.py
index 0ecb1bdf476cb649eb833f4e14eedf65be030e4e..c6a503405fe38f76c2f2e3051db06b188efdd903 100755
--- a/chrome/common/extensions/docs/server2/branch_utility_test.py
+++ b/chrome/common/extensions/docs/server2/branch_utility_test.py
@@ -10,12 +10,12 @@ import unittest
 from branch_utility import BranchUtility
 from fake_url_fetcher import FakeUrlFetcher
 from object_store_creator import ObjectStoreCreator
-from test_object_store import TestObjectStore
 
 class BranchUtilityTest(unittest.TestCase):
   def setUp(self):
     self._branch_util = BranchUtility(
         os.path.join('branch_utility', 'first.json'),
+        os.path.join('branch_utility', 'second.json'),
         FakeUrlFetcher(os.path.join(sys.path[0], 'test_data')),
         ObjectStoreCreator.ForTest())
 
@@ -45,11 +45,80 @@ class BranchUtilityTest(unittest.TestCase):
                       self._branch_util.SplitChannelNameFromPath(
                       'stuff.html'))
 
-  def testGetBranchForChannel(self):
-    self.assertEquals('1145', self._branch_util.GetBranchForChannel('dev'))
-    self.assertEquals('1084', self._branch_util.GetBranchForChannel('beta'))
-    self.assertEquals('1084', self._branch_util.GetBranchForChannel('stable'))
-    self.assertEquals('trunk', self._branch_util.GetBranchForChannel('trunk'))
+  def testNewestChannel(self):
+    self.assertEquals('trunk',
+        self._branch_util.NewestChannel(('trunk', 'dev', 'beta', 'stable')))
+    self.assertEquals('trunk',
+        self._branch_util.NewestChannel(('stable', 'beta', 'dev', 'trunk')))
+    self.assertEquals('dev',
+        self._branch_util.NewestChannel(('stable', 'beta', 'dev')))
+    self.assertEquals('dev',
+        self._branch_util.NewestChannel(('dev', 'beta', 'stable')))
+    self.assertEquals('beta',
+        self._branch_util.NewestChannel(('beta', 'stable')))
+    self.assertEquals('beta',
+        self._branch_util.NewestChannel(('stable', 'beta')))
+    self.assertEquals('stable', self._branch_util.NewestChannel(('stable',)))
+    self.assertEquals('beta', self._branch_util.NewestChannel(('beta',)))
+    self.assertEquals('dev', self._branch_util.NewestChannel(('dev',)))
+    self.assertEquals('trunk', self._branch_util.NewestChannel(('trunk',)))
+
+  def testGetChannelInfo(self):
+    self.assertEquals('trunk',
+      self._branch_util.GetChannelInfo('trunk').channel)
+    self.assertEquals('trunk',
+      self._branch_util.GetChannelInfo('trunk').branch)
+    self.assertEquals('trunk',
+      self._branch_util.GetChannelInfo('trunk').version)
+    self.assertEquals('dev',
+      self._branch_util.GetChannelInfo('dev').channel)
+    self.assertEquals(1500,
+      self._branch_util.GetChannelInfo('dev').branch)
+    self.assertEquals(28,
+      self._branch_util.GetChannelInfo('dev').version)
+    self.assertEquals('beta',
+      self._branch_util.GetChannelInfo('beta').channel)
+    self.assertEquals(1453,
+      self._branch_util.GetChannelInfo('beta').branch)
+    self.assertEquals(27,
+      self._branch_util.GetChannelInfo('beta').version)
+    self.assertEquals('stable',
+      self._branch_util.GetChannelInfo('stable').channel)
+    self.assertEquals(1410,
+      self._branch_util.GetChannelInfo('stable').branch)
+    self.assertEquals(26,
+      self._branch_util.GetChannelInfo('stable').version)
+
+  def testGetLatestVersionNumber(self):
+    self.assertEquals(28, self._branch_util.GetLatestVersionNumber())
+
+  def testGetBranchForVersion(self):
+    self.assertEquals(1453,
+        self._branch_util.GetBranchForVersion(27))
+    self.assertEquals(1410,
+        self._branch_util.GetBranchForVersion(26))
+    self.assertEquals(1364,
+        self._branch_util.GetBranchForVersion(25))
+    self.assertEquals(1312,
+        self._branch_util.GetBranchForVersion(24))
+    self.assertEquals(1271,
+        self._branch_util.GetBranchForVersion(23))
+    self.assertEquals(1229,
+        self._branch_util.GetBranchForVersion(22))
+    self.assertEquals(1180,
+        self._branch_util.GetBranchForVersion(21))
+    self.assertEquals(1132,
+        self._branch_util.GetBranchForVersion(20))
+    self.assertEquals(1084,
+        self._branch_util.GetBranchForVersion(19))
+    self.assertEquals(1025,
+        self._branch_util.GetBranchForVersion(18))
+    self.assertEquals(963,
+        self._branch_util.GetBranchForVersion(17))
+    self.assertEquals(696,
+        self._branch_util.GetBranchForVersion(11))
+    self.assertEquals(396,
+        self._branch_util.GetBranchForVersion(5))
 
 if __name__ == '__main__':
   unittest.main()