mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-17 05:21:49 -05:00
54bf7e13d8
The original tests set no longer works after Django 1.11 due to more strict rules against dynamic model definition. The refactored tests set aims at each existing model that apply named URL rules, instead of abstract general use cases, thus significantly improves maintainability and readability. Signed-off-by: Aaron Tan <jangsutsr@gmail.com>
191 lines
8.3 KiB
Python
191 lines
8.3 KiB
Python
import pytest
|
|
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
|
|
from awx.api.versioning import reverse
|
|
from awx.main.middleware import URLModificationMiddleware
|
|
from awx.main.models import * # noqa
|
|
|
|
|
|
@pytest.fixture(scope='function', autouse=True)
|
|
def init_url_modification_middleware():
|
|
# In real-world scenario, named url graph structure is populated by __init__
|
|
# of URLModificationMiddleware. The way Django bootstraps ensures the initialization
|
|
# will happen *once and only once*, while the number of initialization is uncontrollable
|
|
# in unit test environment. So it is wrapped by try-except block to mute any
|
|
# unwanted exceptions.
|
|
try:
|
|
URLModificationMiddleware()
|
|
except ImproperlyConfigured:
|
|
pass
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_user(get, admin_user):
|
|
test_user = User.objects.create(username='test_user', password='test_user', is_superuser=False)
|
|
url = reverse('api:user_detail', kwargs={'pk': test_user.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_user/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_team(get, admin_user):
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_team = test_org.teams.create(name='test_team')
|
|
url = reverse('api:team_detail', kwargs={'pk': test_team.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_team++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_organization(get, admin_user):
|
|
test_org = Organization.objects.create(name='test_org')
|
|
url = reverse('api:organization_detail', kwargs={'pk': test_org.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_job_template(get, admin_user):
|
|
test_jt = JobTemplate.objects.create(name='test_jt')
|
|
url = reverse('api:job_template_detail', kwargs={'pk': test_jt.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_jt/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_workflow_job_template(get, admin_user):
|
|
test_wfjt = WorkflowJobTemplate.objects.create(name='test_wfjt')
|
|
url = reverse('api:workflow_job_template_detail', kwargs={'pk': test_wfjt.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_wfjt++/')
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_wfjt.organization = test_org
|
|
test_wfjt.save()
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_wfjt++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_label(get, admin_user):
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_label = test_org.labels.create(name='test_label')
|
|
url = reverse('api:label_detail', kwargs={'pk': test_label.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_label++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_project(get, admin_user):
|
|
test_proj = Project.objects.create(name='test_proj')
|
|
url = reverse('api:project_detail', kwargs={'pk': test_proj.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_proj++/')
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_proj.organization = test_org
|
|
test_proj.save()
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_proj++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_notification_template(get, admin_user):
|
|
test_notification_template = NotificationTemplate.objects.create(
|
|
name='test_note', notification_type='slack',
|
|
notification_configuration=dict(channels=["Foo", "Bar"], token="token")
|
|
)
|
|
url = reverse('api:notification_template_detail', kwargs={'pk': test_notification_template.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_note++/')
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_notification_template.organization = test_org
|
|
test_notification_template.save()
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_note++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_instance(get, admin_user):
|
|
test_instance = Instance.objects.create(
|
|
uuid=settings.SYSTEM_UUID, hostname="localhost", capacity=100
|
|
)
|
|
url = reverse('api:instance_detail', kwargs={'pk': test_instance.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/localhost/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_instance_group(get, admin_user):
|
|
test_instance_group = InstanceGroup.objects.create(name='Tower')
|
|
url = reverse('api:instance_group_detail', kwargs={'pk': test_instance_group.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/Tower/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_inventory(get, admin_user):
|
|
test_inv = Inventory.objects.create(name='test_inv')
|
|
url = reverse('api:inventory_detail', kwargs={'pk': test_inv.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_inv++/')
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_inv.organization = test_org
|
|
test_inv.save()
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_inv++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_host(get, admin_user):
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_inv = Inventory.objects.create(name='test_inv', organization=test_org)
|
|
test_host = Host.objects.create(name='test_host', inventory=test_inv)
|
|
url = reverse('api:host_detail', kwargs={'pk': test_host.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_host++test_inv++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_group(get, admin_user):
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_inv = Inventory.objects.create(name='test_inv', organization=test_org)
|
|
test_group = Group.objects.create(name='test_group', inventory=test_inv)
|
|
url = reverse('api:group_detail', kwargs={'pk': test_group.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_group++test_inv++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_inventory_source(get, admin_user):
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_inv = Inventory.objects.create(name='test_inv', organization=test_org)
|
|
test_source = InventorySource.objects.create(name='test_source', inventory=test_inv)
|
|
url = reverse('api:inventory_source_detail', kwargs={'pk': test_source.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_source++test_inv++test_org/')
|
|
test_source.inventory = None
|
|
test_source.save()
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_source++/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_inventory_script(get, admin_user):
|
|
test_script = CustomInventoryScript.objects.create(name='test_script')
|
|
url = reverse('api:inventory_script_detail', kwargs={'pk': test_script.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_script++/')
|
|
test_org = Organization.objects.create(name='test_org')
|
|
test_script.organization = test_org
|
|
test_script.save()
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_script++test_org/')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_credential(get, admin_user, credentialtype_ssh):
|
|
test_cred = Credential.objects.create(name='test_cred', credential_type=credentialtype_ssh)
|
|
url = reverse('api:credential_detail', kwargs={'pk': test_cred.pk})
|
|
response = get(url, user=admin_user, expect=200)
|
|
assert response.data['related']['named_url'].endswith('/test_cred++Machine+ssh++/')
|