def test_deletes(self):
"""Ensures the entity is deleted."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
), ).put()
models.InstanceTemplate(
key=key.parent(),
drained=[
ndb.Key(models.InstanceTemplateRevision, 'fake-key-1'),
key,
ndb.Key(models.InstanceTemplateRevision, 'fake-key-2'),
],
).put()
expected_drained = [
ndb.Key(models.InstanceTemplateRevision, 'fake-key-1'),
ndb.Key(models.InstanceTemplateRevision, 'fake-key-2'),
]
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failIf(key.get())
self.assertItemsEqual(key.parent().get().drained, expected_drained)
def test_deletes(self):
"""Ensures the entity is deleted."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
),
).put()
models.InstanceTemplate(
key=key.parent(),
drained=[
ndb.Key(models.InstanceTemplateRevision, 'fake-key-1'),
key,
ndb.Key(models.InstanceTemplateRevision, 'fake-key-2'),
],
).put()
expected_drained = [
ndb.Key(models.InstanceTemplateRevision, 'fake-key-1'),
ndb.Key(models.InstanceTemplateRevision, 'fake-key-2'),
]
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failIf(key.get())
self.assertItemsEqual(key.parent().get().drained, expected_drained)
def test_entity_not_found(self):
"""Ensures nothing happens when the entity is not found."""
key = ndb.Key(models.InstanceTemplateRevision, 'fake-key')
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failIf(key.get())
def test_entity_not_found(self):
"""Ensures nothing happens when the entity is not found."""
key = ndb.Key(models.InstanceTemplateRevision, 'fake-key')
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failIf(key.get())
def test_parent_doesnt_exist(self):
"""Ensures nothing happens when the parent doesn't exist."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
), ).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())
def test_parent_doesnt_exist(self):
"""Ensures nothing happens when the parent doesn't exist."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
),
).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())
def test_active(self):
"""Ensures nothing happens when the entity is active."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
), ).put()
models.InstanceTemplate(
key=key.parent(),
active=key,
).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())
def test_active(self):
"""Ensures nothing happens when the entity is active."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
),
).put()
models.InstanceTemplate(
key=key.parent(),
active=key,
).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())
def test_url_specified(self):
"""Ensures nothing happens when the entity still has a URL."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
),
url='url',
).put()
models.InstanceTemplate(
key=key.parent(),
drained=[
key,
],
).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())
def test_url_specified(self):
"""Ensures nothing happens when the entity still has a URL."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
),
url='url',
).put()
models.InstanceTemplate(
key=key.parent(),
drained=[
key,
],
).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())
def test_drained_instance_group_managers(self):
"""Ensures nothing happens when there are drained InstanceGroupManagers."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
),
drained=[
ndb.Key(models.InstanceGroupManager, 'fake-key'),
],
).put()
models.InstanceTemplate(
key=key.parent(),
drained=[
key,
],
).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())
def test_drained_instance_group_managers(self):
"""Ensures nothing happens when there are drained InstanceGroupManagers."""
key = models.InstanceTemplateRevision(
key=instance_templates.get_instance_template_revision_key(
'base-name',
'revision',
),
drained=[
ndb.Key(models.InstanceGroupManager, 'fake-key'),
],
).put()
models.InstanceTemplate(
key=key.parent(),
drained=[
key,
],
).put()
future = cleanup.delete_instance_template_revision(key)
future.wait()
self.failUnless(key.get())