| Index: appengine/swarming/handlers_bot_test.py
|
| diff --git a/appengine/swarming/handlers_bot_test.py b/appengine/swarming/handlers_bot_test.py
|
| index aa61a8981114ece741526a209e46900c1aa5e04e..83ba6b0a6cc52f66cee3563a1a0d3bb88b6d4d5a 100755
|
| --- a/appengine/swarming/handlers_bot_test.py
|
| +++ b/appengine/swarming/handlers_bot_test.py
|
| @@ -272,7 +272,7 @@ class BotApiTest(test_env_handlers.AppTestBase):
|
| # A bot polls, gets a task, updates it, completes it.
|
| params = self.do_handshake()
|
| # Enqueue a task.
|
| - _, task_id = self.client_create_task_raw()
|
| + _, _, task_id = self.client_create_task_raw()
|
| self.assertEqual('0', task_id[-1])
|
| # Convert TaskResultSummary reference to TaskRunResult.
|
| task_id = task_id[:-1] + '1'
|
| @@ -341,7 +341,7 @@ class BotApiTest(test_env_handlers.AppTestBase):
|
| def test_poll_task_with_bot_service_account(self):
|
| params = self.do_handshake()
|
|
|
| - _, task_id = self.client_create_task_raw(service_account_token='bot')
|
| + _, _, task_id = self.client_create_task_raw(service_account_token='bot')
|
| self.assertEqual('0', task_id[-1])
|
| task_id = task_id[:-1] + '1'
|
|
|
| @@ -387,7 +387,7 @@ class BotApiTest(test_env_handlers.AppTestBase):
|
| def test_poll_task_with_caches(self):
|
| params = self.do_handshake()
|
|
|
| - _, task_id = self.client_create_task_raw({
|
| + _, _, task_id = self.client_create_task_raw({
|
| 'caches': [{
|
| 'name': 'git_infra',
|
| 'path': 'git_cache',
|
| @@ -502,7 +502,7 @@ class BotApiTest(test_env_handlers.AppTestBase):
|
| # A bot polls, gets a task, updates it, completes it.
|
| params = self.do_handshake()
|
| # Enqueue a task.
|
| - _, task_id = self.client_create_task_isolated()
|
| + _, _, task_id = self.client_create_task_isolated()
|
| self.assertEqual('0', task_id[-1])
|
| # Convert TaskResultSummary reference to TaskRunResult.
|
| task_id = task_id[:-1] + '1'
|
| @@ -638,7 +638,7 @@ class BotApiTest(test_env_handlers.AppTestBase):
|
|
|
| def test_poll_not_enough_time(self):
|
| # Make sure there's a task that we don't get.
|
| - _, task_id = self.client_create_task_raw()
|
| + _, _, task_id = self.client_create_task_raw()
|
| self.assertEqual('0', task_id[-1])
|
| params = self.do_handshake()
|
| bot_info = bot_management.get_info_key('bot1').get()
|
| @@ -657,7 +657,7 @@ class BotApiTest(test_env_handlers.AppTestBase):
|
| self.mock(random, 'getrandbits', lambda _: 0x88)
|
| now = datetime.datetime(2010, 1, 2, 3, 4, 5)
|
| self.mock_now(now)
|
| - _, task_id = self.client_create_task_isolated()
|
| + _, _, task_id = self.client_create_task_isolated()
|
| self.assertEqual('0', task_id[-1])
|
| params = self.do_handshake()
|
| bot_info = bot_management.get_info_key('bot1').get()
|
|
|