abstract_attract_test.py
8470efbcf114master

abstract_attract_test.py
View Options

			from pillar.tests import PillarTestServer, AbstractPillarTest

			MOCK_SVN_URL = 'svn://biserver/mocked'


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

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


			class AbstractAttractTest(AbstractPillarTest):
			pillar_server_class = AttractTestServer

			def tearDown(self):
			self.unload_modules('attract')
			AbstractPillarTest.tearDown(self)

			def ensure_project_exists(self, project_overrides=None):
			from attract.setup import setup_for_attract

			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,
			svn_url=MOCK_SVN_URL)

			return proj_id, attract_project