Skip to content

Commit

Permalink
Add runner tests from ansible core test suite
Browse files Browse the repository at this point in the history
  • Loading branch information
Shrews committed Jan 4, 2024
1 parent 309baad commit d67f44a
Show file tree
Hide file tree
Showing 2 changed files with 143 additions and 0 deletions.
59 changes: 59 additions & 0 deletions test/integration/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -125,3 +125,62 @@ def container_image(request, cli, tmp_path): # pylint: disable=W0621
[runtime, 'rmi', '-f', image_name],
bare=True,
)


@pytest.fixture
def container_image_devel(request, cli, tmp_path): # pylint: disable=W0621
branch = request.getfixturevalue('branch')

DOCKERFILE = f"""
FROM quay.io/centos/centos:stream9
ARG WHEEL
COPY $WHEEL /$WHEEL
# Need python 3.11 minimum for devel
RUN dnf install -y python3.11 python3.11-pip git
RUN alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 0
RUN python3 -m pip install /$WHEEL git+https://github.com/ansible/ansible@{branch}
RUN mkdir -p /runner/{{env,inventory,project,artifacts}} /home/runner/.ansible/tmp
RUN chmod -R 777 /runner /home/runner
WORKDIR /runner
ENV HOME=/home/runner
CMD ["ansible-runner", "run", "/runner"]
"""

try:
containerized = request.getfixturevalue('containerized')
if not containerized:
yield None
return
except Exception:
# Test func doesn't use containerized
pass

if (env_image_name := os.getenv('RUNNER_TEST_IMAGE_NAME')):
yield env_image_name
return

cli(
['pyproject-build', '-w', '-o', str(tmp_path)],
cwd=here.parent.parent,
bare=True,
)

wheel = next(tmp_path.glob('*.whl')) # pylint: disable=R1708

runtime = request.getfixturevalue('runtime')
dockerfile_path = tmp_path / 'Dockerfile'
dockerfile_path.write_text(DOCKERFILE)
random_string = ''.join(random.choice(ascii_lowercase) for i in range(10))
image_name = f'ansible-runner-{random_string}-event-test'

cli(
[runtime, 'build', '--build-arg', f'WHEEL={wheel.name}', '--rm=true', '-t', image_name, '-f', str(dockerfile_path), str(tmp_path)],
bare=True,
)
yield image_name
cli(
[runtime, 'rmi', '-f', image_name],
bare=True,
)
84 changes: 84 additions & 0 deletions test/integration/test_core_integration.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,84 @@
import sys
import pytest

from ansible_runner.interface import run


TEST_BRANCHES = (
'devel',
'milestone',
'stable-2.16', # current stable
'stable-2.15', # stable - 1
)


@pytest.mark.test_all_runtimes
@pytest.mark.parametrize('branch', TEST_BRANCHES)
@pytest.mark.skipif(sys.platform == 'darwin', reason='does not work on macOS')
def test_adhoc(tmp_path, runtime, branch, container_image_devel): # pylint: disable=W0613
# pvt_data_dir is mounted on the container, so it must contain the expected directories
project_dir = tmp_path / 'project'
project_dir.mkdir()
r = run(private_data_dir=str(tmp_path),
host_pattern='localhost',
module='shell',
module_args='pwd',
process_isolation_executable=runtime,
process_isolation=True,
container_image=container_image_devel,
)

assert r.status == 'successful'
assert r.rc == 0
assert 'ok' in r.stats
assert 'localhost' in r.stats['ok']
events = [x['event'] for x in r.events if x['event'] != 'verbose']
assert len(events) == 4


@pytest.mark.test_all_runtimes
@pytest.mark.parametrize('branch', TEST_BRANCHES)
@pytest.mark.skipif(sys.platform == 'darwin', reason='does not work on macOS')
def test_playbook(tmp_path, runtime, branch, container_image_devel): # pylint: disable=W0613
PLAYBOOK = """
- hosts: localhost
gather_facts: False
tasks:
- set_fact:
foo: bar
"""

# pvt_data_dir is mounted on the container, so it must contain the expected directories
project_dir = tmp_path / 'project'
project_dir.mkdir()
inventory_dir = tmp_path / 'inventory'
inventory_dir.mkdir()

hosts_file = inventory_dir / 'hosts'
hosts_file.write_text('localhost\n')

playbook = project_dir / 'test.yml'
playbook.write_text(PLAYBOOK)

r = run(private_data_dir=str(tmp_path),
playbook='test.yml',
process_isolation_executable=runtime,
process_isolation=True,
container_image=container_image_devel,
)

expected_events = [
'playbook_on_start',
'playbook_on_play_start',
'playbook_on_task_start',
'runner_on_start',
'runner_on_ok',
'playbook_on_stats',
]

assert r.status == 'successful'
assert r.rc == 0
assert 'ok' in r.stats
assert 'localhost' in r.stats['ok']
events = [x['event'] for x in r.events if x['event'] != 'verbose']
assert events == expected_events

0 comments on commit d67f44a

Please sign in to comment.