Fix test to handle new IllegalStateException from closed sessions

Co-authored-by: brunoborges <129743+brunoborges@users.noreply.github.com>

Author: Copilot

src/main/java/com/github/copilot/sdk/CopilotSession.java

@@ -97,7 +97,7 @@ public final class CopilotSession implements AutoCloseable {
     private final AtomicReference<SessionHooks> hooksHandler = new AtomicReference<>();
     private volatile EventErrorHandler eventErrorHandler;
     private volatile EventErrorPolicy eventErrorPolicy = EventErrorPolicy.PROPAGATE_AND_LOG_ERRORS;
-    
+
     /** Tracks whether this session instance has been terminated via close(). */
     private volatile boolean isTerminated = false;
 
@@ -189,7 +189,8 @@ public String getWorkspacePath() {
      * @param handler
      *            the error handler, or {@code null} to use only the default logging
      *            behavior
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see EventErrorHandler
      * @see #setEventErrorPolicy(EventErrorPolicy)
      * @since 1.0.8
@@ -229,7 +230,8 @@ public void setEventErrorHandler(EventErrorHandler handler) {
      * @param policy
      *            the error policy (default is
      *            {@link EventErrorPolicy#PROPAGATE_AND_LOG_ERRORS})
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see EventErrorPolicy
      * @see #setEventErrorHandler(EventErrorHandler)
      * @since 1.0.8
@@ -251,7 +253,8 @@ public void setEventErrorPolicy(EventErrorPolicy policy) {
      * @param prompt
      *            the message text to send
      * @return a future that resolves with the message ID assigned by the server
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see #send(MessageOptions)
      */
     public CompletableFuture<String> send(String prompt) {
@@ -269,7 +272,8 @@ public CompletableFuture<String> send(String prompt) {
      *            the message text to send
      * @return a future that resolves with the final assistant message event, or
      *         {@code null} if no assistant message was received
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see #sendAndWait(MessageOptions)
      */
     public CompletableFuture<AssistantMessageEvent> sendAndWait(String prompt) {
@@ -286,7 +290,8 @@ public CompletableFuture<AssistantMessageEvent> sendAndWait(String prompt) {
      * @param options
      *            the message options containing the prompt and attachments
      * @return a future that resolves with the message ID assigned by the server
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see #sendAndWait(MessageOptions)
      * @see #send(String)
      */
@@ -317,7 +322,8 @@ public CompletableFuture<String> send(MessageOptions options) {
      *         {@code null} if no assistant message was received. The future
      *         completes exceptionally with a TimeoutException if the timeout
      *         expires.
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see #sendAndWait(MessageOptions)
      * @see #send(MessageOptions)
      */
@@ -380,7 +386,8 @@ public CompletableFuture<AssistantMessageEvent> sendAndWait(MessageOptions optio
      *            the message options containing the prompt and attachments
      * @return a future that resolves with the final assistant message event, or
      *         {@code null} if no assistant message was received
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see #sendAndWait(MessageOptions, long)
      */
     public CompletableFuture<AssistantMessageEvent> sendAndWait(MessageOptions options) {
@@ -414,7 +421,8 @@ public CompletableFuture<AssistantMessageEvent> sendAndWait(MessageOptions optio
      * @param handler
      *            a callback to be invoked when a session event occurs
      * @return a Closeable that, when closed, unsubscribes the handler
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see #on(Class, Consumer)
      * @see AbstractSessionEvent
      * @see #setEventErrorPolicy(EventErrorPolicy)
@@ -466,7 +474,8 @@ public Closeable on(Consumer<AbstractSessionEvent> handler) {
      * @param handler
      *            a callback invoked when events of this type occur
      * @return a Closeable that unsubscribes the handler when closed
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see #on(Consumer)
      * @see AbstractSessionEvent
      */
@@ -729,7 +738,8 @@ CompletableFuture<Object> handleHooksInvoke(String hookType, JsonNode input) {
      * assistant responses, tool invocations, and other session events.
      *
      * @return a future that resolves with a list of all session events
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      * @see AbstractSessionEvent
      */
     public CompletableFuture<List<AbstractSessionEvent>> getMessages() {
@@ -760,17 +770,19 @@ public CompletableFuture<List<AbstractSessionEvent>> getMessages() {
      * continuing to generate a response.
      *
      * @return a future that completes when the abort is acknowledged
-     * @throws IllegalStateException if this session has been terminated
+     * @throws IllegalStateException
+     *             if this session has been terminated
      */
     public CompletableFuture<Void> abort() {
         ensureNotTerminated();
         return rpc.invoke("session.abort", Map.of("sessionId", sessionId), Void.class);
     }
-    
+
     /**
      * Verifies that this session has not yet been terminated.
-     * 
-     * @throws IllegalStateException if close() has already been invoked
+     *
+     * @throws IllegalStateException
+     *             if close() has already been invoked
      */
     private void ensureNotTerminated() {
         if (isTerminated) {

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

@@ -17,11 +17,11 @@
 
 /**
  * Tests for closed-session guard functionality in CopilotSession.
- * 
+ *
  * <p>
  * Verifies that all public methods that interact with session state throw
- * IllegalStateException when invoked after close(), and that close() itself
- * is idempotent.
+ * IllegalStateException when invoked after close(), and that close() itself is
+ * idempotent.
  * </p>
  */
 public class ClosedSessionGuardTest {
@@ -60,8 +60,8 @@ void testSendStringThrowsAfterTermination() throws Exception {
     }
 
     /**
-     * Verifies that send(MessageOptions) throws IllegalStateException after
-     * session is terminated.
+     * Verifies that send(MessageOptions) throws IllegalStateException after session
+     * is terminated.
      */
     @Test
     void testSendOptionsThrowsAfterTermination() throws Exception {

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

@@ -79,13 +79,15 @@ void testShouldReceiveSessionEvents_createAndDestroy() throws Exception {
 
             session.close();
 
-            // Session should no longer be accessible
+            // Session should no longer be accessible - now throws IllegalStateException
             try {
                 session.getMessages().get();
                 fail("Expected exception for closed session");
             } catch (Exception e) {
-                assertTrue(e.getMessage().toLowerCase().contains("not found")
-                        || e.getCause().getMessage().toLowerCase().contains("not found"));
+                // After our changes, we now get IllegalStateException directly
+                assertTrue(e.getMessage().toLowerCase().contains("closed")
+                        || (e.getCause() != null
+                                && e.getCause().getMessage().toLowerCase().contains("not found")));
             }
         }
     }