External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-26 13:25:02 -02:30
Code Issues Packages Projects Releases Wiki Activity

update tests to reflect new job_event schema

Browse Source
This commit is contained in:
Jim Ladd
2021-03-11 16:01:48 -08:00
parent 8501a45531
commit b98b3ced1c
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
awx/main/tests/functional/api/test_events.py
View File

@@ -1,9 +1,17 @@
import pytest import pytest
from unittest import mock
from awx.api.versioning import reverse from awx.api.versioning import reverse
from awx.main.models import AdHocCommand, AdHocCommandEvent, JobEvent from awx.main.models import AdHocCommand, AdHocCommandEvent, JobEvent
from awx.main.models import Job
# Job.created_or_epoch is used to help retrieve events that were
# created before job event tables were partitioned.
# This test can safely behave as if all job events were created
# after the migration, in which case Job.created_or_epoch == Job.created
@mock.patch('awx.main.models.Job.created_or_epoch',
Job.created)
@pytest.mark.django_db @pytest.mark.django_db
@pytest.mark.parametrize( @pytest.mark.parametrize(
'truncate, expected', 'truncate, expected',
@@ -16,7 +24,7 @@ def test_job_events_sublist_truncation(get, organization_factory, job_template_f
objs = organization_factory("org", superusers=['admin']) objs = organization_factory("org", superusers=['admin'])
jt = job_template_factory("jt", organization=objs.organization, inventory='test_inv', project='test_proj').job_template jt = job_template_factory("jt", organization=objs.organization, inventory='test_inv', project='test_proj').job_template
job = jt.create_unified_job() job = jt.create_unified_job()
JobEvent.create_from_data(job_id=job.pk, uuid='abc123', event='runner_on_start', stdout='a' * 1025).save() JobEvent.create_from_data(job_id=job.pk, uuid='abc123', event='runner_on_start', stdout='a' * 1025, job_created=job.created).save()
url = reverse('api:job_job_events_list', kwargs={'pk': job.pk}) url = reverse('api:job_job_events_list', kwargs={'pk': job.pk})
if not truncate: if not truncate:
@@ -26,6 +34,12 @@ def test_job_events_sublist_truncation(get, organization_factory, job_template_f
assert (len(response.data['results'][0]['stdout']) == 1025) == expected assert (len(response.data['results'][0]['stdout']) == 1025) == expected
# Job.created_or_epoch is used to help retrieve events that were
# created before job event tables were partitioned.
# This test can safely behave as if all job events were created
# after the migration, in which case Job.created_or_epoch == Job.created
@mock.patch('awx.main.models.ad_hoc_commands.AdHocCommand.created_or_epoch',
Job.created)
@pytest.mark.django_db @pytest.mark.django_db
@pytest.mark.parametrize( @pytest.mark.parametrize(
'truncate, expected', 'truncate, expected',
@@ -38,7 +52,7 @@ def test_ad_hoc_events_sublist_truncation(get, organization_factory, job_templat
objs = organization_factory("org", superusers=['admin']) objs = organization_factory("org", superusers=['admin'])
adhoc = AdHocCommand() adhoc = AdHocCommand()
adhoc.save() adhoc.save()
AdHocCommandEvent.create_from_data(ad_hoc_command_id=adhoc.pk, uuid='abc123', event='runner_on_start', stdout='a' * 1025).save() AdHocCommandEvent.create_from_data(ad_hoc_command_id=adhoc.pk, uuid='abc123', event='runner_on_start', stdout='a' * 1025, job_created=adhoc.created).save()
url = reverse('api:ad_hoc_command_ad_hoc_command_events_list', kwargs={'pk': adhoc.pk}) url = reverse('api:ad_hoc_command_ad_hoc_command_events_list', kwargs={'pk': adhoc.pk})
if not truncate: if not truncate: