mirror of
https://github.com/ansible/awx.git
synced 2026-02-18 03:30:02 -03:30
add JobEvent and UnifiedJob workflow property unit tests
This commit is contained in:
@@ -0,0 +1,28 @@
|
|||||||
|
import pytest
|
||||||
|
import mock
|
||||||
|
|
||||||
|
from awx.api.serializers import JobEventSerializer
|
||||||
|
from awx.main.models import (
|
||||||
|
Job,
|
||||||
|
JobTemplate,
|
||||||
|
JobEvent,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def job_event(mocker):
|
||||||
|
job_event = mocker.MagicMock(spec=JobEvent)
|
||||||
|
|
||||||
|
job = Job(id=1, name="job-1")
|
||||||
|
job.job_template = JobTemplate(id=1, name="job-template-1")
|
||||||
|
job_event.job = job
|
||||||
|
|
||||||
|
return job_event
|
||||||
|
|
||||||
|
|
||||||
|
def test_summary_field_workflow_exists(job_event):
|
||||||
|
with mock.patch('awx.api.serializers.BaseSerializer.get_summary_fields', lambda x,y: {'job':{}}):
|
||||||
|
serializer = JobEventSerializer(job_event)
|
||||||
|
summary_fields = serializer.get_summary_fields(job_event)
|
||||||
|
assert 'spawned_by_workflow' in summary_fields['job']
|
||||||
|
assert 'workflow_job_id' in summary_fields['job']
|
||||||
16
awx/main/tests/unit/models/test_unified_job.py
Normal file
16
awx/main/tests/unit/models/test_unified_job.py
Normal file
@@ -0,0 +1,16 @@
|
|||||||
|
import mock
|
||||||
|
|
||||||
|
from awx.main.models import (
|
||||||
|
UnifiedJob,
|
||||||
|
WorkflowJob,
|
||||||
|
WorkflowJobNode,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_unified_job_workflow_attributes():
|
||||||
|
with mock.patch('django.db.ConnectionRouter.db_for_write'):
|
||||||
|
job = UnifiedJob(id=1, name="job-1", launch_type="workflow")
|
||||||
|
job.unified_job_node = WorkflowJobNode(workflow_job=WorkflowJob(pk=1))
|
||||||
|
|
||||||
|
assert job.spawned_by_workflow is True
|
||||||
|
assert job.workflow_job_id == 1
|
||||||
Reference in New Issue
Block a user