def start_proxy(port, backend_port, update_listener, quiet=False, backend_host=DEFAULT_BACKEND_HOST): proxy_thread = GenericProxy(port=port, forward_host='%s:%s' % (backend_host, backend_port), update_listener=update_listener, quiet=quiet) proxy_thread.start() TMP_THREADS.append(proxy_thread)
def serve(port, quiet=True): if quiet: log = logging.getLogger('werkzeug') log.setLevel(logging.ERROR) ssl_context = GenericProxy.get_flask_ssl_context() app.run(port=int(port), threaded=True, host='0.0.0.0', ssl_context=ssl_context)
def test_api_gateway_http_integration(): test_port = 12123 backend_url = 'http://localhost:%s%s' % (test_port, API_PATH_HTTP_BACKEND) # create target HTTP backend def listener(**kwargs): response = Response() response.status_code = 200 response._content = json.dumps( kwargs['data']) if kwargs['data'] else '{}' return response proxy = GenericProxy(test_port, update_listener=listener) proxy.start() # create API Gateway and connect it to the HTTP backend result = connect_api_gateway_to_http('test_gateway2', backend_url, path=API_PATH_HTTP_BACKEND) # make test request to gateway url = INBOUND_GATEWAY_URL_PATTERN.format(api_id=result['id'], stage_name=TEST_STAGE_NAME, path=API_PATH_HTTP_BACKEND) result = requests.get(url) assert result.status_code == 200 assert to_str(result.content) == '{}' data = {"data": 123} result = requests.post(url, data=json.dumps(data)) assert result.status_code == 200 assert json.loads(to_str(result.content)) == data # clean up proxy.stop()
def start_proxy(port, backend_port, update_listener): proxy_thread = GenericProxy(port=port, forward_host='127.0.0.1:%s' % backend_port, update_listener=update_listener) proxy_thread.start() TMP_THREADS.append(proxy_thread)
t.start() TMP_THREADS.append(t) return t else: return run(cmd) def start_dynalite(port=DEFAULT_PORT_DYNAMODB, async=False, update_listener=None): backend_port = DEFAULT_PORT_DYNAMODB_BACKEND cmd = '%s/node_modules/dynalite/cli.js --port %s' % (root_path, backend_port) print("Starting mock DynamoDB (port %s)..." % port) proxy_thread = GenericProxy(port=port, forward_host='127.0.0.1:%s' % backend_port, update_listener=update_listener) proxy_thread.start() TMP_THREADS.append(proxy_thread) return do_run(cmd, async) def start_kinesalite(port=DEFAULT_PORT_KINESIS, async=False, shard_limit=100, update_listener=None): backend_port = DEFAULT_PORT_KINESIS_BACKEND cmd = ('%s/node_modules/kinesalite/cli.js --shardLimit %s --port %s' % (root_path, shard_limit, backend_port)) print("Starting mock Kinesis (port %s)..." % port) proxy_thread = GenericProxy(port=port,
def start_proxy(port, backend_port, update_listener): proxy_thread = GenericProxy(port=port, forward_host='127.0.0.1:%s' % backend_port, update_listener=update_listener) proxy_thread.start() TMP_THREADS.append(proxy_thread)
install.install_elasticsearch() backend_port = DEFAULT_PORT_ELASTICSEARCH_BACKEND es_data_dir = '%s/infra/elasticsearch/data' % (ROOT_PATH) if DATA_DIR: es_data_dir = '%s/elasticsearch' % DATA_DIR # Elasticsearch 5.x cannot be bound to 0.0.0.0 in some Docker environments, # hence we use the default bind address 127.0.0.0 and put a proxy in front of it cmd = (( 'ES_JAVA_OPTS=\"$ES_JAVA_OPTS -Xms200m -Xmx500m\" %s/infra/elasticsearch/bin/elasticsearch ' + '-E http.port=%s -E http.publish_port=%s -E http.compression=false -E path.data=%s ' + '-E xpack.security.enabled=false') % (ROOT_PATH, backend_port, backend_port, es_data_dir)) if USE_SSL: # make sure we have a test cert generated and configured GenericProxy.create_ssl_cert() cmd += ((' -E xpack.ssl.key=%s.key -E xpack.ssl.certificate=%s.crt ' + '-E xpack.security.transport.ssl.enabled=true ' + '-E xpack.security.http.ssl.enabled=true') % (SERVER_CERT_PEM_FILE, SERVER_CERT_PEM_FILE)) print("Starting local Elasticsearch (%s port %s)..." % (get_service_protocol(), port)) if delete_data: run('rm -rf %s' % es_data_dir) # fix permissions run('chmod -R 777 %s/infra/elasticsearch' % ROOT_PATH) run('mkdir -p "%s"; chmod -R 777 "%s"' % (es_data_dir, es_data_dir)) # start proxy and ES process start_proxy(port, backend_port, update_listener,