Adding tower_instance_group module

awx_collection/test/awx/test_instance_group.py

@@ -0,0 +1,69 @@
+from __future__ import (absolute_import, division, print_function)
+__metaclass__ = type
+
+import pytest
+
+from awx.main.models import InstanceGroup, Instance
+from awx.main.tests.functional.conftest import kube_credential, credentialtype_kube
+
+@pytest.mark.django_db
+def test_instance_group_create(run_module, admin_user):
+    result = run_module('tower_instance_group', {
+        'name': 'foo-group',
+        'policy_instance_percentage': 34,
+        'policy_instance_minimum': 12,
+        'state': 'present'
+    }, admin_user)
+    assert not result.get('failed', False), result
+    assert result['changed']
+
+    ig = InstanceGroup.objects.get(name='foo-group')
+    assert ig.policy_instance_percentage == 34
+    assert ig.policy_instance_minimum == 12
+
+    # Create a new instance in the DB
+    new_instance = Instance.objects.create(hostname='foo.example.com')
+
+    # Set the new instance group only to the one instnace
+    result = run_module('tower_instance_group', {
+        'name': 'foo-group',
+        'instances': new_instance.hostname,
+        'state': 'present'
+    }, admin_user)
+    assert not result.get('failed', False), result
+    assert result['changed']
+
+    ig = InstanceGroup.objects.get(name='foo-group')
+    all_instance_names = []
+    for instance in ig.instances.all():
+        all_instance_names.append(instance.hostname)
+
+    assert new_instance.hostname in all_instance_names, 'Failed to add instance to group'
+    assert len(all_instance_names) == 1, 'Too many instances in group {0}'.format(','.join(all_instance_names))
+
+@pytest.mark.django_db
+def test_continer_group_create(run_module, admin_user, kube_credential):
+    pod_spec = "{ 'Nothing': True }"
+
+    result = run_module('tower_instance_group', {
+        'name': 'foo-c-group',
+        'credential': kube_credential.id,
+        'state': 'present'
+    }, admin_user)
+    assert not result.get('failed', False), result['msg']
+    assert result['changed']
+
+    ig = InstanceGroup.objects.get(name='foo-c-group')
+    assert ig.pod_spec_override == ''
+
+    result = run_module('tower_instance_group', {
+        'name': 'foo-c-group',
+        'credential': kube_credential.id,
+        'pod_spec_override': pod_spec,
+        'state': 'present'
+    }, admin_user)
+    assert not result.get('failed', False), result['msg']
+    assert result['changed']
+
+    ig = InstanceGroup.objects.get(name='foo-c-group')
+    assert ig.pod_spec_override == pod_spec