Function TestPerSessionAuthE2E

go/internal/e2e/per_session_auth_e2e_test.go:11–143

(t *testing.T)

)

func TestPerSessionAuthE2E(t *testing.T) {
	ctx := testharness.NewTestContext(t)

	// Create client with COPILOT_DEBUG_GITHUB_API_URL redirected to the proxy
	// so per-session auth token resolution (fetchCopilotUser) is intercepted.
	client := ctx.NewClient(func(opts *copilot.ClientOptions) {
		opts.Env = append(opts.Env, "COPILOT_DEBUG_GITHUB_API_URL="+ctx.ProxyURL)
	})
	t.Cleanup(func() { client.ForceStop() })
	// Register per-token user configs on the proxy
	if err := ctx.SetCopilotUserByToken("token-alice", map[string]interface{}{
		"login":                 "alice",
		"copilot_plan":          "individual_pro",
		"endpoints":             map[string]interface{}{"api": ctx.ProxyURL, "telemetry": "https://localhost:1/telemetry"},
		"analytics_tracking_id": "alice-tracking-id",
	}); err != nil {
		t.Fatalf("Failed to set copilot user for alice: %v", err)
	}

	if err := ctx.SetCopilotUserByToken("token-bob", map[string]interface{}{
		"login":                 "bob",
		"copilot_plan":          "business",
		"endpoints":             map[string]interface{}{"api": ctx.ProxyURL, "telemetry": "https://localhost:1/telemetry"},
		"analytics_tracking_id": "bob-tracking-id",
	}); err != nil {
		t.Fatalf("Failed to set copilot user for bob: %v", err)
	}

	t.Run("should authenticate with per-session token", func(t *testing.T) {
		ctx.ConfigureForTest(t)

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

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

		if !authStatus.IsAuthenticated {
			t.Errorf("Expected session to be authenticated")
		}
		if authStatus.Login == nil || *authStatus.Login != "alice" {
			t.Errorf("Expected login to be 'alice', got %v", authStatus.Login)
		}
	})

	t.Run("should isolate auth between sessions", func(t *testing.T) {
		ctx.ConfigureForTest(t)

		sessionA, err := client.CreateSession(t.Context(), &copilot.SessionConfig{
			OnPermissionRequest: copilot.PermissionHandler.ApproveAll,
			GitHubToken:         "token-alice",