DataObjectCoverageTest.java

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *--------------------------------------------------------------------------------------------*/

package com.github.copilot;

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

import org.junit.jupiter.api.Test;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.github.copilot.rpc.CustomAgentConfig;
import com.github.copilot.rpc.GetForegroundSessionResponse;
import com.github.copilot.rpc.McpHttpServerConfig;
import com.github.copilot.rpc.McpStdioServerConfig;
import com.github.copilot.rpc.ModelCapabilitiesOverride;
import com.github.copilot.rpc.PermissionRequest;
import com.github.copilot.rpc.PermissionRequestResult;
import com.github.copilot.rpc.PostToolUseHookInput;
import com.github.copilot.rpc.PostToolUseHookOutput;
import com.github.copilot.rpc.PreToolUseHookInput;
import com.github.copilot.rpc.PreToolUseHookOutput;
import com.github.copilot.rpc.SectionOverride;
import com.github.copilot.rpc.SetForegroundSessionRequest;
import com.github.copilot.rpc.SetForegroundSessionResponse;
import com.github.copilot.rpc.ToolBinaryResult;
import com.github.copilot.rpc.ToolResultObject;

/**
 * Unit tests for various data transfer objects and record types that were
 * missing coverage, including hook output factory methods, record constructors,
 * and getters for hook inputs.
 */
class DataObjectCoverageTest {

    // ===== PreToolUseHookOutput factory methods =====

    @Test
    void preToolUseHookOutputDenyWithReason() {
        var output = PreToolUseHookOutput.deny("Security policy violation");
        assertEquals("deny", output.permissionDecision());
        assertEquals("Security policy violation", output.permissionDecisionReason());
        assertNull(output.modifiedArgs());
    }

    @Test
    void preToolUseHookOutputAsk() {
        var output = PreToolUseHookOutput.ask();
        assertEquals("ask", output.permissionDecision());
        assertNull(output.permissionDecisionReason());
    }

    @Test
    void preToolUseHookOutputWithModifiedArgs() {
        ObjectNode args = JsonNodeFactory.instance.objectNode();
        args.put("path", "/safe/path");

        var output = PreToolUseHookOutput.withModifiedArgs("allow", args);
        assertEquals("allow", output.permissionDecision());
        assertEquals(args, output.modifiedArgs());
    }

    // ===== PostToolUseHookOutput record =====

    @Test
    void postToolUseHookOutputRecord() {
        var output = new PostToolUseHookOutput(null, "Extra context", false);
        assertNull(output.modifiedResult());
        assertEquals("Extra context", output.additionalContext());
        assertFalse(output.suppressOutput());
    }

    // ===== ToolBinaryResult record =====

    @Test
    void toolBinaryResultRecord() {
        var result = new ToolBinaryResult("base64data==", "image/png", "image", "A chart");
        assertEquals("base64data==", result.data());
        assertEquals("image/png", result.mimeType());
        assertEquals("image", result.type());
        assertEquals("A chart", result.description());
    }

    // ===== GetForegroundSessionResponse record =====

    @Test
    void getForegroundSessionResponseRecord() {
        var response = new GetForegroundSessionResponse("session-123", "/home/user/project");
        assertEquals("session-123", response.sessionId());
        assertEquals("/home/user/project", response.workspacePath());
    }

    // ===== SetForegroundSessionRequest record =====

    @Test
    void setForegroundSessionRequestRecord() {
        var request = new SetForegroundSessionRequest("session-123");
        assertEquals("session-123", request.sessionId());
    }

    // ===== SetForegroundSessionResponse record =====

    @Test
    void setForegroundSessionResponseRecord() {
        var successResponse = new SetForegroundSessionResponse(true, null);
        assertTrue(successResponse.success());
        assertNull(successResponse.error());

        var errorResponse = new SetForegroundSessionResponse(false, "Session not found");
        assertFalse(errorResponse.success());
        assertEquals("Session not found", errorResponse.error());
    }

    // ===== ToolResultObject factory methods =====

    @Test
    void toolResultObjectErrorWithTextAndError() {
        var result = ToolResultObject.error("partial output", "File not found");
        assertEquals("error", result.resultType());
        assertEquals("partial output", result.textResultForLlm());
        assertEquals("File not found", result.error());
    }

    @Test
    void toolResultObjectFailure() {
        var result = ToolResultObject.failure("Tool unavailable", "Unknown tool");
        assertEquals("failure", result.resultType());
        assertEquals("Tool unavailable", result.textResultForLlm());
        assertEquals("Unknown tool", result.error());
    }

    // ===== PermissionRequest additional setters =====

