Dynamically generate a heading for Extension Docs API pages

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

 # Copyright 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.
 
 from branch_utility import BranchUtility
 
+channels = {
+    'trunk': ['trunk', 'trunk'],
+    'dev': [1500, 28],
+    'beta': [1453, 27],
+    'stable': [1410, 26]
+}
+
+branches = {
+  'trunk' : 'trunk',
+  28 : 1500,
+  27 : 1453,
+  26 : 1410,
+  25 : 1364,
+  24 : 1312,
+  23 : 1271,
+  22 : 1229,
+  21 : 1180,
+  20 : 1132,
+  19 : 1084,
+  18 : 1025,
+  17 : 963,
+  16 : 912,
+  15 : 874,
+  14 : 835,
+  13 : 782,
+  12 : 742,
+  11 : 696,
+  10 : 648,
+   9 : 597,
+   8 : 552,
+   7 : 544,
+   6 : 495,
+   5 : 396
+}
+
 class TestBranchUtility(object):
   '''Mimics BranchUtility to return valid-ish data without needing omahaproxy
   data.
   '''
-  def GetBranchForChannel(self, channel_name):
-    return channel_name
+  def GetAllChannelInfo(self):
+    return [self.GetChannelInfo(channel)
+            for channel in BranchUtility.GetAllChannelNames()]
+
+  def GetChannelInfo(self, channel):
+    class TestChannelInfo(object):
+      def __init__(self, channel, branch, version):
+        self.channel = channel
+        self.branch = branch
+        self.version = version
+
+    return TestChannelInfo(channel, channels[channel][0], channels[channel][1])
+
+  def GetBranchNumberForVersion(self, version):
+    return branches[version]