Separates normal & smart inventory logic for script endpoint

awx/api/views.py

@@ -2376,7 +2376,15 @@ class InventoryScriptView(RetrieveAPIView):
             if obj.variables_dict:
                 all_group = data.setdefault('all', OrderedDict())
                 all_group['vars'] = obj.variables_dict
-
+            if obj.kind == 'smart':
+                if len(obj.hosts.all()) == 0:
+                    return Response({})
+                else:
+                    all_group = data.setdefault('all', OrderedDict())
+                    smart_hosts_qs = obj.hosts.all().order_by('name')
+                    smart_hosts = list(smart_hosts_qs.values_list('name', flat=True))
+                    all_group['hosts'] = smart_hosts
+            else:
                 # Add hosts without a group to the all group.
                 groupless_hosts_qs = obj.hosts.filter(groups__isnull=True, **hosts_q).order_by('name')
                 groupless_hosts = list(groupless_hosts_qs.values_list('name', flat=True))
@@ -2422,18 +2430,6 @@ class InventoryScriptView(RetrieveAPIView):
                 for host in obj.hosts.filter(**hosts_q):
                     data['_meta']['hostvars'][host.name] = host.variables_dict
 
-            # workaround for Ansible inventory bug (github #3687), localhost
-            # must be explicitly listed in the all group for dynamic inventory
-            # scripts to pick it up.
-            localhost_names = ('localhost', '127.0.0.1', '::1')
-            localhosts_qs = obj.hosts.filter(name__in=localhost_names, **hosts_q)
-            localhosts = list(localhosts_qs.values_list('name', flat=True))
-            if localhosts:
-                all_group = data.setdefault('all', OrderedDict())
-                all_group_hosts = all_group.get('hosts', [])
-                all_group_hosts.extend(localhosts)
-                all_group['hosts'] = sorted(set(all_group_hosts))
-
         return Response(data)
 
 

awx/main/tests/functional/api/test_script_endpoint.py

@@ -0,0 +1,75 @@
+import pytest
+import json
+from awx.api.versioning import reverse
+
+from awx.main.models import Inventory, Host
+
+
+@pytest.mark.django_db
+def test_empty_inventory(post, get, admin_user, organization, group_factory):
+    inventory = Inventory(name='basic_inventory', 
+                          kind='', 
+                          organization=organization)
+    inventory.save()
+    resp = get(reverse('api:inventory_script_view', kwargs={'version': 'v2', 'pk': inventory.pk}), admin_user)
+    jdata = json.loads(resp.content)
+    
+    assert inventory.hosts.count() == 0
+    assert jdata == {}
+    
+    
+@pytest.mark.django_db
+def test_empty_smart_inventory(post, get, admin_user, organization, group_factory):
+    smart_inventory = Inventory(name='smart',
+                                kind='smart',
+                                organization=organization,
+                                host_filter='enabled=True')
+    smart_inventory.save()
+    resp = get(reverse('api:inventory_script_view', kwargs={'version': 'v2', 'pk': smart_inventory.pk}), admin_user)
+    smartjdata = json.loads(resp.content)
+
+    assert smart_inventory.hosts.count() == 0
+    assert smartjdata == {}
+    
+    
+@pytest.mark.django_db
+def test_ungrouped_hosts(post, get, admin_user, organization, group_factory):
+    inventory = Inventory(name='basic_inventory', 
+                          kind='', 
+                          organization=organization)
+    inventory.save()
+    Host.objects.create(name='first_host', inventory=inventory)
+    Host.objects.create(name='second_host', inventory=inventory)
+    resp = get(reverse('api:inventory_script_view', kwargs={'version': 'v2', 'pk': inventory.pk}), admin_user)
+    jdata = json.loads(resp.content)
+    assert inventory.hosts.count() == 2
+    assert len(jdata['all']['hosts']) == 2
+
+
+@pytest.mark.django_db
+def test_grouped_hosts_smart_inventory(post, get, admin_user, organization, group_factory):
+    inventory = Inventory(name='basic_inventory', 
+                          kind='', 
+                          organization=organization)
+    inventory.save()
+    groupA = group_factory('test_groupA')
+    host1 = Host.objects.create(name='first_host', inventory=inventory)
+    host2 = Host.objects.create(name='second_host', inventory=inventory)
+    Host.objects.create(name='third_host', inventory=inventory)
+    groupA.hosts.add(host1)
+    groupA.hosts.add(host2)       
+    smart_inventory = Inventory(name='smart_inventory',
+                                kind='smart',
+                                organization=organization,
+                                host_filter='enabled=True')
+    smart_inventory.save()
+    resp = get(reverse('api:inventory_script_view', kwargs={'version': 'v2', 'pk': inventory.pk}), admin_user)
+    jdata = json.loads(resp.content)
+    resp = get(reverse('api:inventory_script_view', kwargs={'version': 'v2', 'pk': smart_inventory.pk}), admin_user)
+    smartjdata = json.loads(resp.content)
+
+    assert getattr(smart_inventory, 'kind') == 'smart'
+    assert inventory.hosts.count() == 3
+    assert len(jdata['all']['hosts']) == 1
+    assert smart_inventory.hosts.count() == 3
+    assert len(smartjdata['all']['hosts']) == 3