import factory
from factory import fuzzy

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


class StarredContentFactory(factory.alchemy.SQLAlchemyModelFactory):
    id = factory.Faker("pyint")
    user_id = fuzzy.FuzzyText(length=45)
    uri = fuzzy.FuzzyText(length=45)
    isrc = fuzzy.FuzzyText(length=15)

    class Meta:
        model = StarredContent
        sqlalchemy_session_persistence = "commit"
        sqlalchemy_session = session