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..e88b9b2038644f74c624d1afd862718568430658 100755
--- a/chrome/common/extensions/docs/server2/branch_utility_test.py
+++ b/chrome/common/extensions/docs/server2/branch_utility_test.py
@@ -44,12 +44,64 @@ class BranchUtilityTest(unittest.TestCase):
     self.assertEquals((None, 'stuff.html'),
                       self._branch_util.SplitChannelNameFromPath(
                       'stuff.html'))
+  def testNewestChannel(self):
+    self.assertEquals('trunk',
+        self._branch_util.NewestChannel(['trunk', 'dev', 'beta', 'stable']))
+    self.assertEquals('dev',
+        self._branch_util.NewestChannel(['dev', 'beta', 'stable']))
+    self.assertEquals('beta',
+        self._branch_util.NewestChannel(['beta', 'stable']))
+    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 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 testGetBranchNumberForChannelName(self):
+    self.assertEquals('dev',
+       self._branch_util.GetChannelInfo('dev').name)
+    self.assertEquals('beta',
+       self._branch_util.GetChannelInfo('beta').name)
+    self.assertEquals('stable',
+       self._branch_util.GetChannelInfo('stable').name)
+    self.assertEquals('trunk',
+       self._branch_util.GetChannelInfo('trunk').name)
+
+    self.assertEquals('1145',
+       self._branch_util.GetChannelInfo('dev').branch)
+    self.assertEquals('1084',
+       self._branch_util.GetChannelInfo('beta').branch)
+    self.assertEquals('1084',
+       self._branch_util.GetChannelInfo('stable').branch)
+    self.assertEquals('trunk',
+       self._branch_util.GetChannelInfo('trunk').branch)
+
+    self.assertEquals('21',
+       self._branch_util.GetChannelInfo('dev').version)
+    self.assertEquals('19',
+       self._branch_util.GetChannelInfo('beta').version)
+    self.assertEquals('19',
+       self._branch_util.GetChannelInfo('stable').version)
+    self.assertEquals('trunk',
+       self._branch_util.GetChannelInfo('trunk').version)
+
+  def test_ExtractFromVersionJson(self):
+    self.assertEquals('1145',
+       self._branch_util._ExtractFromVersionJson('dev', 'branch'))
+    self.assertEquals('1084',
+       self._branch_util._ExtractFromVersionJson('beta', 'branch'))
+    self.assertEquals('1084',
+       self._branch_util._ExtractFromVersionJson('stable', 'branch'))
+    self.assertEquals('trunk',
+       self._branch_util._ExtractFromVersionJson('trunk', 'branch'))
+
+    self.assertEquals('21',
+       self._branch_util._ExtractFromVersionJson('dev', 'version'))
+    self.assertEquals('19',
+       self._branch_util._ExtractFromVersionJson('beta', 'version'))
+    self.assertEquals('19',
+       self._branch_util._ExtractFromVersionJson('stable', 'version'))
+    self.assertEquals('trunk',
+       self._branch_util._ExtractFromVersionJson('trunk', 'version'))
 
 if __name__ == '__main__':
   unittest.main()