"""Fixtures for push-notifications models"""
import factory
from factory import fuzzy

from apollo_main_db.push_notifications import models


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'