import factory
from api import models


class DesignerFactory(factory.Factory):

    class Meta:
        model = models.Designer

    name = 'Spiderman'
    username = 'spiderman'
    territory = 'US'
    currency = 'USD'


class ContractCustomerFactory(factory.Factory):

    class Meta:
        model = models.ContractCustomer

    pricepremium = 0


class InquiryFactory(factory.Factory):

    class Meta:
        model = models.Inquiry

    price_retouch = None
    pricepremium = 0