attract/tests/abstract_attract_test.py

from pillar.tests import PillarTestServer, AbstractPillarTest

from attract import AttractExtension
from attract.setup import setup_for_attract


class AttractTestServer(PillarTestServer):
    def __init__(self, *args, **kwargs):
        PillarTestServer.__init__(self, *args, **kwargs)

        self.load_extension(AttractExtension(), '/attract')


class AbstractAttractTest(AbstractPillarTest):
    pillar_server_class = AttractTestServer

    def ensure_project_exists(self, project_overrides=None):
        project_overrides = dict(
            picture_header=None,
            picture_square=None,
            **(project_overrides or {})
        )
        proj_id, project = AbstractPillarTest.ensure_project_exists(self, project_overrides)

        with self.app.test_request_context():
            attract_project = setup_for_attract(project['url'], replace=True)

        return proj_id, attract_project