    @Test
    void permissionRequestSetExtensionData() {
        var req = new PermissionRequest();
        req.setExtensionData(java.util.Map.of("key", "value"));
        assertEquals("value", req.getExtensionData().get("key"));
    }

    @Test
    void permissionRequestPreservesMcpExtensionData() {
        var request = PermissionRequest.fromJsonValue(
                java.util.Map.of("kind", "mcp", "serverName", "playwright", "toolName", "playwright-browser_navigate",
                        "args", java.util.Map.of("url", "http://127.0.0.1:8106/docs/target-app/")));

        assertEquals("mcp", request.getKind());
        assertEquals("playwright", request.getExtensionData().get("serverName"));
        assertEquals("playwright-browser_navigate", request.getExtensionData().get("toolName"));
        @SuppressWarnings("unchecked")
        var args = (java.util.Map<String, Object>) request.getExtensionData().get("args");
        assertEquals("http://127.0.0.1:8106/docs/target-app/", args.get("url"));
    }

    @Test
    void permissionRequestWithoutExtensionDataPreservesNull() {
        var request = PermissionRequest.fromJsonValue(java.util.Map.of("kind", "read", "toolCallId", "tool-123"));

        assertNull(request.getExtensionData());
    }

    // ===== SectionOverride setContent =====

    @Test
    void sectionOverrideSetContent() {
        var override = new SectionOverride();
        override.setContent("Custom content");
        assertEquals("Custom content", override.getContent());
    }

    // ===== PreToolUseHookInput getters =====

    @Test
    void preToolUseHookInputGetters() {
        var input = new PreToolUseHookInput();
        // Default values
        assertEquals(0L, input.getTimestamp());
        assertNull(input.getCwd());
        assertNull(input.getToolArgs());
        assertNull(input.getSessionId());
    }

    @Test
    void preToolUseHookInputSessionIdRoundTrip() {
        var input = new PreToolUseHookInput();
        input.setSessionId("session-abc");
        assertEquals("session-abc", input.getSessionId());
    }

    // ===== PostToolUseHookInput getters =====

    @Test
    void postToolUseHookInputGetters() {
        var input = new PostToolUseHookInput();
        // Default values
        assertEquals(0L, input.getTimestamp());
        assertNull(input.getCwd());
        assertNull(input.getToolArgs());
        assertNull(input.getSessionId());
    }

    @Test
    void postToolUseHookInputSessionIdRoundTrip() {
        var input = new PostToolUseHookInput();
        input.setSessionId("session-xyz");
        assertEquals("session-xyz", input.getSessionId());
    }

    // ===== CustomAgentConfig model fields =====

    @Test
    void customAgentConfigModelGetterAndSetter() {
        var cfg = new CustomAgentConfig();
        assertNull(cfg.getModel());

        cfg.setModel("claude-haiku-4.5");
        assertEquals("claude-haiku-4.5", cfg.getModel());
    }

    @Test
    void customAgentConfigModelFluentChaining() {
        var cfg = new CustomAgentConfig().setName("reviewer").setModel("gpt-5").setDescription("Code reviewer");
        assertEquals("reviewer", cfg.getName());
        assertEquals("gpt-5", cfg.getModel());
        assertEquals("Code reviewer", cfg.getDescription());
    }

    @Test
    void customAgentConfigModelSerializationRoundTrip() throws Exception {
        var mapper = JsonRpcClient.getObjectMapper();
        var cfg = new CustomAgentConfig().setName("my-agent").setModel("claude-haiku-4.5");

        var json = mapper.writeValueAsString(cfg);
        assertTrue(json.contains("\"model\":\"claude-haiku-4.5\""));

        var deserialized = mapper.readValue(json, CustomAgentConfig.class);
        assertEquals("my-agent", deserialized.getName());
        assertEquals("claude-haiku-4.5", deserialized.getModel());
    }

    @Test
    void customAgentConfigModelOmittedWhenNull() throws Exception {
        var mapper = JsonRpcClient.getObjectMapper();
        var cfg = new CustomAgentConfig().setName("no-model-agent");

        var json = mapper.writeValueAsString(cfg);
        assertFalse(json.contains("\"model\""));
    }

    @Test
    void customAgentConfigReasoningEffortGetterAndFluentSetter() {
        var cfg = new CustomAgentConfig();
        assertNull(cfg.getReasoningEffort());

        var result = cfg.setReasoningEffort("high");
        assertSame(cfg, result);
        assertEquals("high", cfg.getReasoningEffort());
    }

