sftp: add /sftp-test endpoint

worker.py

@@ -3,6 +3,7 @@ import time
 import threading
 import requests
 from http.server import BaseHTTPRequestHandler, HTTPServer
+from sftp_store import sftp_test
 
 import sftp_store
 
@@ -17,23 +18,27 @@ if not API_BASE or not WORKER_TOKEN:
 
 class Handler(BaseHTTPRequestHandler):
     def do_GET(self):
-        # HF sometimes calls "/" or "/?logs=container" while rendering logs.
-        if self.path.startswith("/health") or self.path == "/" or self.path.startswith("/?"):
+        path = self.path.split("?", 1)[0]
+
+        if path in ("/", "/health"):
             self.send_response(200)
             self.end_headers()
-            self.wfile.write(b"ok\n")
+            self.wfile.write(b"ok
+")
             return
 
-        if self.path.startswith("/sftp-test"):
+        if path == "/sftp-test":
             try:
-                msg = sftp_store.sftp_test()
+                msg = sftp_test()
                 self.send_response(200)
                 self.end_headers()
-                self.wfile.write((msg + "\n").encode("utf-8"))
+                self.wfile.write((msg + "
+").encode("utf-8"))
             except Exception as e:
                 self.send_response(500)
                 self.end_headers()
-                self.wfile.write((f"sftp fail: {e}\n").encode("utf-8"))
+                self.wfile.write((f"sftp fail: {e}
+").encode("utf-8"))
             return
 
         self.send_response(404)