conftest.py

import pytest
import os
import sys

from djwebdapp.models import Account, Blockchain
from djwebdapp_multisig.models import MultisigContract
from djwebdapp_tezos.models import TezosTransaction


def evil(path, *scripts, variables=None):
    """
    100% Evil code so that test code be documentation too.

    Example usage::

        variables = include(
            'djwebdapp_example/tezos',
            'client', 'deploy', 'blockchain', 'index', 'normalize',
        )

    Or (old syntax)::

        variables = {}
        evil('djwebdapp_example/tezos_deploy.py', variables)
        evil('djwebdapp_example/tezos_index.py', variables)
    """
    variables = variables if variables is not None else {}

    def abspath(script):
        return os.path.abspath(os.path.join(path, f'{script}.py'))

    path = os.path.join(
        os.path.dirname(__file__),
        'src',
        path,
    )
    for number, script in enumerate(scripts, start=0):
        script_path = abspath(script)
        with open(script_path) as f:
            source = f.read()
        code = compile(source, script_path, 'exec')
        try:
            exec(code, variables, variables)
        except:
            if number:
                print('Executed successfuly:')
                for i in range(0, number):
                    print(abspath(scripts[i]))
                print('Failed:\n' + script_path)
            exctype, exc, tb = sys.exc_info()
            raise exctype(exc).with_traceback(tb)
    return variables


@pytest.fixture
def include():
    return evil


@pytest.fixture
def admin_smoketest(admin_client):
    def _():
        urls = (
            '/admin/djwebdapp/account/',
            '/admin/djwebdapp/account/add/',
            '/admin/djwebdapp/blockchain/',
            '/admin/djwebdapp/blockchain/add/',
            '/admin/djwebdapp/node/',
            '/admin/djwebdapp/node/add/',
            '/admin/djwebdapp/transaction/',
            '/admin/djwebdapp/transaction/add/',
            '/admin/djwebdapp_tezos/tezostransaction/',
            '/admin/djwebdapp_tezos/tezostransaction/add/',
            '/admin/djwebdapp_ethereum/ethereumtransaction/',
            '/admin/djwebdapp_ethereum/ethereumtransaction/add/',
        )
        for url in urls:
            assert admin_client.get(url).status_code == 200
    return _


@pytest.fixture
@pytest.mark.django_db
def deploy_and_index():
    def f(transaction, no_assert=False):
        res = transaction.blockchain.provider.spool()
        if not no_assert:
            assert res == transaction
        transaction.blockchain.wait()
        transaction.blockchain.provider.index()
        transaction.refresh_from_db()

    return f


@pytest.fixture
@pytest.mark.django_db
def blockchain():
    return Blockchain.objects.create(
        provider_class='djwebdapp.provider.Success',
    )


@pytest.fixture
@pytest.mark.django_db
def account(blockchain):
    return Account.objects.create(
        address='testacc',
        blockchain=blockchain,
    )