def setup_flask_app():
    global security_config
    if not security_config:
        conf_file_str = read_manager_file('/opt/manager/rest-security.conf')
        security_config = yaml.load(conf_file_str)

    manager_ip = utils.get_manager_ip()
    return _setup_flask_app(manager_ip=manager_ip,
                            hash_salt=security_config['hash_salt'],
                            secret_key=security_config['secret_key'])
示例#2
0
 def test_provider_context(self):
     # Context is already setup during test bootstrap phase,
     # we only verify everything was properly saved and accessible
     name = PROVIDER_NAME
     context = copy.deepcopy(PROVIDER_CONTEXT)
     context['cloudify']['cloudify_agent']['networks']['default'] = \
         get_manager_ip()
     response_context = self.client.manager.get_context()
     self.assertEqual(name, response_context['name'])
     self.assertEqual(context, response_context['context'])
     self.assertRaises(CloudifyClientError,
                       self.client.manager.create_context, name, context)
 def test_provider_context(self):
     # Context is already setup during test bootstrap phase,
     # we only verify everything was properly saved and accessible
     name = PROVIDER_NAME
     context = copy.deepcopy(PROVIDER_CONTEXT)
     context['cloudify']['cloudify_agent']['networks']['default'] = \
         get_manager_ip()
     response_context = self.client.manager.get_context()
     self.assertEqual(name, response_context['name'])
     self.assertEqual(context, response_context['context'])
     self.assertRaises(CloudifyClientError,
                       self.client.manager.create_context, name, context)
def prepare_reset_storage_script():
    reset_script = get_resource('scripts/reset_storage.py')
    copy_file_to_manager(reset_script, SCRIPT_PATH)
    with tempfile.NamedTemporaryFile(delete=False) as f:
        json.dump(
            {
                'config': {
                    '': constants.CONFIG_FILE_LOCATION,
                    'security': SECURITY_FILE_LOCATION,
                    'authorization': constants.AUTHORIZATION_FILE_LOCATION
                },
                'ip': utils.get_manager_ip(),
                'username': utils.get_manager_username(),
                'password': utils.get_manager_password(),
                'provider_context': PROVIDER_CONTEXT
            }, f)
    try:
        copy_file_to_manager(f.name, CONFIG_PATH)
    finally:
        os.unlink(f.name)
示例#5
0
def run_query(query, db_name=None):
    conf = get_postgres_conf()
    manager_ip = utils.get_manager_ip()

    db_name = db_name or conf.db_name
    with psycopg2.connect(database=db_name,
                          user=conf.username,
                          password=conf.password,
                          host=manager_ip) as con:
        con.autocommit = True
        logger.info('Trying to execute SQL query: ' + query)
        with closing(con.cursor()) as cur:
            try:
                cur.execute(query)
                fetchall = cur.fetchall()
                status_message = 'ok'
            except Exception, e:
                fetchall = None
                status_message = str(e)
            return {'status': status_message, 'all': fetchall}