ConfigCloneTest.java

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

package com.github.copilot.sdk;

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

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.CompletableFuture;
import java.util.function.Consumer;

import org.junit.jupiter.api.Test;

import com.github.copilot.sdk.events.AbstractSessionEvent;
import com.github.copilot.sdk.json.CopilotClientOptions;
import com.github.copilot.sdk.json.MessageOptions;
import com.github.copilot.sdk.json.ModelInfo;
import com.github.copilot.sdk.json.ResumeSessionConfig;
import com.github.copilot.sdk.json.SessionConfig;

class ConfigCloneTest {

    @Test
    void copilotClientOptionsCloneBasic() {
        CopilotClientOptions original = new CopilotClientOptions();
        original.setCliPath("/usr/local/bin/copilot");
        original.setLogLevel("debug");
        original.setPort(9000);
        original.setGitHubToken("ghp_test");
        original.setUseLoggedInUser(false);

        CopilotClientOptions cloned = original.clone();

        assertEquals(original.getCliPath(), cloned.getCliPath());
        assertEquals(original.getLogLevel(), cloned.getLogLevel());
        assertEquals(original.getPort(), cloned.getPort());
        assertEquals(original.getGitHubToken(), cloned.getGitHubToken());
        assertEquals(original.getUseLoggedInUser(), cloned.getUseLoggedInUser());
    }

    @Test
    void copilotClientOptionsArrayIndependence() {
        CopilotClientOptions original = new CopilotClientOptions();
        String[] args = {"--flag1", "--flag2"};
        original.setCliArgs(args);

        CopilotClientOptions cloned = original.clone();
        cloned.getCliArgs()[0] = "--changed";

        assertEquals("--flag1", original.getCliArgs()[0]);
        assertEquals("--changed", cloned.getCliArgs()[0]);
    }

    @Test
    void copilotClientOptionsEnvironmentIndependence() {
        CopilotClientOptions original = new CopilotClientOptions();
        Map<String, String> env = new HashMap<>();
        env.put("KEY1", "value1");
        original.setEnvironment(env);

        CopilotClientOptions cloned = original.clone();

        // Mutate the original environment map to test independence
        env.put("KEY2", "value2");

        // The cloned config should be unaffected by mutations to the original map
        assertEquals(1, cloned.getEnvironment().size());
        assertEquals(2, original.getEnvironment().size());
    }

    @Test
    void copilotClientOptionsOnListModelsCloned() {
        CopilotClientOptions original = new CopilotClientOptions();
        List<ModelInfo> models = List.of(new ModelInfo());
        original.setOnListModels(() -> CompletableFuture.completedFuture(models));

        CopilotClientOptions cloned = original.clone();

        assertNotNull(cloned.getOnListModels());
        assertSame(original.getOnListModels(), cloned.getOnListModels());
    }

    @Test
    void sessionConfigCloneBasic() {
        SessionConfig original = new SessionConfig();
        original.setSessionId("my-session");
        original.setClientName("my-app");
        original.setModel("gpt-4o");
        original.setStreaming(true);

        SessionConfig cloned = original.clone();

        assertEquals(original.getSessionId(), cloned.getSessionId());
        assertEquals(original.getClientName(), cloned.getClientName());
        assertEquals(original.getModel(), cloned.getModel());
        assertEquals(original.isStreaming(), cloned.isStreaming());
    }

    @Test
    void sessionConfigListIndependence() {
        SessionConfig original = new SessionConfig();
        List<String> toolList = new ArrayList<>();
        toolList.add("grep");
        toolList.add("bash");
        original.setAvailableTools(toolList);

        SessionConfig cloned = original.clone();

        // Mutate the original list directly to test independence
        toolList.add("web");

        // The cloned config should be unaffected by mutations to the original list
        assertEquals(2, cloned.getAvailableTools().size());
        assertEquals(3, original.getAvailableTools().size());
    }

    @Test
    void sessionConfigAgentAndOnEventCloned() {
        Consumer<AbstractSessionEvent> handler = event -> {
        };
        SessionConfig original = new SessionConfig();
        original.setAgent("my-agent");
        original.setOnEvent(handler);

        SessionConfig cloned = original.clone();

        assertEquals("my-agent", cloned.getAgent());
        assertSame(handler, cloned.getOnEvent());
    }

    @Test
    void resumeSessionConfigCloneBasic() {
        ResumeSessionConfig original = new ResumeSessionConfig();
        original.setModel("o1");
        original.setStreaming(false);

        ResumeSessionConfig cloned = original.clone();

        assertEquals(original.getModel(), cloned.getModel());
        assertEquals(original.isStreaming(), cloned.isStreaming());
    }

    @Test
    void resumeSessionConfigAgentAndOnEventCloned() {
        Consumer<AbstractSessionEvent> handler = event -> {
        };
        ResumeSessionConfig original = new ResumeSessionConfig();
        original.setAgent("my-agent");
        original.setOnEvent(handler);

        ResumeSessionConfig cloned = original.clone();

        assertEquals("my-agent", cloned.getAgent());
        assertSame(handler, cloned.getOnEvent());
    }

    @Test
    void messageOptionsCloneBasic() {
        MessageOptions original = new MessageOptions();
        original.setPrompt("What is 2+2?");
        original.setMode("immediate");

        MessageOptions cloned = original.clone();

        assertEquals(original.getPrompt(), cloned.getPrompt());
        assertEquals(original.getMode(), cloned.getMode());
    }

    @Test
    void clonePreservesNullFields() {
        CopilotClientOptions opts = new CopilotClientOptions();
        CopilotClientOptions optsClone = opts.clone();
        assertNull(optsClone.getCliPath());

        SessionConfig cfg = new SessionConfig();
        SessionConfig cfgClone = cfg.clone();
        assertNull(cfgClone.getModel());

        MessageOptions msg = new MessageOptions();
        MessageOptions msgClone = msg.clone();
        assertNull(msgClone.getMode());
    }
}