change imports to reflect isolated->expect move

This commit is contained in:
AlanCoding
2017-08-15 11:47:29 -04:00
parent e79ca131a6
commit 6944ffa010
6 changed files with 8 additions and 8 deletions

View File

@@ -13,7 +13,7 @@ import logging
from django.conf import settings
import awx
from awx.main.isolated import run
from awx.main.expect import run
from awx.main.utils import OutputEventFilter
from awx.main.queue import CallbackQueueDispatcher

View File

@@ -51,7 +51,7 @@ from awx.main.constants import CLOUD_PROVIDERS, PRIVILEGE_ESCALATION_METHODS
from awx.main.models import * # noqa
from awx.main.models.unified_jobs import ACTIVE_STATES
from awx.main.queue import CallbackQueueDispatcher
from awx.main.isolated import run, isolated_manager
from awx.main.expect import run, isolated_manager
from awx.main.utils import (get_ansible_version, get_ssh_version, decrypt_field, update_scm_url,
check_proot_installed, build_proot_temp_dir, get_licenser,
wrap_args_with_proot, get_system_task_capacity, OutputEventFilter,

View File

@@ -13,7 +13,7 @@ from cryptography.hazmat.primitives.asymmetric import rsa
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization
from awx.main.isolated import run, isolated_manager
from awx.main.expect import run, isolated_manager
from django.conf import settings
@@ -273,7 +273,7 @@ def test_check_isolated_job(private_data_dir, rsa_key):
mgr.host = 'isolated-host'
os.mkdir(os.path.join(private_data_dir, 'artifacts'))
with mock.patch('awx.main.isolated.run.run_pexpect') as run_pexpect:
with mock.patch('awx.main.expect.run.run_pexpect') as run_pexpect:
def _synchronize_job_artifacts(args, cwd, env, buff, **kw):
buff.write('checking job status...')
@@ -325,7 +325,7 @@ def test_check_isolated_job_timeout(private_data_dir, rsa_key):
mgr.started_at = time.time()
mgr.host = 'isolated-host'
with mock.patch('awx.main.isolated.run.run_pexpect') as run_pexpect:
with mock.patch('awx.main.expect.run.run_pexpect') as run_pexpect:
def _synchronize_job_artifacts(args, cwd, env, buff, **kw):
buff.write('checking job status...')

View File

@@ -195,7 +195,7 @@ class TestJobExecution:
# don't emit websocket statuses; they use the DB and complicate testing
mock.patch.object(UnifiedJob, 'websocket_emit_status', mock.Mock()),
mock.patch.object(Job, 'inventory', mock.Mock(pk=1, spec_set=['pk'])),
mock.patch('awx.main.isolated.run.run_pexpect', self.run_pexpect)
mock.patch('awx.main.expect.run.run_pexpect', self.run_pexpect)
]
for p in self.patches:
p.start()