Refactor task enqueuing

appengine/gce-backend/instance_group_managers_test.py

 #!/usr/bin/python
 # Copyright 2016 The LUCI Authors. All rights reserved.
 # Use of this source code is governed under the Apache License, Version 2.0
 # that can be found in the LICENSE file.
 
 """Unit tests for instance_group_managers.py."""
 
 import unittest
 
 import test_env
@@ skipping 869 matching lines @@
     ).put()
     models.InstanceTemplate(key=key.parent().parent()).put()
 
     instance_group_managers.resize(key)
 
 
 class ScheduleCreationTest(test_case.TestCase):
   """Tests for instance_group_managers.schedule_creation."""
 
   def setUp(self, *args, **kwargs):
-    def enqueue_task(*args, **kwargs):
-      self.failUnless(kwargs.get('params', {}).get('key'))
-      entity = ndb.Key(urlsafe=kwargs['params']['key']).get()
-      entity.url = kwargs['params']['key']
+    def enqueue_task(taskqueue, key):
+      entity = key.get()
+      entity.url = key.urlsafe()
       entity.put()
       return True
 
     super(ScheduleCreationTest, self).setUp(*args, **kwargs)
-    self.mock(instance_group_managers.utils, 'enqueue_task', enqueue_task)
+    self.mock(instance_group_managers.utilities, 'enqueue_task', enqueue_task)
 
   def test_enqueues_task(self):
     """Ensures a task is enqueued."""
     key = instance_group_managers.get_instance_group_manager_key(
         'base-name', 'revision', 'zone')
     models.InstanceTemplate(
         key=key.parent().parent(),
         active=key.parent(),
     ).put()
     models.InstanceTemplateRevision(
@@ skipping 196 matching lines @@
         ),
     ).put()
 
     instance_group_managers.update_url(key, 'url')
 
     self.assertEqual(key.get().url, 'url')
 
 
 if __name__ == '__main__':
   unittest.main()