mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-25 01:11:48 -05:00
64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
import pytest
|
|
from unittest import mock
|
|
import os
|
|
import tempfile
|
|
import shutil
|
|
|
|
from awx.main.tasks.system import execution_node_health_check, _cleanup_images_and_files
|
|
from awx.main.models import Instance, Job
|
|
|
|
|
|
@pytest.fixture
|
|
def scm_revision_file(tmpdir_factory):
|
|
# Returns path to temporary testing revision file
|
|
revision_file = tmpdir_factory.mktemp('revisions').join('revision.txt')
|
|
with open(str(revision_file), 'w') as f:
|
|
f.write('1234567890123456789012345678901234567890')
|
|
return os.path.join(revision_file.dirname, 'revision.txt')
|
|
|
|
|
|
@pytest.mark.django_db
|
|
@pytest.mark.parametrize('node_type', ('control', 'hybrid'))
|
|
def test_no_worker_info_on_AWX_nodes(node_type):
|
|
hostname = 'us-south-3-compute.invalid'
|
|
Instance.objects.create(hostname=hostname, node_type=node_type)
|
|
with pytest.raises(RuntimeError):
|
|
execution_node_health_check(hostname)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_job_folder(request):
|
|
pdd_path = tempfile.mkdtemp(prefix='awx_123_')
|
|
|
|
def test_folder_cleanup():
|
|
if os.path.exists(pdd_path):
|
|
shutil.rmtree(pdd_path)
|
|
|
|
request.addfinalizer(test_folder_cleanup)
|
|
|
|
return pdd_path
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_folder_cleanup_stale_file(mock_job_folder, mock_me):
|
|
_cleanup_images_and_files()
|
|
assert os.path.exists(mock_job_folder) # grace period should protect folder from deletion
|
|
|
|
_cleanup_images_and_files(grace_period=0)
|
|
assert not os.path.exists(mock_job_folder) # should be deleted
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_folder_cleanup_running_job(mock_job_folder, mock_me):
|
|
me_inst = Instance.objects.create(hostname='local_node', uuid='00000000-0000-0000-0000-000000000000')
|
|
with mock.patch.object(Instance.objects, 'me', return_value=me_inst):
|
|
|
|
job = Job.objects.create(id=123, controller_node=me_inst.hostname, status='running')
|
|
_cleanup_images_and_files(grace_period=0)
|
|
assert os.path.exists(mock_job_folder) # running job should prevent folder from getting deleted
|
|
|
|
job.status = 'failed'
|
|
job.save(update_fields=['status'])
|
|
_cleanup_images_and_files(grace_period=0)
|
|
assert not os.path.exists(mock_job_folder) # job is finished and no grace period, should delete
|