| 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
|
|
|