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