From 105175b6b44e36b7d905a029029ddef6b32d6430 Mon Sep 17 00:00:00 2001
From: AlanCoding <arominge@redhat.com>
Date: Tue, 8 Nov 2016 09:46:59 -0500
Subject: [PATCH] Fix errors in workflow surveys found through tests

---
 awx/main/models/jobs.py                          | 1 -
 awx/main/models/mixins.py                        | 1 +
 awx/main/tests/unit/models/test_survey_models.py | 1 -
 awx/main/utils.py                                | 3 ++-
 4 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/awx/main/models/jobs.py b/awx/main/models/jobs.py
index 05f02bd5af..7621649889 100644
--- a/awx/main/models/jobs.py
+++ b/awx/main/models/jobs.py
@@ -33,7 +33,6 @@ from awx.main.models.notifications import (
     JobNotificationMixin,
 )
 from awx.main.utils import (
-    decrypt_field,
     ignore_inventory_computed_fields,
     parse_yaml_or_json,
 )
diff --git a/awx/main/models/mixins.py b/awx/main/models/mixins.py
index 2dcc7befef..efe36bcefc 100644
--- a/awx/main/models/mixins.py
+++ b/awx/main/models/mixins.py
@@ -11,6 +11,7 @@ from jsonfield import JSONField
 from awx.main.models.rbac import (
     Role, RoleAncestorEntry, get_roles_on_resource
 )
+from awx.main.utils import parse_yaml_or_json
 
 
 __all__ = ['ResourceMixin', 'SurveyJobTemplateMixin', 'SurveyJobMixin']
diff --git a/awx/main/tests/unit/models/test_survey_models.py b/awx/main/tests/unit/models/test_survey_models.py
index bf6148ec55..4f9653755e 100644
--- a/awx/main/tests/unit/models/test_survey_models.py
+++ b/awx/main/tests/unit/models/test_survey_models.py
@@ -4,7 +4,6 @@ import json
 from awx.main.tasks import RunJob
 from awx.main.models import (
     Job,
-    WorkflowJob,
     WorkflowJobTemplate
 )
 
diff --git a/awx/main/utils.py b/awx/main/utils.py
index c513a9bf5d..3614a99039 100644
--- a/awx/main/utils.py
+++ b/awx/main/utils.py
@@ -5,6 +5,7 @@
 import base64
 import hashlib
 import json
+import yaml
 import logging
 import os
 import re
@@ -490,7 +491,7 @@ def parse_yaml_or_json(vars_str):
     except (ValueError, TypeError):
         try:
             vars_dict = yaml.safe_load(vars_str)
-            assert isinstance(extra_vars, dict)
+            assert isinstance(vars_dict, dict)
         except (yaml.YAMLError, TypeError, AttributeError, AssertionError):
             vars_dict = {}
     return vars_dict