def testGetDimensionsSuccessful(self, mock_fetch):
mock_fetch.return_value = collections.namedtuple(
'Result', ['content', 'status_code', 'headers'])(
status_code=200, content=_MOCK_TASK_DEF_RESPONSE, headers={})
dimensions = swarmbucket.GetDimensionsForBuilder(
'luci.chromium.ci', 'Linux x64', dimensions_whitelist=None)
self.assertEqual(['cpu:x86-64', 'os:Ubuntu-14.04', 'pool:Chrome.LUCI'],
dimensions)
dimensions = swarmbucket.GetDimensionsForBuilder('luci.chromium.ci',
'Linux x64')
self.assertEqual(['cpu:x86-64', 'os:Ubuntu-14.04'], dimensions)
_args, kwargs = mock_fetch.call_args
self.assertEqual(http_client_appengine.urlfetch.POST, kwargs.get('method'))
def testGetDimensionsFailed(self, mock_fetch):
mock_fetch.return_value = collections.namedtuple(
'Result', ['content', 'status_code', 'headers'])(
status_code=501, content=_MOCK_TASK_DEF_RESPONSE, headers={})
dimensions = swarmbucket.GetDimensionsForBuilder('luci.chromium.ci',
'Linux x64')
self.assertFalse(dimensions)
def GetTrybotDimensions(_wf_mastername, wf_buildername, bucket=None):
bucket = bucket or _CI_BUCKET
return MergeDimensions(
_DEFAULT_DIMENSIONS,
swarmbucket.GetDimensionsForBuilder(bucket, wf_buildername))