Update unavailable choice test

Ladysnake · Oct 30, 2023 · 10b2364 · 10b2364
1 parent 580eaf0
commit 10b2364
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/testmod/java/org/ladysnake/babblings/tests/BlabberTestSuite.java b/src/testmod/java/org/ladysnake/babblings/tests/BlabberTestSuite.java
@@ -76,10 +76,11 @@ public void unavailableChoicesCannotGetSelected(TestContext ctx) {
         player.setHealth(10f);
         Blabber.startDialogue(player, new Identifier("babblings:mountain_king"));
         ((DialogueScreenHandler) player.currentScreenHandler).makeChoice(player, 1);
+        GameTestUtil.assertTrue("dialogue should be at state bargain", player.currentScreenHandler instanceof DialogueScreenHandler handler && handler.getCurrentStateKey().equals("bargain"));
         ((DialogueScreenHandler) player.currentScreenHandler).makeChoice(player, 0);
         GameTestUtil.assertTrue("unavailable choice 0 should be ignored", player.currentScreenHandler instanceof DialogueScreenHandler handler && handler.getCurrentStateKey().equals("bargain"));
         ((DialogueScreenHandler) player.currentScreenHandler).makeChoice(player, 2);
-        GameTestUtil.assertTrue("dialogue should end", player.currentScreenHandler == player.playerScreenHandler);
+        GameTestUtil.assertTrue("dialogue should be at state friendship", player.currentScreenHandler instanceof DialogueScreenHandler handler && handler.getCurrentStateKey().equals("friendship"));
         ctx.complete();
     }
 }