mirror of
https://github.com/galaxyproject/galaxy.git
synced 2026-09-19 02:21:32 +08:00
82 lines
3.1 KiB
Python
82 lines
3.1 KiB
Python
"""Integration tests for configurable job handler assignment methods."""
|
|
|
|
import os
|
|
|
|
from base import integration_util
|
|
|
|
SCRIPT_DIRECTORY = os.path.abspath(os.path.dirname(__file__))
|
|
HANDLER_TEMPLATE_JOB_CONFIG_FILE = os.path.join(SCRIPT_DIRECTORY, "handler_template_job_conf.xml")
|
|
|
|
|
|
class WritesConfig(object):
|
|
|
|
def _with_handlers_config(self, assign_with=None, default=None, handlers=None):
|
|
handlers = handlers or []
|
|
template = {
|
|
'assign_with': ' assign_with="%s"' % assign_with if assign_with is not None else '',
|
|
'default': ' default="%s"' % default if default is not None else '',
|
|
'handlers': '\n'.join([
|
|
'<handler id="{id}"{tags}/>'.format(
|
|
id=x['id'],
|
|
tags=' tags="%s"' % x['tags'] if 'tags' in x else ''
|
|
) for x in handlers]),
|
|
}
|
|
self.__write_config_from(HANDLER_TEMPLATE_JOB_CONFIG_FILE, template=template)
|
|
|
|
def __write_config_from(self, path, template=None):
|
|
template = template or {}
|
|
self.__write_config(open(path, "r").read().format(**template))
|
|
|
|
def __write_config(self, contents):
|
|
with open(self._app.config.job_config_file, "w") as f:
|
|
f.write(contents)
|
|
|
|
|
|
class BaseHandlerAssignmentMethodIntegrationTestCase(integration_util.IntegrationTestCase, WritesConfig):
|
|
|
|
framework_tool_and_types = True
|
|
|
|
@classmethod
|
|
def handle_galaxy_config_kwds(cls, config):
|
|
cls.jobs_directory = cls._test_driver.mkdtemp()
|
|
config["jobs_directory"] = cls.jobs_directory
|
|
config["job_config_file"] = os.path.join(cls.jobs_directory, "job_conf.xml")
|
|
# Disable tool dependency resolution.
|
|
config["tool_dependency_dir"] = "none"
|
|
config["enable_beta_mulled_containers"] = "true"
|
|
|
|
|
|
class DBPreassignHandlerAssignmentMethodIntegrationTestCase(BaseHandlerAssignmentMethodIntegrationTestCase):
|
|
|
|
def setUp(self):
|
|
self._with_handlers_config(assign_with='db-preassign', handlers=[{'id': 'main'}])
|
|
super(DBPreassignHandlerAssignmentMethodIntegrationTestCase, self).setUp()
|
|
|
|
def test_handler_assignment(self):
|
|
tool_id = 'config_vars'
|
|
self._run_tool_test(tool_id)
|
|
|
|
|
|
class DBTransactionIsolationHandlerAssignmentMethodIntegrationTestCase(BaseHandlerAssignmentMethodIntegrationTestCase):
|
|
|
|
def setUp(self):
|
|
self._with_handlers_config(assign_with='db-transaction-isolation', handlers=[{'id': 'main'}])
|
|
super(DBTransactionIsolationHandlerAssignmentMethodIntegrationTestCase, self).setUp()
|
|
|
|
def test_handler_assignment(self):
|
|
self._skip_unless_postgres()
|
|
tool_id = 'config_vars'
|
|
self._run_tool_test(tool_id)
|
|
|
|
|
|
class DBSkipLockedHandlerAssignmentMethodIntegrationTestCase(BaseHandlerAssignmentMethodIntegrationTestCase):
|
|
|
|
def setUp(self):
|
|
self._with_handlers_config(assign_with='db-skip-locked', handlers=[{'id': 'main'}])
|
|
super(DBSkipLockedHandlerAssignmentMethodIntegrationTestCase, self).setUp()
|
|
|
|
def test_handler_assignment(self):
|
|
self._skip_unless_postgres()
|
|
tool_id = 'config_vars'
|
|
self._run_tool_test(tool_id)
|