Add tests for PotionEffectArgument.NamespacedKey

DerEchtePilz · DerEchtePilz · commit d80dab03694c · 2024-04-24T17:24:52.000+02:00
diff --git a/commandapi-platforms/commandapi-bukkit/commandapi-bukkit-test/commandapi-bukkit-test-tests/src/test/java/dev/jorel/commandapi/test/arguments/ArgumentPotionTests.java b/commandapi-platforms/commandapi-bukkit/commandapi-bukkit-test/commandapi-bukkit-test-tests/src/test/java/dev/jorel/commandapi/test/arguments/ArgumentPotionTests.java
@@ -5,6 +5,8 @@
 
 import java.util.Arrays;
 
+import org.bukkit.NamespacedKey;
+import org.bukkit.Registry;
 import org.bukkit.potion.PotionEffectType;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
@@ -96,6 +98,34 @@ void executionTestWithPotionEffectArgumentAllPotionEffects() {
 		assertNoMoreResults(results);
 	}
 
+	@Test
+	void executionTestWithPotionEffectArgumentNamespaced() {
+		Mut<NamespacedKey> results = Mut.of();
+
+		new CommandAPICommand("test")
+			.withArguments(new PotionEffectArgument.NamespacedKey("potion"))
+			.executesPlayer((player, args) -> {
+				results.set((NamespacedKey) args.get("potion"));
+			})
+			.register();
+
+		PlayerMock player = server.addPlayer();
+
+		server.dispatchCommand(player, "test speed");
+		assertEquals(NamespacedKey.minecraft("speed"), results.get());
+
+		server.dispatchCommand(player, "test minecraft:speed");
+		assertEquals(NamespacedKey.minecraft("speed"), results.get());
+
+		server.dispatchCommand(player, "test unknowneffect");
+		assertEquals(NamespacedKey.minecraft("unknowneffect"), results.get());
+
+		server.dispatchCommand(player, "test custom:unknowneffect");
+		assertEquals(NamespacedKey.fromString("custom:unknowneffect"), results.get());
+
+		assertNoMoreResults(results);
+	}
+
 	/********************
 	 * Suggestion tests *
 	 ********************/
