Fixed flaky CI tests by replacing httpbin with a simple http_server (#395)
Signed-off-by: saimedhi <[email protected]>
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
#
|
||||
# The OpenSearch Contributors require contributions made to
|
||||
# this file be licensed under the Apache-2.0 license or a
|
||||
# compatible open source license.
|
||||
#
|
||||
# Modifications Copyright OpenSearch Contributors. See
|
||||
# GitHub history for details.
|
||||
|
||||
import json
|
||||
import threading
|
||||
from http.server import BaseHTTPRequestHandler, HTTPServer
|
||||
|
||||
|
||||
class TestHTTPRequestHandler(BaseHTTPRequestHandler):
|
||||
def do_GET(self):
|
||||
self.send_response(200)
|
||||
headers = self.headers
|
||||
self.send_header("Content-type", "application/json")
|
||||
self.end_headers()
|
||||
|
||||
Headers = {}
|
||||
for header, value in headers.items():
|
||||
capitalized_header = "-".join([word.title() for word in header.split("-")])
|
||||
Headers.update({capitalized_header: value})
|
||||
if "Connection" in Headers:
|
||||
Headers.pop("Connection")
|
||||
|
||||
data = {"method": "GET", "headers": Headers}
|
||||
self.wfile.write(json.dumps(data).encode("utf-8"))
|
||||
|
||||
|
||||
class TestHTTPServer(HTTPServer):
|
||||
def __init__(self, host="localhost", port=8080):
|
||||
super().__init__((host, port), TestHTTPRequestHandler)
|
||||
self._server_thread = None
|
||||
|
||||
def start(self):
|
||||
if self._server_thread is not None:
|
||||
return
|
||||
|
||||
self._server_thread = threading.Thread(target=self.serve_forever)
|
||||
self._server_thread.start()
|
||||
|
||||
def stop(self):
|
||||
if self._server_thread is None:
|
||||
return
|
||||
self.socket.close()
|
||||
self.shutdown()
|
||||
self._server_thread.join()
|
||||
self._server_thread = None
|
||||
Reference in New Issue
Block a user