Fix some bugs found from unit tests

This commit is contained in:
Matthew Jones 2014-03-14 15:54:27 -04:00
parent 382e1e96c7
commit 28c9d77f29
2 changed files with 4 additions and 1 deletions

View File

@ -34,7 +34,7 @@ from jsonfield import JSONField
# AWX
from awx.main.models.base import *
from awx.main.utils import encrypt_field
from awx.main.utils import encrypt_field, decrypt_field
# Celery
from celery import chain

View File

@ -836,6 +836,9 @@ class RunJobTest(BaseCeleryTest):
self.assertTrue(job.passwords_needed_to_start)
self.assertTrue('ssh_key_unlock' in job.passwords_needed_to_start)
self.assertFalse(job.signal_start())
job.status = 'failed'
job.save()
job = self.create_test_job(job_template=job_template)
self.assertEqual(job.status, 'new')
self.assertTrue(job.signal_start(ssh_key_unlock=TEST_SSH_KEY_DATA_UNLOCK))
job = Job.objects.get(pk=job.pk)