"""Factory for creating rejection notes in tests."""

import datetime

import factory

from product_workflow.models import rejection_notes


class RejectionNoteFactory(factory.Factory):
    """Factory for test rejection notes."""

    class Meta:
        """Meta class around the rejection notes model."""

        model = rejection_notes.RejectionNote

    rejection_id = factory.Sequence(lambda n: n)
    release_approval_id = factory.Sequence(lambda n: n)
    table_name = factory.Sequence(lambda n: 'Test table_name {}'.format(n))
    field_name = factory.Sequence(lambda n: 'Test field_name {}'.format(n))
    comments = factory.Sequence(lambda n: 'Test comments {}'.format(n))
    corrected = 'N'
    date_added = datetime.datetime(2017, 6, 10, 0, 0, 0)
    key_id = factory.Sequence(lambda n: n)