adding deploy

Author: joroscoSF

.github/workflows/sf_cli_integration.yml

@@ -166,6 +166,21 @@ jobs:
           exit 1
         }
 
+      # ── Script: deploy ───────────────────────────────────────────────────────
+
+    - name: '[script] deploy — sf data-code-extension script deploy'
+      run: |
+        sf data-code-extension script deploy \
+          --name test-script-deploy \
+          --package-version 0.0.1 \
+          --description "Test script deploy" \
+          --package-dir testScript/payload \
+          --cpu-size CPU_2XL \
+          -o dev1 || {
+          echo "::error::sf data-code-extension script deploy FAILED. Check mock server output above for which endpoint failed. The deploy command flags or API contract may have changed."
+          exit 1
+        }
+
       # ── Function: init ────────────────────────────────────────────────────────
 
     - name: '[function] init — sf data-code-extension function init --package-dir testFunction'
@@ -244,3 +259,18 @@ jobs:
           echo "::error::sf data-code-extension function run FAILED. Check mock server output above; the --entrypoint flag or SF CLI org auth contract may have changed."
           exit 1
         }
+
+      # ── Function: deploy ─────────────────────────────────────────────────────
+
+    - name: '[function] deploy — sf data-code-extension function deploy'
+      run: |
+        sf data-code-extension function deploy \
+          --name test-function-deploy \
+          --package-version 0.0.1 \
+          --description "Test function deploy" \
+          --package-dir testFunction/payload \
+          --cpu-size CPU_2XL \
+          -o dev1 || {
+          echo "::error::sf data-code-extension function deploy FAILED. Check mock server output above for which endpoint failed. The deploy command flags or API contract may have changed."
+          exit 1
+        }

scripts/mock_sf_server.py

@@ -1,7 +1,8 @@
 """Minimal mock Salesforce server for CI integration tests.
 
-Intercepts the HTTP calls made during ``sf data-code-extension script|function run``
-so that neither a real Salesforce org nor real Data Cloud data is required.
+Intercepts the HTTP calls made during ``sf data-code-extension script|function``
+``run`` and ``deploy`` so that neither a real Salesforce org nor real Data Cloud
+data is required.
 
 Endpoints handled
 -----------------
@@ -17,6 +18,20 @@
     Returns fake rows with the columns expected by the default script template
     (Account_std__dll: description__c, sfdcorganizationid__c, kq_id__c).
 
+POST /services/data/v63.0/ssot/data-custom-code
+    Called by deploy_full() → create_deployment().
+    Returns a fake fileUploadUrl pointing back at this server.
+
+GET  /services/data/v63.0/ssot/data-custom-code/*
+    Called by deploy_full() → wait_for_deployment() → get_deployments().
+    Returns deploymentStatus=Deployed immediately so the poll loop exits.
+
+PUT  /upload/*
+    The presigned fileUploadUrl target.  Accepts the deployment.zip binary.
+
+POST /services/data/v63.0/ssot/data-transforms
+    Called by deploy_full() → create_data_transform() for script packages.
+
 GET  /* (catch-all)
     Returns {"status": "ok"} for any other SF API path the CLI may probe.
 
@@ -72,6 +87,9 @@
     ],
 }
 
+_DATA_CUSTOM_CODE_PATH = "/services/data/v63.0/ssot/data-custom-code"
+_DATA_TRANSFORMS_PATH = "/services/data/v63.0/ssot/data-transforms"
+
 
 class MockSFHandler(BaseHTTPRequestHandler):
     def _send_json(self, payload: object, status: int = 200) -> None:
@@ -82,29 +100,54 @@ def _send_json(self, payload: object, status: int = 200) -> None:
         self.end_headers()
         self.wfile.write(body)
 
+    def _send_empty(self, status: int = 200) -> None:
+        self.send_response(status)
+        self.send_header("Content-Length", "0")
+        self.end_headers()
+
+    def _drain_body(self) -> bytes:
+        length = int(self.headers.get("Content-Length", 0))
+        return self.rfile.read(length) if length else b""
+
     def log_message(self, fmt: str, *args: object) -> None:  # type: ignore[override]
         print(f"[MOCK SF] {self.command} {self.path} — {fmt % args}", flush=True)
 
     def do_GET(self) -> None:
         path = self.path.split("?")[0]
         if path == "/services/oauth2/userinfo":
             self._send_json(_USERINFO)
+        elif path.startswith(_DATA_CUSTOM_CODE_PATH + "/"):
+            # Deployment status poll — report Deployed immediately
+            self._send_json({"deploymentStatus": "Deployed"})
         else:
             self._send_json({"status": "ok"})
 
     def do_POST(self) -> None:
         path = self.path.split("?")[0]
-        length = int(self.headers.get("Content-Length", 0))
-        body = self.rfile.read(length) if length else b""
-        print(f"[MOCK SF] POST body: {body[:200]!r}", flush=True)
+        body = self._drain_body()
+        print(f"[MOCK SF] POST body: {body[:300]!r}", flush=True)
 
         if path == "/services/oauth2/token":
             self._send_json(_TOKEN_RESPONSE)
         elif path.endswith("/ssot/query-sql"):
+            # Data Cloud query (run command)
             self._send_json(_QUERY_RESPONSE)
+        elif path == _DATA_CUSTOM_CODE_PATH:
+            # create_deployment() — return a presigned upload URL
+            self._send_json(
+                {"fileUploadUrl": f"http://localhost:{PORT}/upload/fake-deployment.zip"}
+            )
+        elif path == _DATA_TRANSFORMS_PATH:
+            # create_data_transform() — script packages only
+            self._send_json({"id": "fake-dt-id", "name": "fake-data-transform"})
         else:
             self._send_json({"status": "ok"})
 
+    def do_PUT(self) -> None:
+        # upload_zip() sends the deployment.zip to the presigned URL
+        self._drain_body()
+        self._send_empty(200)
+
 
 if __name__ == "__main__":
     server = HTTPServer(("localhost", PORT), MockSFHandler)