Offer Semuta before CHOAM Jubba storm protection

tests/choam-jubba.test.ts

@@ -5,6 +5,7 @@ import {
   createGame,
   newPlayer,
   viewGame,
+  normalizeAutomaticGame,
   type Game,
   type Action,
 } from '../game/engine';
@@ -12,6 +13,8 @@ import { baseDeck } from '../game/cards';
 import { MOBILE_LOCATION } from '../game/board';
 import { botActions } from '../game/bots';
 import { DIFFICULTIES } from '../game/bot-profiles';
+import { richeseCards } from '../game/richese-cards';
+import { SEMUTA_DRUG_ID } from '../game/semuta-drug';
 
 const player = (g: Game, id: string) => g.players.find((p) => p.id === id)!;
 const send = (g: Game, id: string, action: Action) =>
@@ -121,6 +124,82 @@ void test('Jubba protects only CHOAM in one territory in basic and advanced game
     conserve(g);
   }
 });
+void test('accepted Jubba discard offers Semuta before protection and resumes one storm', () => {
+  const initial = fixture();
+  initial.players[1] = newPlayer('e', 'Richese', 'richese');
+  forces(initial, 'e', { 'red_chasm:7': 3 });
+  initial.richeseCache = richeseCards();
+  initial.semutaPreview = true;
+  const index = initial.richeseCache.findIndex(card => card.id === SEMUTA_DRUG_ID);
+  assert.ok(index >= 0);
+  const semuta = initial.richeseCache.splice(index, 1)[0];
+  player(initial, 'e').hand.push(semuta);
+  contest(initial, 'b');
+  const card = player(initial, 'c').hand.find(card => card.name === 'Jubba Cloak')!.id;
+  const physical = (state: Game) => [
+    ...state.deck, ...state.discard, ...state.richeseCache!,
+    ...state.players.flatMap(p => p.hand),
+  ].map(card => card.id).sort();
+  const original = physical(initial);
+  const absent = structuredClone(initial);
+  player(absent, 'e').hand = [absent.richeseCache!.splice(0, 1)[0]];
+  absent.richeseCache!.push(semuta);
+  const pending = allow(play(begin(initial)));
+  const neutral = allow(play(begin(absent)));
+  const event = pending.pendingTreacheryDiscard!.batch.event;
+  assert.equal(pending.pendingTreacheryDiscard?.continuation.kind, 'choamJubbaDiscard');
+  assert.equal(pending.pendingTreacheryDiscard?.reaction?.stage, 'offer');
+  assert.equal(neutral.pendingTreacheryDiscard?.reaction?.stage, 'offer');
+  assert.equal(pending.storm, 5);
+  assert.ok(pending.stormResolution);
+  assert.deepEqual(pending.stormResolution.choamProtected, undefined);
+  assert.equal(player(pending, 'c').forces['red_chasm:7'], 4);
+  assert.equal(player(pending, 'e').forces['red_chasm:7'], 3);
+  assert.equal(pending.discard.filter(used => used.id === card).length, 1);
+  assert.deepEqual(viewGame(pending, 'c'), viewGame(neutral, 'c'));
+  assert.deepEqual(viewGame(pending, 'b'), viewGame(neutral, 'b'));
+  assert.equal(viewGame(pending, 'e').semutaReaction?.canCommit, true);
+  assert.equal(botActions(viewGame(pending, 'e'))[0]?.type, 'semutaCommit');
+  assert.deepEqual(normalizeAutomaticGame(JSON.parse(JSON.stringify(pending))), pending);
+  for (const change of [
+    (state: Game) => {
+      const c = state.pendingTreacheryDiscard!.continuation;
+      if (c.kind === 'choamJubbaDiscard') c.territory = 'basin';
+    },
+    (state: Game) => { state.stormResolution!.distance++; },
+    (state: Game) => { player(state, 'c').forces = {}; },
+    (state: Game) => { state.pendingTreacheryDiscard!.batch.entries[0].card.name = 'Forgery'; },
+  ]) {
+    const corrupt: Game = JSON.parse(JSON.stringify(pending));
+    change(corrupt);
+    const before = JSON.stringify(corrupt);
+    assert.throws(() => send(corrupt, 'e', { type: 'semutaCommit', event }));
+    assert.equal(JSON.stringify(corrupt), before);
+  }
+  let declined: Game = JSON.parse(JSON.stringify(pending));
+  for (const id of ['c', 'b', 'e'])
+    declined = send(declined, id, { type: 'semutaPass', event });
+  const claimed = send(JSON.parse(JSON.stringify(pending)), 'e',
+    { type: 'semutaCommit', event });
+  for (const state of [declined, claimed]) {
+    assert.equal(state.pendingTreacheryDiscard, null);
+    assert.equal(state.stormResolution, null);
+    assert.equal(state.storm, 8);
+    assert.equal(player(state, 'c').forces['red_chasm:7'], 4);
+    assert.equal(player(state, 'c').tanks, 0);
+    assert.equal(player(state, 'e').forces['red_chasm:7'], undefined);
+    assert.equal(player(state, 'e').tanks, 3);
+    assert.equal(state.spice['red_chasm:7'], undefined);
+    assert.equal(state.decision?.kind, 'choamMarket');
+    assert.equal(state.log.filter(entry => entry.text.startsWith('Storm moved')).length, 1);
+    assert.deepEqual(physical(state), original);
+    assert.throws(() => send(state, 'e', { type: 'semutaCommit', event }));
+  }
+  assert.equal(declined.discard.filter(used => used.id === card).length, 1);
+  assert.equal(player(claimed, 'e').hand.filter(used => used.id === card).length, 1);
+  assert.equal(claimed.discard.filter(used => used.id === semuta.id).length, 1);
+});
+
 void test('moving-storm decision and public projection do not reveal Jubba ownership', () => {
   const initial = fixture(),
     empty = structuredClone(initial);

tests/semuta-recovery.test.ts

@@ -1120,6 +1120,93 @@ void test('authenticated ordinary Trip Semuta claim returns one force after save
   } finally { f.sqlite.close(); }
 });
 
