mirror of
https://github.com/ansible/awx.git
synced 2026-03-07 19:51:08 -03:30
d0776dabdf
* Make the JT name uniqueness enforced at the database level * Forgot demo project fixture * New approach, done by adding a new field * Update for linters and failures * Fix logical error in migration test * Revert some test changes based on review comment * Do not rename first template, add test * Avoid name-too-long rename errors * Insert migration into place * Move existing files with git * Bump migrations of existing * Update migration test * Awkward bump * Fix migration file link * update test reference again
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import pytest
|
|
|
|
from awx.main.migrations._db_constraints import _rename_duplicates
|
|
from awx.main.models import JobTemplate
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_rename_job_template_duplicates(organization, project):
|
|
ids = []
|
|
for i in range(5):
|
|
jt = JobTemplate.objects.create(name=f'jt-{i}', organization=organization, project=project)
|
|
ids.append(jt.id) # saved in order of creation
|
|
|
|
# Hack to first allow duplicate names of JT to test migration
|
|
JobTemplate.objects.filter(id__in=ids).update(org_unique=False)
|
|
|
|
# Set all JTs to the same name
|
|
JobTemplate.objects.filter(id__in=ids).update(name='same_name_for_test')
|
|
|
|
_rename_duplicates(JobTemplate)
|
|
|
|
first_jt = JobTemplate.objects.get(id=ids[0])
|
|
assert first_jt.name == 'same_name_for_test'
|
|
|
|
for i, pk in enumerate(ids):
|
|
if i == 0:
|
|
continue
|
|
jt = JobTemplate.objects.get(id=pk)
|
|
# Name should be set based on creation order
|
|
assert jt.name == f'same_name_for_test_dup{i}'
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_rename_job_template_name_too_long(organization, project):
|
|
ids = []
|
|
for i in range(3):
|
|
jt = JobTemplate.objects.create(name=f'jt-{i}', organization=organization, project=project)
|
|
ids.append(jt.id) # saved in order of creation
|
|
|
|
JobTemplate.objects.filter(id__in=ids).update(org_unique=False)
|
|
|
|
chars = 512
|
|
# Set all JTs to the same reaaaaaaly long name
|
|
JobTemplate.objects.filter(id__in=ids).update(name='A' * chars)
|
|
|
|
_rename_duplicates(JobTemplate)
|
|
|
|
first_jt = JobTemplate.objects.get(id=ids[0])
|
|
assert first_jt.name == 'A' * chars
|
|
|
|
for i, pk in enumerate(ids):
|
|
if i == 0:
|
|
continue
|
|
jt = JobTemplate.objects.get(id=pk)
|
|
assert jt.name.endswith(f'dup{i}')
|
|
assert len(jt.name) <= 512
|