"""Fake factories."""

from datetime import datetime, timedelta

from apollo_main_db.apollo import models
import factory
from factory import fuzzy

from track_visit_events_handling import config


class ApolloVisitLogFactory(factory.alchemy.SQLAlchemyModelFactory):
    """Fake model for ApolloVisitLog."""
    id = fuzzy.FuzzyInteger(100)
    user_id = fuzzy.FuzzyText(length=37)
    dsp_id = fuzzy.FuzzyInteger(1, 2)
    type_id = fuzzy.FuzzyInteger(1, 2)
    instance_id = fuzzy.FuzzyText(length=22)
    created_at = fuzzy.FuzzyNaiveDateTime(datetime.utcnow() - timedelta(days=config.TRACK_VISITS_LOG_RETENTION_DAYS))

    class Meta:
        model = models.ApolloVisitLog
        sqlalchemy_session_persistence = "commit"