def test_instance_template_revision_missing(self): """Ensures no task is enqueued for missing instance template revisions.""" key = instance_group_managers.get_instance_group_manager_key( 'base-name', 'revision', 'zone') models.InstanceTemplate( key=key.parent().parent(), active=key.parent(), ).put() models.InstanceGroupManager(key=key).put() instance_group_managers.schedule_creation() self.failIf(key.get().url)
def test_instance_group_manager_inactive(self): """Ensures no task is enqueued for inactive instance group managers.""" 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( key=key.parent(), url='url', ).put() models.InstanceGroupManager(key=key).put() instance_group_managers.schedule_creation() self.failIf(key.get().url)
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( key=key.parent(), active=[ key, ], url='url', ).put() models.InstanceGroupManager(key=key).put() expected_url = key.urlsafe() instance_group_managers.schedule_creation() self.assertEqual(key.get().url, expected_url)
def test_instance_group_manager_already_created(self): """Ensures no task is enqueued for existing instance group managers.""" 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( key=key.parent(), active=[ key, ], url='instance-template-url', ).put() models.InstanceGroupManager(key=key, url='url').put() expected_url = 'url' instance_group_managers.schedule_creation() self.assertEqual(key.get().url, expected_url)
def get(self): instance_group_managers.schedule_creation()