Merge pull request #3620 from AlanCoding/gce_instance_id

Add instance_id for gce imported hosts

Reviewed-by: https://github.com/softwarefactory-project-zuul[bot]

awx/main/tests/functional/test_inventory_source_migration.py

@@ -1,4 +1,5 @@
 import pytest
+from unittest import mock
 
 from awx.main.migrations import _inventory_source as invsrc
 from awx.main.models import InventorySource
@@ -45,3 +46,36 @@ def test_azure_inv_src_removal(inventory_source):
     assert InventorySource.objects.filter(pk=inventory_source.pk).exists()
     invsrc.remove_azure_inventory_sources(apps, None)
     assert not InventorySource.objects.filter(pk=inventory_source.pk).exists()
+
+
+@pytest.mark.parametrize('vars,id_var,result', [
+    ({'foo': {'bar': '1234'}}, 'foo.bar', '1234'),
+    ({'cat': 'meow'}, 'cat', 'meow'),
+    ({'dog': 'woof'}, 'cat', '')
+])
+def test_instance_id(vars, id_var, result):
+    assert invsrc._get_instance_id(vars, id_var) == result
+
+
+@pytest.mark.django_db
+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'