"""Factory for creating history's job records in tests."""

import datetime

import factory

from salessheets.constants import salessheets
from salessheets.models import history


DEFAULT_USER = 'oa:1234'


class JobFactory(factory.Factory):
    """Factory for test history's job."""

    class Meta:
        """Meta class around the history's job model."""

        model = history.Job

    job_id = factory.Sequence(lambda n: n + 1)
    user_id = DEFAULT_USER
    context = '123415123,12315136713'
    context_type = salessheets.DISPLAY_UPC
    output_format = salessheets.ALLOWED_GENERATION_METHODS[0]
    status = salessheets.REQUESTED
    timestamp = factory.Sequence(lambda n: datetime.datetime.utcnow())