From 324d2dca72dd6744454634840384c5c737c7af34 Mon Sep 17 00:00:00 2001
From: Wayne Witzel III <wayne@riotousliving.com>
Date: Mon, 14 Nov 2016 14:29:52 -0500
Subject: [PATCH] add JobEvent and UnifiedJob workflow property unit tests

---
 .../serializers/test_job_event_serializers.py | 28 +++++++++++++++++++
 .../tests/unit/models/test_unified_job.py     | 16 +++++++++++
 2 files changed, 44 insertions(+)
 create mode 100644 awx/main/tests/unit/api/serializers/test_job_event_serializers.py
 create mode 100644 awx/main/tests/unit/models/test_unified_job.py

diff --git a/awx/main/tests/unit/api/serializers/test_job_event_serializers.py b/awx/main/tests/unit/api/serializers/test_job_event_serializers.py
new file mode 100644
index 0000000000..a9c7ee7f08
--- /dev/null
+++ b/awx/main/tests/unit/api/serializers/test_job_event_serializers.py
@@ -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']
diff --git a/awx/main/tests/unit/models/test_unified_job.py b/awx/main/tests/unit/models/test_unified_job.py
new file mode 100644
index 0000000000..af8833482a
--- /dev/null
+++ b/awx/main/tests/unit/models/test_unified_job.py
@@ -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