mimesis-factory

account.py

import datetime
import random


class Account(object):
    def __init__(self, username, email):
        self.username = username
        self.email = email
        self.password = ''.join(str(random.randint(0, 10)) for _ in range(10))
        self.date_joined = datetime.datetime.today()

    def __str__(self):
        return '{} ({})'.format(self.username, self.email)

Pipfile

[[source]]

url = "https://pypi.python.org/simple"
verify_ssl = true
name = "pypi"


[packages]

mimesis = "*"
factory-boy = "==2.8.1"
pytest = "*"
pytest-factoryboy = "*"


[dev-packages]



[requires]

python_version = "3.6"

test_basic.py

import factory
from pytest_factoryboy import register

from account import Account


@register
class AccountFactory(factory.Factory):
    class Meta:
        model = Account

    username = factory.Sequence(lambda n: 'john%s' % n)
    email = factory.LazyAttribute(lambda o: '%[email protected]' % o.username)


def test_account_exists(account):
    assert account.username == 'john0'
    assert account.email == '[email protected]'

test_idea.py

# =======
# Now with mimesis

import factory
from factory import declarations
from mimesis import Generic
from pytest_factoryboy import register

from account import Account


class Mimesis(declarations.OrderedDeclaration):
    def __init__(self, provider, field, locale=None, **kwargs):
        self.locale = locale
        self.mimesis = Generic(self.locale)
        self.provider = provider
        self.provider_kwargs = kwargs
        self.field = field

    def generate(self, extra_kwargs):
        print(extra_kwargs)
        provider = getattr(self.mimesis, self.provider)
        field = getattr(provider, self.field)()
        return field

    def evaluate(self, sequence, obj, create, extra=None, containers=()):
        print(sequence, obj, create, extra, containers)
        return self.generate(extra or {})


@register
class AccountFactory(factory.Factory):
    class Meta:
        model = Account

    username = Mimesis('personal', 'username')
    email = factory.LazyAttribute(
        lambda o: '%[email protected]' % o.username
    )


def test_account_exists(account):
    assert account.username != 'john0'
    assert account.email != '[email protected]'