"""Fixtures for push-notifications models"""
import factory
from apollo_main_db.push_notifications import models
from factory import fuzzy


class UserDeviceFactory(factory.alchemy.SQLAlchemyModelFactory):
    """Fake model for UserDeviceToken"""
    user_id = fuzzy.FuzzyText(length=37)
    token = fuzzy.FuzzyText(length=255)
    is_active = factory.Faker("boolean")

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