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