move test data to a separate repo root dir

To make pytest happier when running on it's own since it prefers tests
in a root-level dir.

1 files changed, 112 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 00000000..0a1838cd
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,112 @@
+import os
+import tempfile
+import textwrap
+from contextlib import ExitStack
+from unittest.mock import patch
+
+import pytest
+from pkgcheck.cli import Tool
+from pkgcheck.reporters import StrReporter
+from pkgcheck.results import Result
+from pkgcheck.scripts import pkgcheck
+from pkgcore import const as pkgcore_const
+from snakeoil.formatters import PlainTextFormatter
+from snakeoil.osutils import pjoin
+
+pytest_plugins = ['pkgcore']
+REPO_ROOT = os.path.dirname(os.path.dirname(__file__))
+
+
+def pytest_configure():
+    # export repo root for test modules to use
+    pytest.REPO_ROOT = REPO_ROOT
+
+
+def pytest_assertrepr_compare(op, left, right):
+    """Custom assertion failure output."""
+    if isinstance(left, Result) and isinstance(right, Result) and op == "==":
+        with tempfile.TemporaryFile() as f:
+            with StrReporter(out=PlainTextFormatter(f)) as reporter:
+                reporter.report(left)
+                reporter.report(right)
+                f.seek(0)
+                left_val, right_val = f.read().decode().splitlines()
+        return ["Result instances !=:", left_val, right_val]
+
+
+@pytest.fixture(scope="session", autouse=True)
+def default_session_fixture(request):
+    """Fixture run globally for the entire test session."""
+    stack = ExitStack()
+    # don't load the default system or user config files
+    stack.enter_context(patch('pkgcheck.cli.ConfigFileParser.default_configs', ()))
+
+    def unpatch():
+        stack.close()
+
+    request.addfinalizer(unpatch)
+
+
+@pytest.fixture(scope="session")
+def stubconfig():
+    """Generate a portage config that sets the default repo to pkgcore's stubrepo."""
+    return pjoin(pkgcore_const.DATA_PATH, 'stubconfig')
+
+
+@pytest.fixture(scope="session")
+def testconfig(tmp_path_factory):
+    """Generate a portage config that sets the default repo to the bundled standalone repo.
+
+    Also, repo entries for all the bundled test repos.
+    """
+    config = tmp_path_factory.mktemp('testconfig')
+    repos_conf = config / 'repos.conf'
+    stubrepo = pjoin(pkgcore_const.DATA_PATH, 'stubrepo')
+    testdir = pjoin(REPO_ROOT, 'testdata', 'repos')
+    with open(repos_conf, 'w') as f:
+        f.write(textwrap.dedent(f"""\
+            [DEFAULT]
+            main-repo = standalone
+            [stubrepo]
+            location = {stubrepo}
+        """))
+        for repo in os.listdir(testdir):
+            f.write(f'[{repo}]\n')
+            f.write(f'location = {pjoin(testdir, repo)}\n')
+    profile_path = pjoin(stubrepo, 'profiles', 'default')
+    os.symlink(profile_path, str(config / 'make.profile'))
+    return str(config)
+
+
+@pytest.fixture(scope="session")
+def cache_dir(tmp_path_factory):
+    """Generate a cache directory for pkgcheck."""
+    cache_dir = tmp_path_factory.mktemp('cache')
+    return str(cache_dir)
+
+
+@pytest.fixture
+def fakerepo(tmp_path_factory):
+    """Generate a stub repo."""
+    fakerepo = str(tmp_path_factory.mktemp('fakerepo'))
+    os.makedirs(pjoin(fakerepo, 'profiles'))
+    os.makedirs(pjoin(fakerepo, 'metadata'))
+    with open(pjoin(fakerepo, 'profiles', 'repo_name'), 'w') as f:
+        f.write('fakerepo\n')
+    with open(pjoin(fakerepo, 'metadata', 'layout.conf'), 'w') as f:
+        f.write('masters =\n')
+    return fakerepo
+
+
+@pytest.fixture(scope="session")
+def tool(testconfig):
+    """Generate a tool utility for running pkgcheck."""
+    tool = Tool(pkgcheck.argparser)
+    tool.parser.set_defaults(override_config=testconfig)
+    return tool
+
+
+@pytest.fixture
+def parser():
+    """Return a shallow copy of the main pkgcheck argparser."""
+    return pkgcheck.argparser.copy()