"""Factories for testing models."""

from factory.alchemy import SQLAlchemyModelFactory

from royalty_common.connectors.database import db
from royalty_common.utils.dates import current_timestamp


class BaseModelFactory(SQLAlchemyModelFactory):
    """Base model factory."""

    class Meta:
        """Configures the session."""

        sqlalchemy_session = db.session
        sqlalchemy_session_persistence = 'commit'


class BaseModelFactoryWithTimestamps(BaseModelFactory):
    """Base model factory with timestamp fields."""

    created_at = current_timestamp()
    last_modified = current_timestamp()