Add tests for exit-plan-mode and auto-mode-switch handler APIs

Co-authored-by: edburns <75821+edburns@users.noreply.github.com>

Author: Copilot

src/test/java/com/github/copilot/sdk/ConfigCloneTest.java

@@ -26,6 +26,9 @@
 import com.github.copilot.sdk.json.SystemMessageConfig;
 import com.github.copilot.sdk.json.TelemetryConfig;
 
+import com.github.copilot.sdk.json.AutoModeSwitchResponse;
+import com.github.copilot.sdk.json.ExitPlanModeResult;
+
 class ConfigCloneTest {
 
     @Test
@@ -375,4 +378,32 @@ void copilotClientOptionsSessionIdleTimeoutCloned() {
 
         assertEquals(600, cloned.getSessionIdleTimeoutSeconds());
     }
+
+    @Test
+    void sessionConfigModeSwitchHandlersCloned() {
+        SessionConfig original = new SessionConfig();
+        original.setOnExitPlanMode(
+                (request, invocation) -> CompletableFuture.completedFuture(new ExitPlanModeResult()));
+        original.setOnAutoModeSwitch(
+                (request, invocation) -> CompletableFuture.completedFuture(AutoModeSwitchResponse.NO));
+
+        SessionConfig cloned = original.clone();
+
+        assertSame(original.getOnExitPlanMode(), cloned.getOnExitPlanMode());
+        assertSame(original.getOnAutoModeSwitch(), cloned.getOnAutoModeSwitch());
+    }
+
+    @Test
+    void resumeSessionConfigModeSwitchHandlersCloned() {
+        ResumeSessionConfig original = new ResumeSessionConfig();
+        original.setOnExitPlanMode(
+                (request, invocation) -> CompletableFuture.completedFuture(new ExitPlanModeResult()));
+        original.setOnAutoModeSwitch(
+                (request, invocation) -> CompletableFuture.completedFuture(AutoModeSwitchResponse.NO));
+
+        ResumeSessionConfig cloned = original.clone();
+
+        assertSame(original.getOnExitPlanMode(), cloned.getOnExitPlanMode());
+        assertSame(original.getOnAutoModeSwitch(), cloned.getOnAutoModeSwitch());
+    }
 }

src/test/java/com/github/copilot/sdk/SessionRequestBuilderTest.java

@@ -12,11 +12,15 @@
 
 import org.junit.jupiter.api.Test;
 
+import com.github.copilot.sdk.json.AutoModeSwitchHandler;
+import com.github.copilot.sdk.json.AutoModeSwitchResponse;
 import com.github.copilot.sdk.json.CreateSessionRequest;
 import com.github.copilot.sdk.json.DefaultAgentConfig;
 import com.github.copilot.sdk.json.ElicitationHandler;
 import com.github.copilot.sdk.json.ElicitationResult;
 import com.github.copilot.sdk.json.ElicitationResultAction;
+import com.github.copilot.sdk.json.ExitPlanModeHandler;
+import com.github.copilot.sdk.json.ExitPlanModeResult;
 import com.github.copilot.sdk.json.ResumeSessionConfig;
 import com.github.copilot.sdk.json.ResumeSessionRequest;
 import com.github.copilot.sdk.json.SessionConfig;
@@ -535,4 +539,136 @@ void testResumeRequestOmitsEnableSessionTelemetryWhenNull() throws Exception {
         var json = mapper.writeValueAsString(request);
         assertFalse(json.contains("enableSessionTelemetry"), "enableSessionTelemetry should be omitted when null");
     }
