From 535f979fd7aca2e293cfb672b36cb32aff4443dd Mon Sep 17 00:00:00 2001
From: Aaron Tan <jangsutsr@gmail.com>
Date: Thu, 22 Sep 2016 22:34:40 -0400
Subject: [PATCH] Unit test added.

---
 .../unit/models/test_job_template_unit.py     | 29 +++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/awx/main/tests/unit/models/test_job_template_unit.py b/awx/main/tests/unit/models/test_job_template_unit.py
index a156d1e920..9720692f1f 100644
--- a/awx/main/tests/unit/models/test_job_template_unit.py
+++ b/awx/main/tests/unit/models/test_job_template_unit.py
@@ -50,3 +50,32 @@ def test_job_template_survey_password_redaction(job_template_with_survey_passwor
     """Tests the JobTemplate model's funciton to redact passwords from
     extra_vars - used when creating a new job"""
     assert job_template_with_survey_passwords_unit.survey_password_variables() == ['secret_key', 'SSN']
+
+def test_job_template_survey_variable_validation(job_template_factory):
+    objects = job_template_factory(
+        'survey_variable_validation',
+        organization='org1',
+        inventory='inventory1',
+        credential='cred1',
+        persisted=False,
+    )
+    obj = objects.job_template
+    obj.survey_spec = {
+    "description": "",
+    "spec": [
+        {
+            "required": True,
+            "min": 0,
+            "default": "5",
+            "max": 1024,
+            "question_description": "",
+            "choices": "",
+            "variable": "a",
+            "question_name": "Whosyourdaddy",
+            "type": "text"
+        }
+    ],
+    "name": ""
+    }
+    obj.survey_enabled = True
+    assert obj.survey_variable_validation({"a": 5}) == ["Value 5 for 'a' expected to be a string."]