from fansifter_common.testing.db import SessionMixin
from snowflake.snowpark import (
    Session as SnowparkSession,  # type: ignore[reportMissingTypeStubs]
)
from snowflake.snowpark.mock._connection import MockServerConnection

from ows_text_campaigns.adapters.db import DB, PgDB


class TestDB(SessionMixin, DB):
    def create_snowpark_session(self) -> SnowparkSession:
        return SnowparkSession(conn=MockServerConnection(options={}))


class TestPgDB(SessionMixin, PgDB):
    pass