+
+    @Test
+    void buildCreateRequest_setsRequestExitPlanModeWhenHandlerPresent() {
+        var config = new SessionConfig().setOnExitPlanMode(
+                (request, invocation) -> CompletableFuture.completedFuture(new ExitPlanModeResult()));
+
+        CreateSessionRequest result = SessionRequestBuilder.buildCreateRequest(config, "sess-epm");
+
+        assertEquals(true, result.getRequestExitPlanMode());
+    }
+
+    @Test
+    void buildCreateRequest_omitsRequestExitPlanModeWhenNoHandler() {
+        var config = new SessionConfig();
+
+        CreateSessionRequest result = SessionRequestBuilder.buildCreateRequest(config, "sess-epm-none");
+
+        assertNull(result.getRequestExitPlanMode());
+    }
+
+    @Test
+    void buildCreateRequest_setsRequestAutoModeSwitchWhenHandlerPresent() {
+        var config = new SessionConfig().setOnAutoModeSwitch(
+                (request, invocation) -> CompletableFuture.completedFuture(AutoModeSwitchResponse.NO));
+
+        CreateSessionRequest result = SessionRequestBuilder.buildCreateRequest(config, "sess-ams");
+
+        assertEquals(true, result.getRequestAutoModeSwitch());
+    }
+
+    @Test
+    void buildCreateRequest_omitsRequestAutoModeSwitchWhenNoHandler() {
+        var config = new SessionConfig();
+
+        CreateSessionRequest result = SessionRequestBuilder.buildCreateRequest(config, "sess-ams-none");
+
+        assertNull(result.getRequestAutoModeSwitch());
+    }
+
+    @Test
+    void buildResumeRequest_setsRequestExitPlanModeWhenHandlerPresent() {
+        var config = new ResumeSessionConfig().setOnExitPlanMode(
+                (request, invocation) -> CompletableFuture.completedFuture(new ExitPlanModeResult()));
+
+        ResumeSessionRequest result = SessionRequestBuilder.buildResumeRequest("sess-epm", config);
+
+        assertEquals(true, result.getRequestExitPlanMode());
+    }
+
+    @Test
+    void buildResumeRequest_setsRequestAutoModeSwitchWhenHandlerPresent() {
+        var config = new ResumeSessionConfig().setOnAutoModeSwitch(
+                (request, invocation) -> CompletableFuture.completedFuture(AutoModeSwitchResponse.YES));
+
+        ResumeSessionRequest result = SessionRequestBuilder.buildResumeRequest("sess-ams", config);
+
+        assertEquals(true, result.getRequestAutoModeSwitch());
+    }
+
+    @Test
+    void configureSessionWithExitPlanModeHandler_registersHandler() {
+        CopilotSession session = new CopilotSession("session-1", null);
+
+        ExitPlanModeHandler handler = (request, invocation) -> CompletableFuture
+                .completedFuture(new ExitPlanModeResult());
+        var config = new SessionConfig().setOnExitPlanMode(handler);
+
+        SessionRequestBuilder.configureSession(session, config);
+    }
+
+    @Test
+    void configureSessionWithAutoModeSwitchHandler_registersHandler() {
+        CopilotSession session = new CopilotSession("session-1", null);
+
+        AutoModeSwitchHandler handler = (request, invocation) -> CompletableFuture
+                .completedFuture(AutoModeSwitchResponse.NO);
+        var config = new SessionConfig().setOnAutoModeSwitch(handler);
+
+        SessionRequestBuilder.configureSession(session, config);
+    }
+
+    @Test
+    void configureResumedSessionWithExitPlanModeHandler_registersHandler() {
+        CopilotSession session = new CopilotSession("session-1", null);
+
+        ExitPlanModeHandler handler = (request, invocation) -> CompletableFuture
+                .completedFuture(new ExitPlanModeResult());
+        var config = new ResumeSessionConfig().setOnExitPlanMode(handler);
+
+        SessionRequestBuilder.configureSession(session, config);
+    }
+
+    @Test
+    void configureResumedSessionWithAutoModeSwitchHandler_registersHandler() {
+        CopilotSession session = new CopilotSession("session-1", null);
+
+        AutoModeSwitchHandler handler = (request, invocation) -> CompletableFuture
+                .completedFuture(AutoModeSwitchResponse.YES);
+        var config = new ResumeSessionConfig().setOnAutoModeSwitch(handler);
+
+        SessionRequestBuilder.configureSession(session, config);
+    }
+
+    @Test
+    void buildCreateRequest_serializesExitPlanModeAndAutoModeSwitchFlags() throws Exception {
+        var config = new SessionConfig()
+                .setOnExitPlanMode((request, invocation) -> CompletableFuture.completedFuture(new ExitPlanModeResult()))
+                .setOnAutoModeSwitch(
+                        (request, invocation) -> CompletableFuture.completedFuture(AutoModeSwitchResponse.NO));
+
+        CreateSessionRequest result = SessionRequestBuilder.buildCreateRequest(config, "sess-flags");
+
+        var mapper = JsonRpcClient.getObjectMapper();
+        var json = mapper.writeValueAsString(result);
+        assertTrue(json.contains("\"requestExitPlanMode\":true"));
+        assertTrue(json.contains("\"requestAutoModeSwitch\":true"));
+    }
+
+    @Test
+    void buildResumeRequest_serializesExitPlanModeAndAutoModeSwitchFlags() throws Exception {
+        var config = new ResumeSessionConfig()
+                .setOnExitPlanMode((request, invocation) -> CompletableFuture.completedFuture(new ExitPlanModeResult()))
+                .setOnAutoModeSwitch(
+                        (request, invocation) -> CompletableFuture.completedFuture(AutoModeSwitchResponse.YES));
+
+        ResumeSessionRequest result = SessionRequestBuilder.buildResumeRequest("sess-flags", config);
+
+        var mapper = JsonRpcClient.getObjectMapper();
+        var json = mapper.writeValueAsString(result);
+        assertTrue(json.contains("\"requestExitPlanMode\":true"));
+        assertTrue(json.contains("\"requestAutoModeSwitch\":true"));
+    }
 }