Merge branch 'devel' of github.com:ansible/ansible-tower into merge-devel

awx/main/tests/functional/conftest.py

@@ -1,8 +1,23 @@
-import pytest
 
+# Python
+import pytest
+import mock
+import json
+import os
+from datetime import timedelta
+
+# Django
 from django.core.urlresolvers import resolve
 from django.utils.six.moves.urllib.parse import urlparse
+from django.utils import timezone
 from django.contrib.auth.models import User
+from django.conf import settings
+
+# AWX
+from awx.main.models.projects import Project
+from awx.main.models.base import PERM_INVENTORY_READ
+from awx.main.models.ha import Instance
+from awx.main.models.fact import Fact
 
 from rest_framework.test import (
     APIRequestFactory,
@@ -10,20 +25,34 @@ from rest_framework.test import (
 )
 
 from awx.main.models.credential import Credential
-from awx.main.models.projects import Project
 from awx.main.models.jobs import JobTemplate
-from awx.main.models.ha import Instance
 from awx.main.models.inventory import (
-    Inventory,
     Group,
 )
 from awx.main.models.organization import (
     Organization,
-    Team,
+    Permission,
 )
 
 from awx.main.models.rbac import Role
 
+'''
+Disable all django model signals.
+'''
+@pytest.fixture(scope="session", autouse=False)
+def disable_signals():
+    mocked = mock.patch('django.dispatch.Signal.send', autospec=True)
+    mocked.start()
+
+'''
+FIXME: Not sure how "far" just setting the BROKER_URL will get us.
+We may need to incluence CELERY's configuration like we do in the old unit tests (see base.py)
+
+Allows django signal code to execute without the need for redis
+'''
+@pytest.fixture(scope="session", autouse=True)
+def celery_memory_broker():
+    settings.BROKER_URL='memory://localhost/'
 
 @pytest.fixture
 def user():
@@ -60,11 +89,15 @@ def deploy_jobtemplate(project, inventory, credential):
 
 @pytest.fixture
 def team(organization):
-    return Team.objects.create(organization=organization, name='test-team')
+    return organization.teams.create(name='test-team')
 
 @pytest.fixture
-def project(organization):
-    prj = Project.objects.create(name="test-project",  description="test-project-desc")
+@mock.patch.object(Project, "update", lambda self, **kwargs: None)
+def project(instance, organization):
+    prj = Project.objects.create(name="test-proj",
+                                 description="test-proj-desc",
+                                 scm_type="git",
+                                 scm_url="https://github.com/jlaska/ansible-playbooks")
     prj.organizations.add(organization)
     return prj
 
@@ -87,7 +120,7 @@ def credential():
 
 @pytest.fixture
 def inventory(organization):
-    return Inventory.objects.create(name="test-inventory", organization=organization)
+    return organization.inventories.create(name="test-inv")
 
 @pytest.fixture
 def role():
@@ -105,12 +138,38 @@ def alice(user):
 def bob(user):
     return user('bob', False)
 
+@pytest.fixture
+def organizations(instance):
+    def rf(organization_count=1):
+        orgs = []
+        for i in xrange(0, organization_count):
+            o = Organization.objects.create(name="test-org-%d" % i, description="test-org-desc")
+            orgs.append(o)
+        return orgs
+    return rf
+
 @pytest.fixture
 def group(inventory):
     def g(name):
         return Group.objects.create(inventory=inventory, name=name)
     return g
 
+@pytest.fixture
+def hosts(group):
+    def rf(host_count=1):
+        hosts = []
+        for i in xrange(0, host_count):
+            name = '%s-host-%s' % (group.name, i)
+            (host, created) = group.inventory.hosts.get_or_create(name=name)
+            if created:
+                group.hosts.add(host)
+            hosts.append(host)
+        return hosts
+    return rf
+
+
+
+
 @pytest.fixture
 def permissions():
     return {
@@ -244,7 +303,46 @@ def options():
         return response
     return rf
 
-@pytest.fixture(scope="session", autouse=True)
-def celery_memory_broker():
-    from django.conf import settings
-    settings.BROKER_URL='memory://localhost/'
+
+
+@pytest.fixture
+def fact_scans(group, fact_ansible_json, fact_packages_json, fact_services_json):
+    def rf(fact_scans=1, timestamp_epoch=timezone.now()):
+        facts_json = {}
+        facts = []
+        module_names = ['ansible', 'services', 'packages']
+        timestamp_current = timestamp_epoch
+
+        facts_json['ansible'] = fact_ansible_json
+        facts_json['packages'] = fact_packages_json
+        facts_json['services'] = fact_services_json
+
+        for i in xrange(0, fact_scans):
+            for host in group.hosts.all():
+                for module_name in module_names:
+                    facts.append(Fact.objects.create(host=host, timestamp=timestamp_current, module=module_name, facts=facts_json[module_name]))
+            timestamp_current += timedelta(days=1)
+        return facts
+    return rf
+
+def _fact_json(module_name):
+    current_dir = os.path.dirname(os.path.realpath(__file__))
+    with open('%s/%s.json' % (current_dir, module_name)) as f:
+        return json.load(f)
+
+@pytest.fixture
+def fact_ansible_json():
+    return _fact_json('ansible')
+
+@pytest.fixture
+def fact_packages_json():
+    return _fact_json('packages')
+
+@pytest.fixture
+def fact_services_json():
+    return _fact_json('services')
+
+@pytest.fixture
+def permission_inv_read(organization, inventory, team):
+    return Permission.objects.create(inventory=inventory, team=team, permission_type=PERM_INVENTORY_READ)
+