client_e2e_test.go

package e2e

import (
	"testing"
	"time"

	copilot "github.com/github/copilot-sdk/go"
	"github.com/github/copilot-sdk/go/internal/e2e/testharness"
)

func TestClientE2E(t *testing.T) {
	cliPath := testharness.CLIPath()
	if cliPath == "" {
		t.Fatal("CLI not found. Run 'npm install' in the nodejs directory first.")
	}

	t.Run("should start and connect to server using stdio", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.StdioConnection{Path: cliPath},
		})
		t.Cleanup(func() { client.ForceStop() })

		if err := client.Start(t.Context()); err != nil {
			t.Fatalf("Failed to start client: %v", err)
		}

		pong, err := client.Ping(t.Context(), "test message")
		if err != nil {
			t.Fatalf("Failed to ping: %v", err)
		}

		if pong.Message != "pong: test message" {
			t.Errorf("Expected pong.message to be 'pong: test message', got %q", pong.Message)
		}

		if pong.Timestamp.IsZero() {
			t.Errorf("Expected non-zero pong.timestamp, got %s", pong.Timestamp)
		}

		if err := client.Stop(); err != nil {
			t.Errorf("Expected no errors on stop, got %v", err)
		}
	})

	t.Run("should start and connect to server using tcp", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.TCPConnection{Path: cliPath},
		})
		t.Cleanup(func() { client.ForceStop() })

		if err := client.Start(t.Context()); err != nil {
			t.Fatalf("Failed to start client: %v", err)
		}

		pong, err := client.Ping(t.Context(), "test message")
		if err != nil {
			t.Fatalf("Failed to ping: %v", err)
		}

		if pong.Message != "pong: test message" {
			t.Errorf("Expected pong.message to be 'pong: test message', got %q", pong.Message)
		}

		if pong.Timestamp.IsZero() {
			t.Errorf("Expected non-zero pong.timestamp, got %s", pong.Timestamp)
		}

		if err := client.Stop(); err != nil {
			t.Errorf("Expected no errors on stop, got %v", err)
		}
	})

	t.Run("should return errors on failed cleanup", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.StdioConnection{Path: cliPath},
		})
		t.Cleanup(func() { client.ForceStop() })

		_, err := client.CreateSession(t.Context(), &copilot.SessionConfig{
			OnPermissionRequest: copilot.PermissionHandler.ApproveAll,
		})
		if err != nil {
			t.Fatalf("Failed to create session: %v", err)
		}

		// Kill the server process to force cleanup to fail
		client.ForceStop()
		time.Sleep(100 * time.Millisecond)

		if err := client.Stop(); err != nil {
			t.Logf("Got expected errors: %v", err)
		}
	})

	t.Run("should forceStop without cleanup", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.StdioConnection{Path: cliPath},
		})
		t.Cleanup(func() { client.ForceStop() })

		_, err := client.CreateSession(t.Context(), &copilot.SessionConfig{
			OnPermissionRequest: copilot.PermissionHandler.ApproveAll,
		})
		if err != nil {
			t.Fatalf("Failed to create session: %v", err)
		}

		client.ForceStop()
	})

	t.Run("should get status with version and protocol info", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.StdioConnection{Path: cliPath},
		})
		t.Cleanup(func() { client.ForceStop() })

		if err := client.Start(t.Context()); err != nil {
			t.Fatalf("Failed to start client: %v", err)
		}

		status, err := client.GetStatus(t.Context())
		if err != nil {
			t.Fatalf("Failed to get status: %v", err)
		}

		if status.Version == "" {
			t.Error("Expected status.Version to be non-empty")
		}

		if status.ProtocolVersion < 1 {
			t.Errorf("Expected status.ProtocolVersion >= 1, got %d", status.ProtocolVersion)
		}

		client.Stop()
	})

	t.Run("should get auth status", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.StdioConnection{Path: cliPath},
		})
		t.Cleanup(func() { client.ForceStop() })

		if err := client.Start(t.Context()); err != nil {
			t.Fatalf("Failed to start client: %v", err)
		}

		authStatus, err := client.GetAuthStatus(t.Context())
		if err != nil {
			t.Fatalf("Failed to get auth status: %v", err)
		}

		// isAuthenticated is a bool, just verify we got a response
		if authStatus.IsAuthenticated {
			if authStatus.AuthType == nil {
				t.Error("Expected authType to be set when authenticated")
			}
			if authStatus.StatusMessage == nil {
				t.Error("Expected statusMessage to be set when authenticated")
			}
		}

		client.Stop()
	})

	t.Run("should list models when authenticated", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.StdioConnection{Path: cliPath},
		})
		t.Cleanup(func() { client.ForceStop() })

		if err := client.Start(t.Context()); err != nil {
			t.Fatalf("Failed to start client: %v", err)
		}

		authStatus, err := client.GetAuthStatus(t.Context())
		if err != nil {
			t.Fatalf("Failed to get auth status: %v", err)
		}

		if !authStatus.IsAuthenticated {
			// Skip if not authenticated - models.list requires auth
			client.Stop()
			return
		}

		models, err := client.ListModels(t.Context())
		if err != nil {
			t.Fatalf("Failed to list models: %v", err)
		}

		if len(models) > 0 {
			model := models[0]
			if model.ID == "" {
				t.Error("Expected model.ID to be non-empty")
			}
			if model.Name == "" {
				t.Error("Expected model.Name to be non-empty")
			}
		}

		client.Stop()
	})

	t.Run("should report error when CLI fails to start", func(t *testing.T) {
		client := copilot.NewClient(&copilot.ClientOptions{
			Connection: copilot.StdioConnection{
				Path: cliPath,
				Args: []string{"--nonexistent-flag-for-testing"},
			},
		})
		t.Cleanup(func() { client.ForceStop() })

		err := client.Start(t.Context())
		if err == nil {
			t.Fatal("Expected Start to fail with invalid CLI args")
		}

		// Verify subsequent calls also fail (don't hang)
		session, err := client.CreateSession(t.Context(), nil)
		if err == nil {
			_, err = session.Send(t.Context(), copilot.MessageOptions{Prompt: "test"})
		}
		if err == nil {
			t.Fatal("Expected CreateSession/Send to fail after CLI exit")
		}
	})
}