import datetime
import factory
from factory import fuzzy

from apollo_main_db.apollo import models
from src.db.base import session


class ChartsMonitoringFactory(factory.alchemy.SQLAlchemyModelFactory):
    """Fake model for tblChartsMonitoring."""

    # id = Column("Id", BigInteger, primary_key=True)
    dsp = fuzzy.FuzzyText(length=10)
    chart_type = fuzzy.FuzzyText(length=10)
    breakdown = fuzzy.FuzzyText(length=10)
    market = fuzzy.FuzzyText(length=2)
    chart_id = fuzzy.FuzzyText(length=20)
    updated_at = factory.LazyFunction(datetime.datetime.utcnow())
    chart_datetime = factory.LazyFunction(datetime.datetime.utcnow())
    data = fuzzy.FuzzyText()

    class Meta:
        model = models.ChartsMonitoring
        sqlalchemy_session_persistence = "commit"
        sqlalchemy_session = session