add JobEvent and UnifiedJob workflow property unit tests

This commit is contained in:
Wayne Witzel III 2016-11-14 14:29:52 -05:00
parent 5bf00fe96b
commit 324d2dca72
2 changed files with 44 additions and 0 deletions

View File

@ -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']

View 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