from snowflake.snowpark import (
    Session as SnowparkSession,
)
from snowflake.snowpark.mock._connection import MockServerConnection

from dmp.adapters.db import DefaultDB, ReportingDB


class TestDefaultDB(DefaultDB):
    pass


class TestReportingDB(ReportingDB):
    def create_snowpark_session(self) -> SnowparkSession:
        return SnowparkSession(conn=MockServerConnection(options={}))