mirror of
https://github.com/ansible/awx.git
synced 2026-01-15 11:50:42 -03:30
add JobEvent and UnifiedJob workflow property unit tests
This commit is contained in:
parent
5bf00fe96b
commit
324d2dca72
@ -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
|
||||
Loading…
x
Reference in New Issue
Block a user