Merge branch 'mongoectomy_implementation' into devel

awx/main/tests/functional/conftest.py

@@ -1,18 +1,48 @@
+
+# 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 awx.main.models.organization import Organization
-from awx.main.models.projects import Project
-from awx.main.models.ha import Instance
+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.organization import Organization, Permission
+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,
     force_authenticate,
 )
 
+'''
+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():
     def u(name, is_superuser=False):
@@ -158,3 +188,79 @@ def project(instance):
                                   description="test-proj-desc",
                                   scm_type="git",
                                   scm_url="https://github.com/jlaska/ansible-playbooks")
+@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 inventory(organization):
+    return organization.inventories.create(name="test-inv")
+
+@pytest.fixture
+def group(inventory):
+    return inventory.groups.create(name='group-1')
+
+@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 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 team(organization):
+    return organization.teams.create(name='test-team')
+
+@pytest.fixture
+def permission_inv_read(organization, inventory, team):
+    return Permission.objects.create(inventory=inventory, team=team, permission_type=PERM_INVENTORY_READ)
+