    @Test
    void customAgentConfigReasoningEffortSerializationRoundTrip() throws Exception {
        var mapper = JsonRpcClient.getObjectMapper();
        var cfg = new CustomAgentConfig().setName("reasoning-agent").setReasoningEffort("high");

        var json = mapper.writeValueAsString(cfg);
        assertTrue(json.contains("\"reasoningEffort\":\"high\""));

        var deserialized = mapper.readValue(json, CustomAgentConfig.class);
        assertEquals("high", deserialized.getReasoningEffort());
    }

    @Test
    void customAgentConfigReasoningEffortOmittedWhenNull() throws Exception {
        var mapper = JsonRpcClient.getObjectMapper();
        var json = mapper.writeValueAsString(new CustomAgentConfig().setName("default-agent"));

        assertFalse(json.contains("\"reasoningEffort\""));
    }

    // ===== PermissionRequestResult setRules =====

    @Test
    void permissionRequestResultSetRules() {
        var result = new PermissionRequestResult().setKind("allow");
        var rules = new java.util.ArrayList<Object>();
        rules.add("bash:read");
        rules.add("bash:write");
        result.setRules(rules);
        assertEquals(2, result.getRules().size());
        assertEquals("bash:read", result.getRules().get(0));
    }

    @Test
    void mcpHttpServerConfigCoversGettersAndFluentSetters() {
        var headers = java.util.Map.of("Authorization", "Bearer token");
        var tools = java.util.List.of("*", "search");
        var oauthScopes = java.util.List.of("tools:read", "resources:read");

        var cfg = new McpHttpServerConfig().setUrl("https://mcp.example.com/sse").setHeaders(headers).setTools(tools)
                .setTimeout(45).setOauthClientId("static-client").setOauthScopes(oauthScopes).setOauthPublicClient(true)
                .setOauthGrantType("authorization_code");

        assertEquals("http", cfg.getType());
        assertEquals("https://mcp.example.com/sse", cfg.getUrl());
        assertEquals("Bearer token", cfg.getHeaders().get("Authorization"));
        assertEquals(tools, cfg.getTools());
        assertEquals(45, cfg.getTimeout());
        assertEquals("static-client", cfg.getOauthClientId());
        assertEquals(oauthScopes, cfg.getOauthScopes());
        assertTrue(cfg.getOauthPublicClient());
        assertEquals("authorization_code", cfg.getOauthGrantType());

        var json = new ObjectMapper().valueToTree(cfg);
        assertEquals("static-client", json.path("oauthClientId").asText());
        assertEquals("tools:read", json.path("oauthScopes").get(0).asText());
        assertEquals("resources:read", json.path("oauthScopes").get(1).asText());
        assertTrue(json.path("oauthPublicClient").asBoolean());
        assertEquals("authorization_code", json.path("oauthGrantType").asText());
    }

    @Test
    void mcpStdioServerConfigCoversGettersAndFluentSetters() {
        var args = java.util.List.of("-y", "@modelcontextprotocol/server-filesystem");
        var env = java.util.Map.of("DEBUG", "1");
        var tools = java.util.List.of("*");

        var cfg = new McpStdioServerConfig().setCommand("npx").setArgs(args).setEnv(env).setWorkingDirectory("/tmp")
                .setTools(tools).setTimeout(30);

        assertEquals("stdio", cfg.getType());
        assertEquals("npx", cfg.getCommand());
        assertEquals(args, cfg.getArgs());
        assertEquals("1", cfg.getEnv().get("DEBUG"));
        assertEquals("/tmp", cfg.getWorkingDirectory());
        assertEquals(tools, cfg.getTools());
        assertEquals(30, cfg.getTimeout());
    }

    @Test
    void mcpStdioServerConfigSerializesWorkingDirectoryAsCwd() {
        var json = new ObjectMapper()
                .valueToTree(new McpStdioServerConfig().setCommand("node").setWorkingDirectory("/workspace"));

        assertEquals("/workspace", json.path("cwd").asText());
        assertFalse(json.has("workingDirectory"));
    }

    @Test
    void modelCapabilitiesOverrideCoversNestedSupportsAndLimits() {
        var supports = new ModelCapabilitiesOverride.Supports().setVision(true).setReasoningEffort(false);
        var limits = new ModelCapabilitiesOverride.Limits().setMaxPromptTokens(2048).setMaxOutputTokens(512)
                .setMaxContextWindowTokens(8192);

        var override = new ModelCapabilitiesOverride().setSupports(supports).setLimits(limits);

        assertTrue(override.getSupports().getVision().orElse(false));
        assertFalse(override.getSupports().getReasoningEffort().orElse(true));
        assertEquals(2048, override.getLimits().getMaxPromptTokens().getAsInt());
        assertEquals(512, override.getLimits().getMaxOutputTokens().getAsInt());
        assertEquals(8192, override.getLimits().getMaxContextWindowTokens().getAsInt());
    }
}