You've already forked directdnsonly
Aaron Guise
8c1c2b4abc
- SQLAlchemy 1.4 → 2.0.46: migrate all session.query() calls to select() / session.execute() style; move declarative_base import from ext.declarative to sqlalchemy.orm; explicit conn.commit() after DDL in _migrate(); drop sessionmaker(bind=) keyword - persist-queue 1.0 → 1.1, pymysql 1.1.1 → 1.1.2, dnspython 2.7 → 2.8, pyyaml 6.0.2 → 6.0.3 - pytest 8.3 → 9.0.2, pytest-cov 6.1 → 7.0, pytest-mock 3.14 → 3.15.1, black 25.1 → 26.1 97 tests pass, zero deprecation warnings
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
"""Shared test fixtures for directdnsonly test suite."""
|
|
|
|
import pytest
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from directdnsonly.app.db import Base
|
|
from directdnsonly.app.db.models import (
|
|
Domain,
|
|
Key,
|
|
) # noqa: F401 — registers models with Base
|
|
|
|
|
|
@pytest.fixture
|
|
def engine():
|
|
eng = create_engine("sqlite:///:memory:")
|
|
Base.metadata.create_all(eng)
|
|
yield eng
|
|
eng.dispose()
|
|
|
|
|
|
@pytest.fixture
|
|
def db_session(engine):
|
|
session = sessionmaker(engine)()
|
|
yield session
|
|
session.close()
|
|
|
|
|
|
@pytest.fixture
|
|
def patch_connect(db_session, monkeypatch):
|
|
"""Patch connect() at every call-site, returning the shared test session.
|
|
|
|
Modules that import connect() directly (e.g. utils, reconciler) are
|
|
patched at their local name so the in-memory SQLite session is used
|
|
instead of trying to read from vyper config.
|
|
"""
|
|
_factory = lambda: db_session # noqa: E731
|
|
monkeypatch.setattr("directdnsonly.app.utils.connect", _factory)
|
|
monkeypatch.setattr("directdnsonly.app.reconciler.connect", _factory)
|
|
monkeypatch.setattr("directdnsonly.app.peer_sync.connect", _factory)
|
|
return db_session
|