Refactor task enqueuing

appengine/gce-backend/instance_group_managers_test.py

Index: appengine/gce-backend/instance_group_managers_test.py
diff --git a/appengine/gce-backend/instance_group_managers_test.py b/appengine/gce-backend/instance_group_managers_test.py
index a83be8b0027b1cb7a6fb18fa22b773b33382a2f2..e48ac34d128522b5bc811944ffb95e3612024cb8 100755
--- a/appengine/gce-backend/instance_group_managers_test.py
+++ b/appengine/gce-backend/instance_group_managers_test.py
@@ -887,15 +887,14 @@ 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."""