Sybren A. Stüvel
cf41599e20
The tests were logging in incorrectly, which came to light due to Pillar commit 4b5a961e1422d8e976b2bf8bb9a4f91addf9bbec.
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
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):
|
|
from attract import subversion
|
|
|
|
subversion.task_logged._clear_state()
|
|
self.unload_modules('attract')
|
|
|
|
AbstractPillarTest.tearDown(self)
|
|
|
|
@property
|
|
def attract(self):
|
|
return self.app.pillar_extensions['attract']
|
|
|
|
def ensure_project_exists(self, project_overrides=None):
|
|
from attract.setup import setup_for_attract
|
|
import pillar.tests.common_test_data as ctd
|
|
|
|
project_overrides = dict(
|
|
picture_header=None,
|
|
picture_square=None,
|
|
**(project_overrides or {})
|
|
)
|
|
proj_id, project = AbstractPillarTest.ensure_project_exists(self, project_overrides)
|
|
ctd.EXAMPLE_USER['groups'].append(ctd.EXAMPLE_ADMIN_GROUP_ID)
|
|
|
|
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
|