def test_setup_settings(self):
config = {'sftp.jail.base_path': '/jail',
'extract.sftp.route.base_dir': 'newRoute'}
setup_settings(config)
self.assertEqual(get_setting(Config.MAX_RETRIES), 1)
self.assertEqual(get_setting(Config.TIMEOUT), 20)
self.assertEqual(get_setting(Config.PICKUP_ROUTE_BASE_DIR), 'newRoute')
def setUp(self):
settings = {
Config.HOMEDIR: '~/.gpg',
Config.BINARYFILE: 'gpg',
Config.KEYSERVER: 'the_key_keeper'
}
setup_settings(settings)
def test_setup_settings(self):
config = {
'sftp.jail.base_path': '/jail',
'extract.sftp.route.base_dir': 'newRoute'
}
setup_settings(config)
self.assertEqual(get_setting(Config.MAX_RETRIES), 1)
self.assertEqual(get_setting(Config.TIMEOUT), 20)
self.assertEqual(get_setting(Config.PICKUP_ROUTE_BASE_DIR), 'newRoute')
def setUp(self):
settings = {Config.HOMEDIR: '~/.gpg',
Config.BINARYFILE: 'gpg',
Config.KEYSERVER: 'the_key_keeper',
Config.PICKUP_ROUTE_BASE_DIR: 'route',
'extract.celery.CELERY_ALWAYS_EAGER': 'True',
'extract.retries_allowed': '1',
'extract.retry_delay': '3'}
setup_celery(settings)
setup_settings(settings)
def setUp(self):
settings = {
Config.HOMEDIR: '~/.gpg',
Config.BINARYFILE: 'gpg',
Config.KEYSERVER: 'the_key_keeper',
Config.PICKUP_ROUTE_BASE_DIR: 'route',
'extract.celery.CELERY_ALWAYS_EAGER': 'True',
'extract.retries_allowed': '1',
'extract.retry_delay': '3'
}
setup_celery(settings)
setup_settings(settings)
def setup_celery(settings, prefix=PREFIX):
'''
Setup celery based on parameters defined in setting (ini file).
This calls by client application when dictionary of settings is given
:param settings: dict of configurations
:param prefix: prefix in configurations used for configuring celery
'''
setup(celery, settings, prefix)
setup_settings(settings)
initialize_hpz(settings)
run_cron_cleanup(settings)
def setUp(self):
here = os.path.abspath(os.path.dirname(__file__))
gpg_home = os.path.abspath(
os.path.join(here, "..", "..", "..", "..", "config", "gpg"))
settings = {
'extract.celery.BROKER_URL': 'memory',
'extract.gpg.keyserver': None,
'extract.gpg.homedir': gpg_home,
'extract.gpg.public_key.cat': '*****@*****.**',
'extract.celery.CELERY_ALWAYS_EAGER': 'True',
'extract.retries_allowed': '1',
'extract.retry_delay': '3',
'hpz.file_upload_base_url': 'http://somehost:82/files',
}
setup_celery(settings)
setup_settings(settings)
self._tenant = get_unittest_tenant_name()
self.__files = ['a.txt', 'b.txt', 'c.txt']
self.__tmp_dir = tempfile.mkdtemp('file_archiver_test')
self.__tmp_zip = os.path.join(tempfile.mkdtemp('zip'), 'test.zip')
for file in self.__files:
with open(os.path.join(self.__tmp_dir, file), "w") as f:
f.write('hello ' + file)
self.maxDiff = None
def test_defaults_in_settings(self):
config = {}
setup_settings(config)
self.assertEqual(get_setting(Config.PICKUP_ROUTE_BASE_DIR), 'route')
from edextract import run_cron_cleanup
from hpz_client.frs.config import initialize as initialize_hpz
PREFIX = 'extract.celery'
def setup_celery(settings, prefix=PREFIX):
'''
Setup celery based on parameters defined in setting (ini file).
This calls by client application when dictionary of settings is given
:param settings: dict of configurations
:param prefix: prefix in configurations used for configuring celery
'''
setup(celery, settings, prefix)
setup_settings(settings)
initialize_hpz(settings)
run_cron_cleanup(settings)
# Create an instance of celery, check if it's for prod celeryd mode and configure it for prod mode if so
celery, conf = configure_celeryd(PREFIX, prefix=PREFIX)
prod_config = get_config_file()
if prod_config:
# We should only need to setup db connection in prod mode
setup_db_connection(conf)
setup_settings(conf)
initialize_hpz(conf)
run_cron_cleanup(conf)
def setUp(self):
settings = {Config.HOMEDIR: '~/.gpg',
Config.BINARYFILE: 'gpg',
Config.KEYSERVER: 'the_key_keeper'}
setup_settings(settings)
def test_defaults_in_settings(self):
config = {}
setup_settings(config)
self.assertEqual(get_setting(Config.PICKUP_ROUTE_BASE_DIR), 'route')