Switch away from deprecated ssl API (#18472)

DataDog · Aug 30, 2024 · 9693964 · 9693964
1 parent f4062f5
commit 9693964
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spark/tests/test_spark.py b/spark/tests/test_spark.py
@@ -1306,9 +1306,11 @@ def do_GET(self):
 
 def run_ssl_server():
     cert_file = os.path.join(CERTIFICATE_DIR, 'server.pem')
+    context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
+    context.load_cert_chain(cert_file)
 
     httpd = BaseHTTPServer.HTTPServer((SSL_SERVER_ADDRESS, SSL_SERVER_PORT), StandaloneAppsResponseHandler)
-    httpd.socket = ssl.wrap_socket(httpd.socket, certfile=cert_file, server_side=False)
+    httpd.socket = context.wrap_socket(httpd.socket, server_side=False)
     httpd.timeout = 5
 
     threading.Thread(target=httpd.handle_request).start()