Claude
7b02fcfd58
Wildcard options, fixed values and combines can carry a `negative`, and a
derivation action gains the `negative` ("Add to negative") mode. The
engine files negatives per variable in `__wp_negatives__`, carries them
across nodes, and the Assembler gets a `negative` output built from the
variables its template rendered, filled into a `negative_template` at
`$negatives`. Negatives resolve on their own seed stream and refs inside
them resolve quietly, so picks never shift. Injector rows can set a
negative, the Prompt Cleaner cleans an optional negative with its own
rule column, and the preview endpoint returns the negatives table.
Schema v8 stamps payloads that use a negative.
219 lines
7.7 KiB
Python
219 lines
7.7 KiB
Python
"""Tests for /wp/api/preview/resolve — RNG-faithful chain preview."""
|
|
|
|
|
|
async def test_preview_empty_chain_returns_empty(wp_client):
|
|
resp = await wp_client.post("/wp/api/preview/resolve", json={"chain": []})
|
|
assert resp.status == 200
|
|
body = await resp.json()
|
|
assert body == {"resolved": {}, "axes": {}, "negatives": {}}
|
|
|
|
|
|
async def test_preview_resolves_fixed_values_module(wp_client):
|
|
"""A fixed_values module's entries surface as resolved bindings."""
|
|
chain = [[
|
|
{
|
|
"id": "abcdef00",
|
|
"type": "fixed_values",
|
|
"enabled": True,
|
|
"meta": {"name": "stub"},
|
|
"entries": [],
|
|
"payload": {
|
|
"values": [
|
|
{"id": "v0", "name": "style", "value": "noir"},
|
|
{"id": "v1", "name": "subject", "value": "fox"},
|
|
],
|
|
},
|
|
},
|
|
]]
|
|
resp = await wp_client.post(
|
|
"/wp/api/preview/resolve",
|
|
json={"chain": chain, "seed": 42},
|
|
)
|
|
assert resp.status == 200
|
|
body = await resp.json()
|
|
assert body["resolved"]["style"] == "noir"
|
|
assert body["resolved"]["subject"] == "fox"
|
|
|
|
|
|
async def test_preview_seed_42_is_deterministic_across_calls(wp_client):
|
|
"""Same seed + same chain → identical resolved map every call."""
|
|
chain = [[
|
|
{
|
|
"id": "abcdef00",
|
|
"type": "wildcard",
|
|
"enabled": True,
|
|
"meta": {"name": "color"},
|
|
"entries": [],
|
|
"payload": {
|
|
"var_binding": "color",
|
|
"options": [
|
|
{"id": "o1", "value": "red", "weight": 1},
|
|
{"id": "o2", "value": "blue", "weight": 1},
|
|
{"id": "o3", "value": "green", "weight": 1},
|
|
],
|
|
},
|
|
},
|
|
]]
|
|
r1 = await wp_client.post("/wp/api/preview/resolve", json={"chain": chain, "seed": 42})
|
|
r2 = await wp_client.post("/wp/api/preview/resolve", json={"chain": chain, "seed": 42})
|
|
body1, body2 = await r1.json(), await r2.json()
|
|
assert body1["resolved"] == body2["resolved"]
|
|
assert body1["resolved"]["color"] in {"red", "blue", "green"}
|
|
|
|
|
|
async def test_preview_chains_upstream_to_downstream(wp_client):
|
|
"""Earlier chain step's bindings feed into later steps' ctx."""
|
|
chain = [
|
|
# Step 1: writes $base
|
|
[{
|
|
"id": "11111111",
|
|
"type": "fixed_values",
|
|
"enabled": True,
|
|
"meta": {"name": "s1"},
|
|
"entries": [],
|
|
"payload": {"values": [{"id": "v0", "name": "base", "value": "wide"}]},
|
|
}],
|
|
# Step 2: writes $combo from $base via combine module
|
|
[{
|
|
"id": "22222222",
|
|
"type": "combine",
|
|
"enabled": True,
|
|
"meta": {"name": "s2"},
|
|
"entries": [],
|
|
"payload": {"output_var": "combo", "template": "$base shot", "input_vars": ["base"]},
|
|
}],
|
|
]
|
|
resp = await wp_client.post("/wp/api/preview/resolve", json={"chain": chain, "seed": 42})
|
|
body = await resp.json()
|
|
assert body["resolved"]["base"] == "wide"
|
|
assert body["resolved"]["combo"] == "wide shot"
|
|
|
|
|
|
async def test_preview_invalid_body_400(wp_client):
|
|
resp = await wp_client.post("/wp/api/preview/resolve", data="not json")
|
|
assert resp.status == 400
|
|
|
|
|
|
async def test_preview_chain_must_be_list(wp_client):
|
|
resp = await wp_client.post("/wp/api/preview/resolve", json={"chain": "nope"})
|
|
assert resp.status == 400
|
|
|
|
|
|
async def test_preview_multi_pick_returns_listvar_shape(wp_client):
|
|
"""SP2a: a multi-pick wildcard binds a ListVar. The endpoint must NOT 500
|
|
on the dataclass — it serialises it as {"items": [...], "sep": ...} so the
|
|
TS preview can join (bare $name) or index ($name.K) it like the engine."""
|
|
chain = [[
|
|
{
|
|
"id": "abcdef00",
|
|
"type": "wildcard",
|
|
"enabled": True,
|
|
"meta": {"name": "color"},
|
|
"entries": [],
|
|
"instance": {"pick_min": 2, "pick_max": 2, "pick_separator": ", "},
|
|
"payload": {
|
|
"var_binding": "color",
|
|
"options": [
|
|
{"id": "o1", "value": "red", "weight": 1},
|
|
{"id": "o2", "value": "blue", "weight": 1},
|
|
{"id": "o3", "value": "green", "weight": 1},
|
|
],
|
|
},
|
|
},
|
|
]]
|
|
resp = await wp_client.post("/wp/api/preview/resolve", json={"chain": chain, "seed": 42})
|
|
assert resp.status == 200
|
|
body = await resp.json()
|
|
val = body["resolved"]["color"]
|
|
assert isinstance(val, dict)
|
|
assert val["sep"] == ", "
|
|
assert len(val["items"]) == 2
|
|
assert set(val["items"]).issubset({"red", "blue", "green"})
|
|
|
|
|
|
async def test_preview_internals_stripped(wp_client):
|
|
"""Engine ctx grows __wp_* internals; the response must not leak them."""
|
|
chain = [[
|
|
{
|
|
"id": "abcdef00",
|
|
"type": "fixed_values",
|
|
"enabled": True,
|
|
"meta": {"name": "stub"},
|
|
"entries": [],
|
|
"payload": {"values": [{"id": "v0", "name": "x", "value": "y"}]},
|
|
},
|
|
]]
|
|
resp = await wp_client.post("/wp/api/preview/resolve", json={"chain": chain, "seed": 42})
|
|
body = await resp.json()
|
|
assert all(not k.startswith("__") for k in body["resolved"])
|
|
|
|
|
|
async def test_preview_returns_the_rolled_accepts_axes(wp_client):
|
|
"""`$var.AXIS` in the assembler preview has to read the tag the preview
|
|
run actually rolled, so the axes table rides beside the values. An option
|
|
with no tag on the axis rolls nothing, which is what renders empty."""
|
|
chain = [[
|
|
{
|
|
"id": "abcdef01",
|
|
"type": "wildcard",
|
|
"enabled": True,
|
|
"meta": {"name": "outfit"},
|
|
"entries": [],
|
|
"payload": {
|
|
"var_binding": "outfit",
|
|
"sub_categories": ["sandals", "boots"],
|
|
"tag_groups": {"SHOES": ["sandals", "boots"]},
|
|
"tag_group_kinds": {"SHOES": "accepts"},
|
|
"options": [
|
|
{"id": "o1", "value": "jeans", "weight": 1,
|
|
"sub_categories": ["boots"]},
|
|
],
|
|
},
|
|
},
|
|
{
|
|
"id": "abcdef02",
|
|
"type": "wildcard",
|
|
"enabled": True,
|
|
"meta": {"name": "hat"},
|
|
"entries": [],
|
|
"payload": {
|
|
"var_binding": "hat",
|
|
"sub_categories": ["sandals"],
|
|
"tag_groups": {"SHOES": ["sandals"]},
|
|
"tag_group_kinds": {"SHOES": "accepts"},
|
|
"options": [{"id": "o1", "value": "cap", "weight": 1}],
|
|
},
|
|
},
|
|
]]
|
|
resp = await wp_client.post(
|
|
"/wp/api/preview/resolve", json={"chain": chain, "seed": 42},
|
|
)
|
|
body = await resp.json()
|
|
assert body["resolved"]["outfit"] == "jeans"
|
|
assert body["axes"]["outfit"] == {"SHOES": "boots"}
|
|
assert body["axes"]["hat"] == {}
|
|
|
|
|
|
async def test_preview_returns_each_variables_negatives(wp_client):
|
|
"""Send-to-negative: the Assembler builds its negative preview locally
|
|
from the variables its template uses, so the table rides along."""
|
|
chain = [[{
|
|
"id": "abcdef03",
|
|
"type": "wildcard",
|
|
"enabled": True,
|
|
"meta": {"name": "hair"},
|
|
"entries": [],
|
|
"payload": {
|
|
"var_binding": "hair",
|
|
"options": [{"id": "o1", "value": "red hair", "weight": 1,
|
|
"negative": "blonde"}],
|
|
},
|
|
}]]
|
|
resp = await wp_client.post(
|
|
"/wp/api/preview/resolve", json={"chain": chain, "seed": 42},
|
|
)
|
|
body = await resp.json()
|
|
assert body["negatives"]["hair"] == [
|
|
{"text": "blonde", "pick": None, "source": "hair"},
|
|
]
|