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