JSONRPCRequestMcpValidationTest.java

/*
 * Copyright 2024-2024 the original author or authors.
 */

package io.modelcontextprotocol.spec;

import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.*;

/**
 * Tests for MCP-specific validation of JSONRPCRequest ID requirements.
 *
 * @author Christian Tzolov
 */
public class JSONRPCRequestMcpValidationTest {

	@Test
	public void testValidStringId() {
		assertDoesNotThrow(() -> {
			var request = new McpSchema.JSONRPCRequest("2.0", "test/method", "string-id", null);
			assertEquals("string-id", request.id());
		});
	}

	@Test
	public void testValidIntegerId() {
		assertDoesNotThrow(() -> {
			var request = new McpSchema.JSONRPCRequest("2.0", "test/method", 123, null);
			assertEquals(123, request.id());
		});
	}

	@Test
	public void testValidLongId() {
		assertDoesNotThrow(() -> {
			var request = new McpSchema.JSONRPCRequest("2.0", "test/method", 123L, null);
			assertEquals(123L, request.id());
		});
	}

	@Test
	public void testNullIdThrowsException() {
		IllegalArgumentException exception = assertThrows(IllegalArgumentException.class, () -> {
			new McpSchema.JSONRPCRequest("2.0", "test/method", null, null);
		});

		assertTrue(exception.getMessage().contains("MCP requests MUST include an ID"));
		assertTrue(exception.getMessage().contains("null IDs are not allowed"));
	}

	@Test
	public void testDoubleIdTypeThrowsException() {
		IllegalArgumentException exception = assertThrows(IllegalArgumentException.class, () -> {
			new McpSchema.JSONRPCRequest("2.0", "test/method", 123.45, null);
		});

		assertTrue(exception.getMessage().contains("MCP requests MUST have an ID that is either a string or integer"));
		assertTrue(exception.getMessage().contains("Double"));
	}

	@Test
	public void testBooleanIdThrowsException() {
		IllegalArgumentException exception = assertThrows(IllegalArgumentException.class, () -> {
			new McpSchema.JSONRPCRequest("2.0", "test/method", true, null);
		});

		assertTrue(exception.getMessage().contains("MCP requests MUST have an ID that is either a string or integer"));
		assertTrue(exception.getMessage().contains("Boolean"));
	}

	@Test
	public void testArrayIdThrowsException() {
		IllegalArgumentException exception = assertThrows(IllegalArgumentException.class, () -> {
			new McpSchema.JSONRPCRequest("2.0", "test/method", new String[] { "array" }, null);
		});

		assertTrue(exception.getMessage().contains("MCP requests MUST have an ID that is either a string or integer"));
	}

	@Test
	public void testObjectIdThrowsException() {
		IllegalArgumentException exception = assertThrows(IllegalArgumentException.class, () -> {
			new McpSchema.JSONRPCRequest("2.0", "test/method", new Object(), null);
		});

		assertTrue(exception.getMessage().contains("MCP requests MUST have an ID that is either a string or integer"));
		assertTrue(exception.getMessage().contains("Object"));
	}

}