d539ca9bab
* feat(sql): swap mattn/sqlite with modernc.org/sqlite (#9325) * feat(sql): swap mattn/sqlite with modernc.org/sqlite * feat(sql): revert the dashboard testing changes * feat(sql): enable WAL mode for sqlite * feat(sql): revert enable WAL mode for sqlite * feat(sql): use sensible defaults for busy_timeout * feat(sql): add ldflags * feat(sql): enable WAL mode for sqlite * feat(sql): some fixes * feat(sql): some fixes * feat(sql): fix yarn lock and config defaults * feat(sql): update the defaults in example.conf * feat(sql): remove wal mode from integration tests
82 lines
2.1 KiB
Python
82 lines
2.1 KiB
Python
from collections import namedtuple
|
|
from typing import Any, Generator
|
|
|
|
import pytest
|
|
from sqlalchemy import create_engine, sql
|
|
|
|
from fixtures import dev, types
|
|
|
|
ConnectionTuple = namedtuple("ConnectionTuple", "connection config")
|
|
|
|
|
|
@pytest.fixture(name="sqlite", scope="package")
|
|
def sqlite(
|
|
tmpfs: Generator[types.LegacyPath, Any, None],
|
|
request: pytest.FixtureRequest,
|
|
pytestconfig: pytest.Config,
|
|
) -> types.TestContainerSQL:
|
|
"""
|
|
Package-scoped fixture for SQLite.
|
|
"""
|
|
|
|
def create() -> types.TestContainerSQL:
|
|
tmpdir = tmpfs("sqlite")
|
|
path = tmpdir / "signoz.db"
|
|
|
|
engine = create_engine(f"sqlite:///{path}")
|
|
with engine.connect() as conn:
|
|
result = conn.execute(sql.text("SELECT 1"))
|
|
assert result.fetchone()[0] == 1
|
|
|
|
|
|
return types.TestContainerSQL(
|
|
container=types.TestContainerDocker(
|
|
id="",
|
|
host_configs={},
|
|
container_configs={},
|
|
),
|
|
conn=engine,
|
|
env={
|
|
"SIGNOZ_SQLSTORE_PROVIDER": "sqlite",
|
|
"SIGNOZ_SQLSTORE_SQLITE_PATH": str(path),
|
|
},
|
|
)
|
|
|
|
def delete(_: types.TestContainerSQL):
|
|
pass
|
|
|
|
def restore(cache: dict) -> types.TestContainerSQL:
|
|
path = cache["env"].get("SIGNOZ_SQLSTORE_SQLITE_PATH")
|
|
|
|
engine = create_engine(f"sqlite:///{path}")
|
|
with engine.connect() as conn:
|
|
result = conn.execute(sql.text("SELECT 1"))
|
|
assert result.fetchone()[0] == 1
|
|
|
|
|
|
return types.TestContainerSQL(
|
|
container=types.TestContainerDocker(
|
|
id="",
|
|
host_configs={},
|
|
container_configs={},
|
|
),
|
|
conn=engine,
|
|
env=cache["env"],
|
|
)
|
|
|
|
return dev.wrap(
|
|
request,
|
|
pytestconfig,
|
|
"sqlite",
|
|
lambda: types.TestContainerSQL(
|
|
container=types.TestContainerDocker(
|
|
id="", host_configs={}, container_configs={}
|
|
),
|
|
conn=None,
|
|
env={},
|
|
),
|
|
create,
|
|
delete,
|
|
restore,
|
|
)
|