def create_new_agent_config(old_agent): new_agent = CloudifyAgentConfig() _set_default_new_agent_config_values(old_agent, new_agent) _copy_values_from_old_agent_config(old_agent, new_agent) new_agent.set_default_values() new_agent.set_installation_params(runner=None) return new_agent
def _create_agent(): old_agent = CloudifyAgentConfig({ 'install_method': 'remote', 'ip': '10.0.4.47', 'rest_host': '10.0.4.46', 'distro': 'ubuntu', 'distro_codename': 'trusty', 'basedir': '/home/vagrant', 'user': '******', 'key': '~/.ssh/id_rsa', 'windows': False, 'package_url': 'http://10.0.4.46:53229/packages/agents/' 'ubuntu-trusty-agent.tar.gz', 'version': '4.4', 'broker_config': { 'broker_ip': '10.0.4.46', 'broker_pass': '******', 'broker_user': '******', 'broker_ssl_cert': '' } }) old_agent.set_execution_params() old_agent.set_default_values() old_agent.set_installation_params(runner=None) return old_agent
def _create_agent(): old_agent = CloudifyAgentConfig({ 'install_method': 'remote', 'ip': '10.0.4.47', 'rest_host': '10.0.4.46', 'distro': 'ubuntu', 'distro_codename': 'trusty', 'basedir': '/home/vagrant', 'user': '******', 'key': '~/.ssh/id_rsa', 'windows': False, 'package_url': 'http://10.0.4.46:53229/packages/agents/' 'ubuntu-trusty-agent.tar.gz', 'version': '4.4', 'broker_config': { 'broker_ip': '10.0.4.46', 'broker_pass': '******', 'broker_user': '******', 'broker_ssl_cert': '' } }) old_agent.set_execution_params() old_agent.set_default_values() old_agent.set_installation_params(runner=None) return old_agent
def create_new_agent_config(old_agent): new_agent = CloudifyAgentConfig() _set_default_new_agent_config_values(old_agent, new_agent) _copy_values_from_old_agent_config(old_agent, new_agent) new_agent.set_default_values() new_agent.set_installation_params(runner=None) return new_agent
def test_prepare_secured(self): cloudify_agent = CloudifyAgentConfig({ 'local': True, 'rest_port': '443' }) cloudify_agent.set_execution_params() cloudify_agent.set_default_values() cloudify_agent.set_installation_params(None) user = getpass.getuser() basedir = utils.get_home_dir(user) agent_dir = os.path.join(basedir, 'test_deployment') envdir = os.path.join(agent_dir, 'env') workdir = os.path.join(agent_dir, 'work') # This test needs to be adapted to security settings expected = { 'agent_dir': agent_dir, 'process_management': { 'name': 'init.d' if os.name == 'posix' else 'nssm' }, 'basedir': basedir, 'name': 'test_deployment', 'rest_port': '443', 'rest_host': 'localhost', 'queue': 'test_deployment', 'envdir': envdir, 'user': user, 'local': True, 'install_method': 'local', 'disable_requiretty': True, 'env': {}, 'fabric_env': {}, 'max_workers': 5, 'min_workers': 0, 'workdir': workdir, 'broker_ssl_cert_path': os.environ[constants.BROKER_SSL_CERT_PATH], 'windows': os.name == 'nt', 'system_python': 'python', 'bypass_maintenance': False, 'rest_token': 'test_token', 'rest_tenant': {} } if os.name == 'posix': distro = platform.dist()[0].lower() distro_codename = platform.dist()[2].lower() expected['distro'] = platform.dist()[0].lower() expected['distro_codename'] = platform.dist()[2].lower() expected['package_url'] = 'localhost/packages/agents/' \ '{0}-{1}-agent.tar.gz' \ .format(distro, distro_codename) else: expected['package_url'] = 'localhost/packages/agents/' \ 'cloudify-windows-agent.exe' self.maxDiff = None self.assertDictEqual(expected, cloudify_agent)
def _test_prepare(self, agent_config, expected_values, context=None): user = getpass.getuser() basedir = utils.get_home_dir(user) agent_dir = os.path.join(basedir, 'test_deployment') envdir = os.path.join(agent_dir, 'env') workdir = os.path.join(agent_dir, 'work') # This test needs to be adapted to security settings expected = { 'agent_dir': agent_dir, 'process_management': { 'name': 'init.d' if os.name == 'posix' else 'nssm' }, 'basedir': basedir, 'name': 'test_deployment', 'rest_host': 'localhost', 'broker_ip': ['localhost'], 'broker_ssl_cert': '', 'heartbeat': None, 'queue': 'test_deployment', 'envdir': envdir, 'user': user, 'local': True, 'install_method': 'local', 'disable_requiretty': True, 'env': {}, 'fabric_env': {}, 'max_workers': 5, 'min_workers': 0, 'workdir': workdir, 'broker_ssl_cert_path': os.environ[constants.BROKER_SSL_CERT_PATH], 'windows': os.name == 'nt', 'system_python': 'python', 'bypass_maintenance': False, 'network': 'default', 'version': utils.get_agent_version(), 'node_instance_id': 'test_node', 'log_level': 'info', 'log_max_bytes': 5242880, 'log_max_history': 7 } expected.update(expected_values) self.maxDiff = None context = context or {} ctx = mock_context(**context) with patch('cloudify_agent.installer.config.agent_config.ctx', ctx), \ patch('cloudify.utils.ctx', mock_context()): cloudify_agent = CloudifyAgentConfig() cloudify_agent.set_initial_values(True, agent_config=agent_config) cloudify_agent.set_execution_params() cloudify_agent.set_default_values() cloudify_agent.set_installation_params(None) self.assertDictEqual(expected, cloudify_agent)
def _test_prepare(self, agent_config, expected_values): cloudify_agent = CloudifyAgentConfig() cloudify_agent.set_initial_values(True, agent_config=agent_config) cloudify_agent.set_execution_params() cloudify_agent.set_default_values() cloudify_agent.set_installation_params(None) user = getpass.getuser() basedir = utils.get_home_dir(user) agent_dir = os.path.join(basedir, 'test_deployment') envdir = os.path.join(agent_dir, 'env') workdir = os.path.join(agent_dir, 'work') # This test needs to be adapted to security settings expected = { 'agent_dir': agent_dir, 'process_management': { 'name': 'init.d' if os.name == 'posix' else 'nssm' }, 'basedir': basedir, 'name': 'test_deployment', 'rest_host': 'localhost', 'broker_ip': 'localhost', 'heartbeat': None, 'queue': 'test_deployment', 'envdir': envdir, 'user': user, 'local': True, 'install_method': 'local', 'disable_requiretty': True, 'env': {}, 'fabric_env': {}, 'max_workers': 20, 'min_workers': 0, 'workdir': workdir, 'broker_ssl_cert_path': os.environ[constants.BROKER_SSL_CERT_PATH], 'windows': os.name == 'nt', 'system_python': 'python', 'bypass_maintenance': False, 'network': 'default', 'version': utils.get_agent_version() } expected.update(expected_values) self.maxDiff = None self.assertDictEqual(expected, cloudify_agent)
def _test_prepare(self, agent_config, expected_values, context=None): user = getpass.getuser() basedir = utils.get_home_dir(user) agent_dir = os.path.join(basedir, 'test_deployment') envdir = os.path.join(agent_dir, 'env') workdir = os.path.join(agent_dir, 'work') # This test needs to be adapted to security settings expected = { 'agent_dir': agent_dir, 'process_management': { 'name': 'init.d' if os.name == 'posix' else 'nssm' }, 'basedir': basedir, 'name': 'test_deployment', 'rest_host': ['127.0.0.1'], 'broker_ip': ['127.0.0.1'], 'broker_ssl_cert': agent_ssl_cert.DUMMY_CERT, 'heartbeat': None, 'queue': 'test_deployment', 'envdir': envdir, 'user': user, 'local': True, 'install_method': 'local', 'disable_requiretty': True, 'env': {}, 'fabric_env': {}, 'file_server_url': 'http://127.0.0.1:80/resources', 'max_workers': 5, 'min_workers': 0, 'workdir': workdir, 'broker_ssl_cert_path': os.environ[constants.BROKER_SSL_CERT_PATH], 'windows': os.name == 'nt', 'system_python': 'python', 'bypass_maintenance': False, 'network': 'default', 'version': utils.get_agent_version(), 'node_instance_id': 'test_node', 'log_level': 'info', 'log_max_bytes': 5242880, 'log_max_history': 7, 'rest_ssl_cert': agent_ssl_cert.DUMMY_CERT, 'tenant': { 'name': 'default_tenant', 'rabbitmq_username': '******', 'rabbitmq_password': '******', 'rabbitmq_vhost': '/' } } expected.update(expected_values) self.maxDiff = None context = context or {} ctx = mock_context(**context) patches = [ patch('cloudify_agent.installer.config.agent_config.ctx', ctx), patch('cloudify.utils.ctx', mock_context()), patch('cloudify.utils.get_manager_name', return_value='cloudify') ] # it's originally a string because it comes from envvars if expected['rest_port'] != '443': patches.append( patch('cloudify.utils.get_manager_file_server_scheme', return_value='http') ) for p in patches: p.start() self.addCleanup(p.stop) cloudify_agent = CloudifyAgentConfig() cloudify_agent.set_initial_values( True, agent_config=agent_config) cloudify_agent.set_execution_params() cloudify_agent.set_default_values() cloudify_agent.set_installation_params(None) self.assertDictEqual(expected, cloudify_agent)
def _test_prepare(agent_ssl_cert, agent_config, expected_values, context=None): user = getpass.getuser() is_windows = os.name == 'nt' basedir = utils.get_agent_basedir(is_windows) agent_dir = os.path.join(basedir, 'test_deployment') envdir = os.path.join(agent_dir, 'env') workdir = os.path.join(agent_dir, 'work') # This test needs to be adapted to security settings expected = { 'agent_dir': agent_dir, 'process_management': { 'name': 'init.d' if os.name == 'posix' else 'nssm' }, 'basedir': basedir, 'name': 'test_deployment', 'rest_host': ['127.0.0.1'], 'broker_ip': ['127.0.0.1'], 'broker_ssl_cert': agent_ssl_cert.DUMMY_CERT, 'heartbeat': None, 'queue': 'test_deployment', 'envdir': envdir, 'user': user, 'local': True, 'install_method': 'local', 'disable_requiretty': True, 'env': {}, 'fabric_env': {}, 'file_server_url': 'https://127.0.0.1:80/resources', 'max_workers': 5, 'min_workers': 0, 'workdir': workdir, 'broker_ssl_cert_path': os.environ[constants.BROKER_SSL_CERT_PATH], 'windows': is_windows, 'system_python': 'python', 'bypass_maintenance': False, 'network': 'default', 'version': utils.get_agent_version(), 'node_instance_id': 'test_node', 'log_level': 'info', 'log_max_bytes': 5242880, 'log_max_history': 7, 'rest_ssl_cert': agent_ssl_cert.DUMMY_CERT, 'tenant': { 'name': 'default_tenant', 'rabbitmq_username': '******', 'rabbitmq_password': '******', 'rabbitmq_vhost': '/' } } expected.update(expected_values) # port is originally a string because it comes from envvars with agent_config_patches(agent_ssl_cert, expected['rest_port'] == '443', context): cloudify_agent = CloudifyAgentConfig() cloudify_agent.set_initial_values( True, agent_config=agent_config) cloudify_agent.set_execution_params() cloudify_agent.set_default_values() cloudify_agent.set_installation_params(None) assert expected == cloudify_agent