diff --git a/src/test/java/world/bentobox/aoneblock/listeners/CheckPhaseTest.java b/src/test/java/world/bentobox/aoneblock/listeners/CheckPhaseTest.java index 71720df..6b2ae3f 100644 --- a/src/test/java/world/bentobox/aoneblock/listeners/CheckPhaseTest.java +++ b/src/test/java/world/bentobox/aoneblock/listeners/CheckPhaseTest.java @@ -20,8 +20,10 @@ import org.junit.jupiter.api.AfterEach; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; +import org.mockito.ArgumentCaptor; import org.mockito.Mock; +import net.kyori.adventure.text.Component; import net.kyori.adventure.title.Title; import world.bentobox.aoneblock.AOneBlock; @@ -152,8 +154,10 @@ void testSetNewPhase() { verify(previous).getFirstTimeEndCommands(); // Verify phase name change assertEquals("Next Phase", is.getPhaseName()); - // Verify title shown - verify(mockPlayer).showTitle(any(Title.class)); + // Verify title shown with correct phase name + ArgumentCaptor titleCaptor = ArgumentCaptor.forClass(Title.class); + verify(mockPlayer).showTitle(titleCaptor.capture()); + assertEquals(Component.text("Next Phase"), titleCaptor.getValue().title()); } @@ -185,8 +189,10 @@ void testSetNewPhaseSecondTime() { verify(previous, never()).getFirstTimeEndCommands(); // Verify phase name change assertEquals("Next Phase", is.getPhaseName()); - // Verify title shown - verify(mockPlayer).showTitle(any(Title.class)); + // Verify title shown with correct phase name + ArgumentCaptor<Title> titleCaptor = ArgumentCaptor.forClass(Title.class); + verify(mockPlayer).showTitle(titleCaptor.capture()); + assertEquals(Component.text("Next Phase"), titleCaptor.getValue().title()); } @@ -218,8 +224,10 @@ void testSetNewPhaseNullPlayer() { verify(previous).getFirstTimeEndCommands(); // Verify phase name change assertEquals("Next Phase", is.getPhaseName()); - // Verify title shown - verify(mockPlayer).showTitle(any(Title.class)); + // Verify title shown with correct phase name + ArgumentCaptor<Title> titleCaptor = ArgumentCaptor.forClass(Title.class); + verify(mockPlayer).showTitle(titleCaptor.capture()); + assertEquals(Component.text("Next Phase"), titleCaptor.getValue().title()); } @@ -251,8 +259,10 @@ void testCheckPhaseNPCPlayer() { verify(previous).getFirstTimeEndCommands(); // Verify phase name change assertEquals("Next Phase", is.getPhaseName()); - // Verify title shown - verify(mockPlayer).showTitle(any(Title.class)); + // Verify title shown with correct phase name + ArgumentCaptor<Title> titleCaptor = ArgumentCaptor.forClass(Title.class); + verify(mockPlayer).showTitle(titleCaptor.capture()); + assertEquals(Component.text("Next Phase"), titleCaptor.getValue().title()); }