Dynamically generate a heading for Extension Docs API pages

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

 #!/usr/bin/env python
 # Copyright (c) 2012 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 os
 import sys
 import unittest
 
 from in_memory_object_store import InMemoryObjectStore
@@ skipping 28 matching lines @@
                       self._branch_util.SplitChannelNameFromPath(
                       'apps/stuff.html'))
     self.assertEquals((None, 'extensions/dev/stuff.html'),
                       self._branch_util.SplitChannelNameFromPath(
                       'extensions/dev/stuff.html'))
     self.assertEquals((None, 'stuff.html'),
                       self._branch_util.SplitChannelNameFromPath(
                       'stuff.html'))
 
   def testGetBranchNumberForChannelName(self):
+    self.assertEquals('dev',
+       self._branch_util.GetChannelInfoForChannelName('dev')['name'])
+    self.assertEquals('beta',
+       self._branch_util.GetChannelInfoForChannelName('beta')['name'])
+    self.assertEquals('stable',
+       self._branch_util.GetChannelInfoForChannelName('stable')['name'])
+    self.assertEquals('trunk',
+       self._branch_util.GetChannelInfoForChannelName('trunk')['name'])
+
     self.assertEquals('1145',
-                      self._branch_util.GetBranchNumberForChannelName('dev'))
+       self._branch_util.GetChannelInfoForChannelName('dev')['branch'])
     self.assertEquals('1084',
-                      self._branch_util.GetBranchNumberForChannelName('beta'))
+       self._branch_util.GetChannelInfoForChannelName('beta')['branch'])
     self.assertEquals('1084',
-                      self._branch_util.GetBranchNumberForChannelName('stable'))
+       self._branch_util.GetChannelInfoForChannelName('stable')['branch'])
     self.assertEquals('trunk',
-                      self._branch_util.GetBranchNumberForChannelName('trunk'))
+       self._branch_util.GetChannelInfoForChannelName('trunk')['branch'])
+
+    self.assertEquals('21',
+       self._branch_util.GetChannelInfoForChannelName('dev')['version'])
+    self.assertEquals('19',
+       self._branch_util.GetChannelInfoForChannelName('beta')['version'])
+    self.assertEquals('19',
+       self._branch_util.GetChannelInfoForChannelName('stable')['version'])
+    self.assertEquals('trunk',
+       self._branch_util.GetChannelInfoForChannelName('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()