From dc3e744dda48060c79dfaebd9994ac2bedb80603 Mon Sep 17 00:00:00 2001 From: AlanCoding Date: Wed, 23 Nov 2016 16:06:23 -0500 Subject: [PATCH] change unit tests to work with JobTemplateMixin serializer --- .../unit/api/serializers/test_job_template_serializers.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/awx/main/tests/unit/api/serializers/test_job_template_serializers.py b/awx/main/tests/unit/api/serializers/test_job_template_serializers.py index 77c7e69dc4..c48633573c 100644 --- a/awx/main/tests/unit/api/serializers/test_job_template_serializers.py +++ b/awx/main/tests/unit/api/serializers/test_job_template_serializers.py @@ -11,6 +11,7 @@ from awx.main.models import ( Role, User, Job, + JobTemplate, ) from rest_framework.test import APIRequestFactory @@ -24,7 +25,9 @@ def mock_JT_resource_data(): @pytest.fixture def job_template(mocker): - mock_jt = mocker.MagicMock(pk=5) + mock_jt = mocker.MagicMock(spec=JobTemplate) + mock_jt.pk = 5 + mock_jt.host_config_key = '9283920492' mock_jt.resource_validation_data = mock_JT_resource_data return mock_jt