1 files changed, 92 insertions, 0 deletions
diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 0000000..15bcf40
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,92 @@
+import sys
+import asyncio
+import selectors
+from typing import List
+
+pytest_plugins = (
+    "tests.fix_db",
+    "tests.fix_pq",
+    "tests.fix_mypy",
+    "tests.fix_faker",
+    "tests.fix_proxy",
+    "tests.fix_psycopg",
+    "tests.fix_crdb",
+    "tests.pool.fix_pool",
+)
+
+
+def pytest_configure(config):
+    markers = [
+        "slow: this test is kinda slow (skip with -m 'not slow')",
+        "flakey(reason): this test may fail unpredictably')",
+        # There are troubles on travis with these kind of tests and I cannot
+        # catch the exception for my life.
+        "subprocess: the test import psycopg after subprocess",
+        "timing: the test is timing based and can fail on cheese hardware",
+        "dns: the test requires dnspython to run",
+        "postgis: the test requires the PostGIS extension to run",
+    ]
+
+    for marker in markers:
+        config.addinivalue_line("markers", marker)
+
+
+def pytest_addoption(parser):
+    parser.addoption(
+        "--loop",
+        choices=["default", "uvloop"],
+        default="default",
+        help="The asyncio loop to use for async tests.",
+    )
+
+
+def pytest_report_header(config):
+    rv = []
+
+    rv.append(f"default selector: {selectors.DefaultSelector.__name__}")
+    loop = config.getoption("--loop")
+    if loop != "default":
+        rv.append(f"asyncio loop: {loop}")
+
+    return rv
+
+
+def pytest_sessionstart(session):
+    # Detect if there was a segfault in the previous run.
+    #
+    # In case of segfault, pytest doesn't get a chance to write failed tests
+    # in the cache. As a consequence, retries would find no test failed and
+    # assume that all tests passed in the previous run, making the whole test pass.
+    cache = session.config.cache
+    if cache.get("segfault", False):
+        session.warn(Warning("Previous run resulted in segfault! Not running any test"))
+        session.warn(Warning("(delete '.pytest_cache/v/segfault' to clear this state)"))
+        raise session.Failed
+    cache.set("segfault", True)
+
+    # Configure the async loop.
+    loop = session.config.getoption("--loop")
+    if loop == "uvloop":
+        import uvloop
+
+        uvloop.install()
+    else:
+        assert loop == "default"
+
+    if sys.platform == "win32":
+        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
+
+allow_fail_messages: List[str] = []
+
+
+def pytest_sessionfinish(session, exitstatus):
+    # Mark the test run successful (in the sense -weak- that we didn't segfault).
+    session.config.cache.set("segfault", False)
+
+
+def pytest_terminal_summary(terminalreporter, exitstatus, config):
+    if allow_fail_messages:
+        terminalreporter.section("failed tests ignored")
+        for msg in allow_fail_messages:
+            terminalreporter.line(msg)