from fansifter_common.testing.factories import SQLAlchemyFactory

from app.dsp.enums import DSPId
from app.dsp.models import DSPClient
from app.fandata.models import FanCredentials
from app.pipeline.models import PipelineRun


class DSPClientFactory(SQLAlchemyFactory[DSPClient]):
    dsp_id = DSPId.spotify


class PipelineRunFactory(SQLAlchemyFactory[PipelineRun]):
    filters = None


class FanCredentialsFactory(SQLAlchemyFactory[FanCredentials]):
    refresh_token_encrypted = "placeholder"