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)