Sync from deal-sage/deal-sage@13a420d134

test/stub_server.py

@@ -0,0 +1,69 @@
+#!/usr/bin/env python3
+"""Stub of the DealSage endpoints the pipeline talks to, for local tests.
+
+Records every request as JSON lines in the file named by STUB_LOG, and answers
+like the real functions would:
+  POST /record-app-version  allocate -> version 7 + upload token
+                            promote  -> promoted per STUB_POLICY (auto|manual)
+                            fail     -> ok
+  POST /record-app-check    -> {"id": "..."}
+  PUT  /__upload/<app>/<v>/<path> -> 204
+"""
+import json, os, sys
+from http.server import BaseHTTPRequestHandler, HTTPServer
+
+LOG = os.environ.get("STUB_LOG", "/tmp/stub-requests.jsonl")
+POLICY = os.environ.get("STUB_POLICY", "auto")
+seq = {"n": 0}
+
+class H(BaseHTTPRequestHandler):
+    def log_message(self, *a):  # quiet
+        pass
+
+    def _record(self, body):
+        with open(LOG, "a") as f:
+            f.write(json.dumps({"method": self.command, "path": self.path,
+                                "auth": self.headers.get("Authorization"), "body": body}) + "\n")
+
+    def _json(self, code, obj):
+        data = json.dumps(obj).encode()
+        self.send_response(code); self.send_header("content-type", "application/json")
+        self.send_header("content-length", str(len(data))); self.end_headers(); self.wfile.write(data)
+
+    def do_POST(self):
+        n = int(self.headers.get("content-length") or 0)
+        raw = self.rfile.read(n).decode() if n else ""
+        try: body = json.loads(raw) if raw else {}
+        except json.JSONDecodeError: body = {"_raw": raw}
+        self._record(body)
+        if self.headers.get("Authorization") != "Bearer test-publish-token":
+            return self._json(401, {"error": "bad token"})
+        if self.path.endswith("/record-app-version"):
+            a = body.get("action")
+            if a == "allocate":
+                return self._json(200, {"version": 7, "r2Prefix": f"apps/{body.get('appId')}/7/", "uploadToken": "upl-token"})
+            if a == "promote":
+                if POLICY == "auto":
+                    return self._json(200, {"version": 7, "currentVersion": 7, "promoted": True, "channel": "live"})
+                return self._json(200, {"version": 7, "currentVersion": 6, "promoted": False, "channel": "branch/feature/x"})
+            if a == "fail":
+                return self._json(200, {"ok": True})
+            return self._json(400, {"error": "bad action"})
+        if self.path.endswith("/record-app-check"):
+            seq["n"] += 1
+            return self._json(200, {"id": f"chk-{seq['n']}"})
+        self._json(404, {"error": "no route"})
+
+    def do_PUT(self):
+        n = int(self.headers.get("content-length") or 0)
+        self.rfile.read(n)
+        self._record({"bytes": n})
+        if not self.path.startswith("/__upload/"):
+            return self._json(404, {"error": "no route"})
+        if self.headers.get("Authorization") != "Bearer upl-token":
+            return self._json(401, {"error": "bad upload token"})
+        self.send_response(204); self.end_headers()
+
+if __name__ == "__main__":
+    port = int(sys.argv[1]) if len(sys.argv) > 1 else 8765
+    HTTPServer(("127.0.0.1", port), H).serve_forever()