def test_plugin_config(self):
class TPlugin(object):
name = 'test'
author = 'test1'
bucket = ''
def init_config(self, config):
self.bucket = config.get("bucket", "bad")
master_config = Config()
master_config['PLUGIN_CONFIGS'] = {
"test-test1": {
"bucket": "test_bucket"
}
}
pm = PluginManager(master_config, plugins=[TPlugin()])
p = pm.get_plugin('test', False)
assert p.bucket == ''
p = pm.get_plugin('test', True)
assert p.bucket == 'test_bucket'
def test_write_config(cleandir):
config = Config()
prod_config = ProdConfig()
config.from_object(prod_config)
cm = ConfigManager()
cm._write_config_py(config)
assert 'client_config.py' in os.listdir(os.getcwd())
def test_uploader(self, cleandir, shared_datadir, upload_plugin_type):
uploaders = [
TestUploaderPass(),
TestUploaderFail(),
TestUploaderRetryPass(),
TestUploaderRetryFail(),
]
# Don't mind the name, should only contain 1 plugin
upload_plugins = [u for u in uploaders if u.name == upload_plugin_type]
if sys.version_info[1] == 5:
data_dir = str(shared_datadir)
else:
data_dir = shared_datadir
if not os.path.exists(data_dir):
os.makedirs(data_dir)
with ChDir(data_dir):
uploader = Uploader(config=Config(), plugins=upload_plugins)
upload_plugin = upload_plugins[0]
uploader.set_uploader(upload_plugin_type)
assert uploader.uploader.name == upload_plugin_type
if "success" in upload_plugin.name:
assert uploader.upload(["/file1"]) is True
elif "fail" in upload_plugin.name:
assert uploader.upload(["/file2"]) is False
def test_init(self):
data_dir = os.getcwd()
t_config = TConfig()
t_config.DATA_DIR = data_dir
config = Config()
config.from_object(t_config)
p = PackageHandler(config)
assert p.files_dir == os.path.join(data_dir, user_data_dir, "files")
assert p.deploy_dir == os.path.join(data_dir, user_data_dir, "deploy")
def test_process_packages(self):
data_dir = os.getcwd()
t_config = TConfig()
t_config.DATA_DIR = data_dir
t_config.UPDATE_PATCHES = False
config = Config()
config.from_object(t_config)
p = PackageHandler(config)
p.process_packages()
def init_app(self, obj, refresh=False, test=False):
"""Sets up client with config values from obj
Args:
obj (instance): config object
Kwargs:
refresh (bool) Meaning:
True: Refresh update manifest on object initialization
False: Don't refresh update manifest on object initialization
"""
# Used to add missing required information
# i.e. APP_NAME
config = Config()
config.from_object(obj)
self.FROZEN = jms_utils.app.FROZEN
# Grabbing config information
update_urls = config.get('UPDATE_URLS')
# Here we combine all urls & add trailing / if one isn't present
self.update_urls = self._sanatize_update_url(update_urls)
self.app_name = config.get('APP_NAME', 'PyUpdater')
self.company_name = config.get('COMPANY_NAME', 'Digital Sapphire')
if test:
# Making platform deterministic for tests.
# No need to test for other platforms at the moment
self.data_dir = obj.DATA_DIR
self.platform = 'mac'
else: # pragma: no cover
# Getting platform specific application directory
self.data_dir = appdirs.user_data_dir(self.app_name,
self.company_name,
roaming=True)
# Setting the platform to pass when requesting updates
self.platform = jms_utils.system.get_system()
# Creating update folder. Using settings to ease change in future
self.update_folder = os.path.join(self.data_dir,
settings.UPDATE_FOLDER)
# Attempting to sanitize incorrect inputs types
self.root_key = config.get('PUBLIC_KEY', '')
self.app_key = None
# Config option to disable tls cert verification
self.verify = config.get('VERIFY_SERVER_CERT', True)
self.version_file = settings.VERSION_FILE
self.key_file = settings.KEY_FILE
self._setup()
if refresh is True:
self.refresh()
def test_plugin_manager_load(self, n, a):
class Plugin(object):
name = n
author = a
pm = PluginManager(Config(), plugins=[Plugin()])
if isinstance(n, six.string_types) and isinstance(a, six.string_types):
assert len(pm.plugins) == 1
else:
assert len(pm.plugins) == 0
def init(): # pragma: no cover
if not os.path.exists(os.path.join(settings.CONFIG_DATA_FOLDER,
settings.CONFIG_FILE_USER)):
config = Config()
config = initial_setup(config)
log.info('Creating pyu-data dir...')
pyu = PyUpdater(config)
pyu.setup()
loader = Loader()
loader.save_config(config)
log.info('Setup complete')
else:
sys.exit('Not an empty PyUpdater repository')
def test_plugin_unique_names(self):
class Plugin1(object):
name = "test"
author = "test1"
class Plugin2(object):
name = "test"
author = "test2"
pm = PluginManager(Config(), plugins=[Plugin1(), Plugin2()])
plugin_names = [n["name"] for n in pm.plugins]
assert "test" in plugin_names
assert "test2" in plugin_names
def test_plugin_unique_names(self):
class Plugin1(object):
name = 'test'
author = 'test1'
class Plugin2(object):
name = 'test'
author = 'test2'
pm = PluginManager(Config(), plugins=[Plugin1(), Plugin2()])
plugin_names = [n['name'] for n in pm.plugins]
assert 'test' in plugin_names
assert 'test2' in plugin_names
def _cmd_init(*args): # pragma: no cover
if not os.path.exists(
os.path.join(settings.CONFIG_DATA_FOLDER,
settings.CONFIG_FILE_USER)):
# Load a basic config.
config = Config()
# Run config through all of the setup functions
config = initial_setup(config)
log.info('Creating pyu-data dir...')
# Initialize PyUpdater with newly created config
pyu = PyUpdater(config)
# Setup repository
pyu.setup()
# Load config manager & save config to disk
cm = ConfigManager()
cm.save_config(config)
log.info('Setup complete')
else:
log.error('Not an empty PyUpdater repository')
def test_plugin_config(self):
class TPlugin(object):
name = "test"
author = "test1"
bucket = ""
def init_config(self, config):
self.bucket = config.get("bucket", "bad")
master_config = Config()
master_config["PLUGIN_CONFIGS"] = {
"test-test1": {
"bucket": "test_bucket"
}
}
pm = PluginManager(master_config, plugins=[TPlugin()])
p = pm.get_plugin("test", False)
assert p.bucket == ""
p = pm.get_plugin("test", True)
assert p.bucket == "test_bucket"
def test_default_plugin_detection_scp(self):
pm = PluginManager(Config())
plugin_names = [n["name"] for n in pm.plugins]
assert "scp" in plugin_names
def __init__(self, config=None):
self.config = Config()
# Important to keep this before updating config
if config is not None:
self.update_config(config)
def test_prod_bad_atter():
config = Config()
prod_config = ProdConfig()
config.from_object(prod_config)
assert config.get('DEBUG', None) is not None
def test_prod_config():
config = Config()
prod_config = ProdConfig()
config.from_object(prod_config)
assert config['MORE_INFO'] == 'Yes Please'
def test_dev_config_bad_attr():
config = Config()
test_config = DevConfig()
config.from_object(test_config)
assert config.get('BAD_ATTR', None) is None
def test_dev_config():
config = Config()
test_config = DevConfig()
config.from_object(test_config)
assert config['TESTING'] is True
def test_prod_config():
config = Config()
prod_config = ProdConfig()
config.from_object(prod_config)
assert config["MORE_INFO"] == "Yes Please"
def test_default_plugin_detection_scp(self):
pm = PluginManager(Config())
plugin_names = [n['name'] for n in pm.plugins]
assert 'scp' in plugin_names
