Refactor task enqueuing

appengine/gce-backend/instance_templates_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_templates.py."""
 
 import unittest
 
 import test_env
@@ skipping 345 matching lines @@
     ]
 
     actual = instance_templates.get_drained_instance_template_revisions()
     self.assertItemsEqual(actual, expected)
 
 
 class ScheduleCreationTest(test_case.TestCase):
   """Tests for instance_templates.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_templates.utils, 'enqueue_task', enqueue_task)
+    self.mock(instance_templates.utilities, 'enqueue_task', enqueue_task)
 
   def test_enqueues_task(self):
     """Ensures a task is enqueued."""
     key = instance_templates.get_instance_template_revision_key(
         'base-name', 'revision')
     models.InstanceTemplate(key=key.parent(), active=key).put()
     models.InstanceTemplateRevision(key=key).put()
     expected_url = key.urlsafe()
 
     instance_templates.schedule_creation()
@@ skipping 84 matching lines @@
             'revision',
         ),
     ).put()
 
     instance_templates.update_url(key, 'url')
 
     self.assertEqual(key.get().url, 'url')
 
 if __name__ == '__main__':
   unittest.main()