def test_apply_new_instance_id(inventory_source):
host1 = inventory_source.hosts.create(name='foo1',
inventory=inventory_source.inventory,
variables={'foo': 'bar'},
instance_id='')
host2 = inventory_source.hosts.create(name='foo2',
inventory=inventory_source.inventory,
variables={'foo': 'bar'},
instance_id='bad_user')
with mock.patch(
'django.conf.settings.{}_INSTANCE_ID_VAR'.format(
inventory_source.source.upper()), 'foo'):
invsrc.set_new_instance_id(apps, inventory_source.source, 'foo')
host1.refresh_from_db()
host2.refresh_from_db()
assert host1.instance_id == 'bar'
assert host2.instance_id == 'bad_user'
with mock.patch(
'django.conf.settings.{}_INSTANCE_ID_VAR'.format(
inventory_source.source.upper()), 'foo'):
invsrc.back_out_new_instance_id(apps, inventory_source.source, 'foo')
host1.refresh_from_db()
host2.refresh_from_db()
assert host1.instance_id == ''
assert host2.instance_id == 'bad_user'
def gce_id_forward(apps, schema_editor):
set_new_instance_id(apps, 'gce', GCE_INSTANCE_ID_VAR)