Index: chrome/common/extensions/docs/server2/test_branch_utility.py |
diff --git a/chrome/common/extensions/docs/server2/test_branch_utility.py b/chrome/common/extensions/docs/server2/test_branch_utility.py |
index 1c365a56869fe64d5538bf619c64fc0207b4e3a6..aaabdcac36886148ec84f1909c8b078433c06501 100644 |
--- a/chrome/common/extensions/docs/server2/test_branch_utility.py |
+++ b/chrome/common/extensions/docs/server2/test_branch_utility.py |
@@ -2,11 +2,20 @@ |
# 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 |
+from branch_utility import BranchUtility, ChannelInfo |
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): |
+ return ChannelInfo(channel, |
+ 'fakebranch-%s' % channel, |
+ 'fakeversion-%s' % channel) |
+ |
+ def GetBranchForVersion(self, version): |
+ return 'fakebranch-%s' % version |