Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 4965d374a5 | |||
| 24d1b199ea | |||
| 46906cc333 | |||
| cb180b511f |
+8
-21
@@ -51,21 +51,12 @@ def disable_rate_limiting():
|
|||||||
|
|
||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def engine():
|
def engine():
|
||||||
"""Sync in-memory SQLite engine for model unit tests.
|
"""Sync in-memory SQLite engine for model unit tests."""
|
||||||
|
|
||||||
Strips PostgreSQL-specific server_default expressions so SQLite can
|
|
||||||
handle all column inserts without missing-function errors.
|
|
||||||
"""
|
|
||||||
eng = create_engine("sqlite:///:memory:")
|
eng = create_engine("sqlite:///:memory:")
|
||||||
|
from cartsnitch_api.models.user import User
|
||||||
|
|
||||||
for table in Base.metadata.tables.values():
|
col = User.__table__.columns["email_inbound_token"]
|
||||||
for col in table.columns.values():
|
col.server_default = None
|
||||||
sd = col.server_default
|
|
||||||
if sd is not None:
|
|
||||||
expr_str = str(sd.expression).lower()
|
|
||||||
if "gen_random_uuid" in expr_str or "gen_random_bytes" in expr_str:
|
|
||||||
col.server_default = None
|
|
||||||
|
|
||||||
Base.metadata.create_all(eng)
|
Base.metadata.create_all(eng)
|
||||||
yield eng
|
yield eng
|
||||||
eng.dispose()
|
eng.dispose()
|
||||||
@@ -89,16 +80,12 @@ async def db_engine():
|
|||||||
cursor.execute("PRAGMA foreign_keys=ON")
|
cursor.execute("PRAGMA foreign_keys=ON")
|
||||||
cursor.close()
|
cursor.close()
|
||||||
|
|
||||||
for table in Base.metadata.tables.values():
|
|
||||||
for col in table.columns.values():
|
|
||||||
sd = col.server_default
|
|
||||||
if sd is not None:
|
|
||||||
expr_str = str(sd.expression).lower()
|
|
||||||
if "gen_random_uuid" in expr_str or "gen_random_bytes" in expr_str:
|
|
||||||
col.server_default = None
|
|
||||||
|
|
||||||
async with engine.begin() as conn:
|
async with engine.begin() as conn:
|
||||||
|
from cartsnitch_api.models.user import User
|
||||||
|
|
||||||
|
User.__table__.columns["email_inbound_token"].server_default = None
|
||||||
await conn.run_sync(Base.metadata.create_all)
|
await conn.run_sync(Base.metadata.create_all)
|
||||||
|
# Create Better-Auth tables (not managed by SQLAlchemy models)
|
||||||
await conn.execute(
|
await conn.execute(
|
||||||
text("""
|
text("""
|
||||||
CREATE TABLE IF NOT EXISTS sessions (
|
CREATE TABLE IF NOT EXISTS sessions (
|
||||||
|
|||||||
@@ -17,15 +17,6 @@ from cartsnitch_api.models.user import User, UserStoreAccount
|
|||||||
@pytest.fixture
|
@pytest.fixture
|
||||||
def engine():
|
def engine():
|
||||||
eng = create_engine("sqlite:///:memory:")
|
eng = create_engine("sqlite:///:memory:")
|
||||||
|
|
||||||
for table in Base.metadata.tables.values():
|
|
||||||
for col in table.columns.values():
|
|
||||||
sd = col.server_default
|
|
||||||
if sd is not None:
|
|
||||||
expr_str = str(sd.expression).lower()
|
|
||||||
if "gen_random_uuid" in expr_str or "gen_random_bytes" in expr_str:
|
|
||||||
col.server_default = None
|
|
||||||
|
|
||||||
Base.metadata.create_all(eng)
|
Base.metadata.create_all(eng)
|
||||||
yield eng
|
yield eng
|
||||||
eng.dispose()
|
eng.dispose()
|
||||||
|
|||||||
Reference in New Issue
Block a user