+void test('authenticated Jubba Semuta claim protects CHOAM before one saved storm', async () => {
+  const f = unitStore();
+  try {
+    const made = await f.rooms.createRoom('Semuta Jubba', 'choam', false, ['choam']);
+    const code = made.view.code;
+    const richese = await f.rooms.joinRoom(code, 'Richese', 'richese');
+    const bg = await f.rooms.joinRoom(code, 'BG', 'beneGesserit');
+    const auths = await Promise.all([made.token!, richese.token!, bg.token!]
+      .map(token => f.restart().authenticate(code, token)));
+    const existing = await f.restart().readRoom(code);
+    const g = createGame(code, newPlayer(auths[0].playerId, 'CHOAM', 'choam'), false, ['choam']);
+    g.players.push(newPlayer(auths[1].playerId, 'Richese', 'richese'),
+      newPlayer(auths[2].playerId, 'BG', 'beneGesserit'));
+    Object.assign(g, { status: 'playing', phase: 0, turn: 2, storm: 5,
+      stormPending: 3, active: auths[0].playerId,
+      order: auths.map(auth => auth.playerId),
+      deck: baseDeck(), richeseCache: richeseCards(),
+      spice: { 'red_chasm:7': 8 }, semutaPreview: true });
+    for (const p of g.players) {
+      p.spice = 10;
+      p.hand = [];
+      p.forces = {};
+      p.reserves = 20;
+      p.traitors = p.leaders.length ? [p.leaders[0].id] : [];
+      p.traitorChoices = [];
+    }
+    const semuta = take(g, SEMUTA_DRUG_ID), karama = take(g, 'karama');
+    const index = g.deck.findIndex(card => card.name === 'Jubba Cloak');
+    assert.ok(index >= 0);
+    const used = g.deck.splice(index, 1)[0];
+    g.players[0].hand.push(used);
+    g.players[0].forces = { 'red_chasm:7': 4 };
+    g.players[0].reserves = 16;
+    g.players[1].hand.push(semuta);
+    g.players[1].forces = { 'red_chasm:7': 3 };
+    g.players[1].reserves = 17;
+    g.players[2].hand.push(karama);
+    g.version = existing.version + 1;
+    assert.equal(f.sqlite.prepare('UPDATE rooms SET state=?,version=? WHERE code=? AND version=?')
+      .run(JSON.stringify(g), g.version, code, existing.version).changes, 1);
+    const seats = f.sqlite.prepare('SELECT * FROM seats').all();
+    const act = async (index: number, action: Action) => {
+      const current = await f.restart().readRoom(code);
+      await f.restart().act(code, auths[index], current.version, action, clock);
+      return f.restart().readRoom(code);
+    };
+    let current = await f.restart().readRoom(code);
+    for (let i = 0; i < auths.length; i++)
+      current = await act(i, { type: 'ready' });
+    assert.equal(current.decision?.kind, 'choamStorm');
+    current = await act(0, { type: 'card', mode: 'choam',
+      card: used.id, territory: 'red_chasm' });
+    assert.equal(current.response?.kind, 'choamWorthless');
+    for (let i = 0; current.response?.kind === 'choamWorthless' && i < 3; i++) {
+      const index = auths.findIndex(auth =>
+        !viewGame(current, auth.playerId).responseControls?.hasPassed &&
+        !!viewGame(current, auth.playerId).responseControls?.cancelCards.length);
+      assert.ok(index >= 0);
+      current = await act(index, { type: 'passResponse' });
+    }
+    const event = current.pendingTreacheryDiscard!.batch.event;
+    assert.equal(current.pendingTreacheryDiscard?.continuation.kind, 'choamJubbaDiscard');
+    assert.equal(current.pendingTreacheryDiscard?.reaction?.stage, 'offer');
+    assert.equal(current.storm, 5);
+    assert.equal(current.stormResolution?.choamProtected, undefined);
+    assert.equal(current.players[0].forces['red_chasm:7'], 4);
+    assert.equal(current.players[1].forces['red_chasm:7'], 3);
+    await f.restart().continueRoomAutomatic(code, clock);
+    assert.equal((await f.restart().readRoom(code)).version, current.version);
+    assert.equal((await f.restart().readSeatView(code, auths[1])).semutaReaction?.canCommit, true);
+    const claimed = await act(1, { type: 'semutaCommit', event });
+    assert.equal(claimed.pendingTreacheryDiscard, null);
+    assert.equal(claimed.stormResolution, null);
+    assert.equal(claimed.storm, 8);
+    assert.equal(claimed.players[0].forces['red_chasm:7'], 4);
+    assert.equal(claimed.players[1].forces['red_chasm:7'], undefined);
+    assert.equal(claimed.players[1].tanks, 3);
+    assert.equal(claimed.players[1].hand.filter(card => card.id === used.id).length, 1);
+    assert.equal(claimed.discard.filter(card => card.id === semuta.id).length, 1);
+    await assert.rejects(f.restart().act(code, auths[1], current.version,
+      { type: 'semutaCommit', event }, clock));
+    assert.deepEqual(f.sqlite.prepare('SELECT * FROM seats').all(), seats);
+    for (const auth of auths)
+      assert.deepEqual(await f.restart().readSeatView(code, auth), viewGame(claimed, auth.playerId));
+  } finally { f.sqlite.close(); }
+});
+
 void test('authenticated CHOAM sale retains paid spice and market after a saved Semuta claim', async () => {
   const f = unitStore();
   try {