"""Helpers for mocking the python-bugzilla package in integration tests."""
import json
from pathlib import Path
MOCK_BUGS_PATH = "/tmp/anaconda-bugzilla-mock-bugs.json"
MOCK_SCRIPTS_DIR = Path(__file__).parent.parent / "mock_bugzilla_scripts"
def _python_cmd(machine):
return machine.execute(
"command -v /usr/libexec/platform-python || command -v python3"
).strip()
def _site_packages(machine):
py = _python_cmd(machine)
return machine.execute(
f"{py} -c 'import sysconfig; print(sysconfig.get_paths()[\"purelib\"])'"
).strip()
def install_mock_bugzilla(machine, bugs=None):
"""Install a mock bugzilla package and optional canned search results.
:param machine: test machine
:param bugs: list of dicts with id/summary/status, or empty list for no matches
"""
if bugs is None:
bugs = []
machine.write(MOCK_BUGS_PATH, json.dumps(bugs))
site = _site_packages(machine)
machine.execute(f"""
set -eu
SITE="{site}"
if [ -e "$SITE/bugzilla" ] && [ ! -e "$SITE/bugzilla.anaconda-test-bak" ]; then
mv "$SITE/bugzilla" "$SITE/bugzilla.anaconda-test-bak"
fi
if [ -e "$SITE/bugzilla.py" ] && [ ! -e "$SITE/bugzilla.py.anaconda-test-bak" ]; then
mv "$SITE/bugzilla.py" "$SITE/bugzilla.py.anaconda-test-bak"
fi
mkdir -p "$SITE/bugzilla"
""")
mock_bugzilla_init = (MOCK_SCRIPTS_DIR / "bugzilla_init.py").read_text()
machine.write(f"{site}/bugzilla/__init__.py", mock_bugzilla_init)
def set_mock_bugzilla_bugs(machine, bugs):
"""Update canned search results for an already installed mock."""
machine.write(MOCK_BUGS_PATH, json.dumps(bugs))
def uninstall_mock_bugzilla(machine):
"""Restore the original bugzilla package if it was backed up."""
site = _site_packages(machine)
machine.execute(f"""
set -eu
SITE="{site}"
rm -rf "$SITE/bugzilla"
if [ -e "$SITE/bugzilla.anaconda-test-bak" ]; then
mv "$SITE/bugzilla.anaconda-test-bak" "$SITE/bugzilla"
fi
if [ -e "$SITE/bugzilla.py.anaconda-test-bak" ]; then
mv "$SITE/bugzilla.py.anaconda-test-bak" "$SITE/bugzilla.py"
fi
rm -f "{MOCK_BUGS_PATH}"
""")
