test(unit): cover steps, workflows, runtime errors, and health

Author: joaodaher

tests/unit/test_arg_expr.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+from fastapi_cloudflow.core.arg import Arg, ArgExpr
+
+
+def test_arg_env_and_str() -> None:
+    e = Arg.env("FOO")
+    assert isinstance(e, ArgExpr)
+    assert str(e) == '${sys.get_env("FOO")}'  # wrapped as ${...}
+
+
+def test_arg_concat_and_div() -> None:
+    base = Arg.env("BASE")
+    p = base / "steps" / Arg.env("NAME")
+    s = base + "/hello"
+    # Validate expression strings
+    assert isinstance(p, ArgExpr)
+    assert "/steps" in p.expr and 'sys.get_env("NAME")' in p.expr
+    assert isinstance(s, ArgExpr)
+    assert "/hello" in s.expr
+
+
+def test_arg_param_and_ctx() -> None:
+    assert str(Arg.param("foo")) == "${params.foo}"
+    assert str(Arg.ctx("run_id")) == "${ctx.run_id}"

tests/unit/test_echo_name_steps.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import app
+
+
+def test_extract_name_and_name_shout() -> None:
+    c = TestClient(app)
+
+    # extract-name expects the httpbin response shape: {"json": {...}}
+    r1 = c.post(
+        "/steps/extract-name",
+        headers={"X-Workflow-Name": "unit"},
+        json={"json": {"name": "Ada Lovelace"}},
+    )
+    assert r1.status_code == 200
+    extracted = r1.json()
+    assert extracted["echoed_name"] == "Ada Lovelace"
+
+    # name-shout expects EchoOut
+    r2 = c.post(
+        "/steps/name-shout",
+        headers={"X-Workflow-Name": "unit", "X-Workflow-Run-Id": "test"},
+        json=extracted,
+    )
+    assert r2.status_code == 200
+    shouted = r2.json()
+    assert shouted["name_upper"] == "ADA LOVELACE"
+    assert shouted["length"] == len("ADA LOVELACE")

tests/unit/test_health.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import app
+
+
+def test_health() -> None:
+    c = TestClient(app)
+    r = c.get("/health")
+    assert r.status_code == 200
+    assert r.json()["status"] == "ok"

tests/unit/test_joke_flow_steps.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import app
+
+
+def test_joke_split_and_rate() -> None:
+    c = TestClient(app)
+
+    # Manually simulate the payload of joke-fetch output
+    fake = {
+        "id": "abc",
+        "joke": "Why did the chicken cross the road? To get to the other side!",
+        "status": 200,
+    }
+
+    r1 = c.post(
+        "/steps/joke-split",
+        headers={"X-Workflow-Name": "unit"},
+        json=fake,
+    )
+    assert r1.status_code == 200
+    bits = r1.json()
+    assert "setup" in bits and "punch" in bits
+
+    r2 = c.post(
+        "/steps/joke-rate",
+        headers={"X-Workflow-Name": "unit", "X-Workflow-Run-Id": "test"},
+        json=bits,
+    )
+    assert r2.status_code == 200
+    rated = r2.json()
+    assert isinstance(rated.get("rating"), int)

tests/unit/test_order_flow.py

@@ -0,0 +1,31 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import app
+
+
+def test_price_order() -> None:
+    c = TestClient(app)
+    r = c.post(
+        "/steps/price-order",
+        headers={"X-Workflow-Name": "unit"},
+        json={"account_id": 1, "sku": "abc", "qty": 1},
+    )
+    assert r.status_code == 200
+    body = r.json()
+    assert body["order_id"].startswith("o-")
+    assert "price" in body
+
+
+def test_auth_payment() -> None:
+    c = TestClient(app)
+    draft = {"order_id": "o-test", "price": "12.34"}
+    r = c.post(
+        "/steps/auth-payment",
+        headers={"X-Workflow-Name": "unit", "X-Workflow-Run-Id": "test"},
+        json=draft,
+    )
+    assert r.status_code == 200
+    body = r.json()
+    assert body["order_id"].startswith("o-")
+    assert body["status"] == "approved"

tests/unit/test_post_story_steps.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import app
+
+
+def test_build_story_and_summarize() -> None:
+    c = TestClient(app)
+
+    story = {"topic": "AI", "mood": "playful", "author_id": 7}
+    r1 = c.post(
+        "/steps/build-story",
+        headers={"X-Workflow-Name": "unit"},
+        json=story,
+    )
+    assert r1.status_code == 200
+    post = r1.json()
+    assert post["title"]
+    assert post["userId"] == 7
+
+    # summarize-post expects PostOut shape
+    r2 = c.post(
+        "/steps/summarize-post",
+        headers={"X-Workflow-Name": "unit", "X-Workflow-Run-Id": "test"},
+        json=post,
+    )
+    assert r2.status_code == 200
+    summary = r2.json()
+    assert "slug" in summary and "short_title" in summary

tests/unit/test_runtime_errors.py

@@ -0,0 +1,36 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import app
+
+
+def test_malformed_json_returns_422() -> None:
+    c = TestClient(app)
+    r = c.post(
+        "/steps/price-order",
+        headers={"X-Workflow-Name": "unit", "Content-Type": "application/json"},
+        data="{not-json}",
+    )
+    assert r.status_code == 422
+    assert "Malformed JSON" in r.text
+
+
+def test_missing_body_returns_422() -> None:
+    c = TestClient(app)
+    r = c.post("/steps/price-order", headers={"X-Workflow-Name": "unit"})
+    assert r.status_code == 422
+    assert ("Request body required" in r.text) or ("Malformed JSON body" in r.text)
+
+
+def test_payload_wrapper_is_accepted() -> None:
+    c = TestClient(app)
+    wrapped = {"payload": {"account_id": 1, "sku": "abc", "qty": 1}}
+    r = c.post("/steps/price-order", headers={"X-Workflow-Name": "unit"}, json=wrapped)
+    assert r.status_code == 200
+
+
+def test_validation_error_returns_422() -> None:
+    c = TestClient(app)
+    # Missing required fields
+    r = c.post("/steps/price-order", headers={"X-Workflow-Name": "unit"}, json={})
+    assert r.status_code == 422

tests/unit/test_user_and_payments.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+from fastapi.testclient import TestClient
+from main import app
+
+
+def test_hash_password() -> None:
+    c = TestClient(app)
+    r = c.post(
+        "/steps/hash-password",
+        headers={"X-Workflow-Name": "unit"},
+        json={"email": "user@example.com", "password": "hunter2asdf"},
+    )
+    assert r.status_code == 200
+    draft = r.json()
+    assert draft["hashed_password"].startswith("hashed:")
+
+
+def test_validate_cart_and_summarize() -> None:
+    c = TestClient(app)
+    r1 = c.post(
+        "/steps/validate-cart",
+        headers={"X-Workflow-Name": "unit"},
+        json={"total": 10.0, "currency": "USD"},
+    )
+    assert r1.status_code == 200
+    r2 = c.post(
+        "/steps/summarize-charge",
+        headers={"X-Workflow-Name": "unit", "X-Workflow-Run-Id": "test"},
+        json={"status": "approved", "psp_id": "p-1"},
+    )
+    assert r2.status_code == 200
+    res = r2.json()
+    assert res["ok"] is True
+    assert res["txn_id"] == "p-1"