"""Factory for creating territory records in tests."""

import factory

from product_digital_marketing.models import territory


class TerritoryFactory(factory.Factory):
    """Factory for test highlights."""

    class Meta:
        """Meta class around the territory model."""

        model = territory.Territory

    pk = factory.Sequence(lambda n: n + 1)
    name = factory.Sequence(
        lambda n: str('Territory Name {}'.format(n + 1)))
    category = 'Country'