"""Factory for the Subaccount model."""

import factory

from users.models.user_info import Subaccount


class SubaccountFactory(factory.Factory):
    """Factory for creating Subaccount instances for testing."""

    class Meta:
        """Meta definition for the factory."""

        model = Subaccount

    subaccount_id = factory.Sequence(lambda n: n)
    subaccount_name = factory.Sequence(lambda n: 'subacc {}'.format(n))