package executor import (	"bytes"	"context"	"net/http"	"net/http/httptest"	"strings"	"testing"	"time" 	"github.com/router-for-me/CLIProxyAPI/v7/internal/config"	_ "github.com/router-for-me/CLIProxyAPI/v7/internal/translator"	cliproxyauth "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/auth"	cliproxyexecutor "github.com/router-for-me/CLIProxyAPI/v7/sdk/cliproxy/executor"	sdktranslator "github.com/router-for-me/CLIProxyAPI/v7/sdk/translator"	"github.com/tidwall/gjson") func TestCodexExecutorExecute_EmptyStreamCompletionOutputUsesOutputItemDone(t *testing.T) {	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {		w.Header().Set("Content-Type", "text/event-stream")		_, _ = w.Write([]byte("data: {\"type\":\"response.output_item.done\",\"item\":{\"type\":\"message\",\"role\":\"assistant\",\"content\":[{\"type\":\"output_text\",\"text\":\"ok\"}]},\"output_index\":0}\n"))		_, _ = w.Write([]byte("data: {\"type\":\"response.completed\",\"response\":{\"id\":\"resp_1\",\"object\":\"response\",\"created_at\":1775555723,\"status\":\"completed\",\"model\":\"gpt-5.4-mini-2026-03-17\",\"output\":[],\"usage\":{\"input_tokens\":8,\"output_tokens\":28,\"total_tokens\":36}}}\n\n"))	}))	defer server.Close() 	executor := NewCodexExecutor(&config.Config{})	auth := &cliproxyauth.Auth{Attributes: map[string]string{		"base_url": server.URL,		"api_key":  "test",	}} 	resp, err := executor.Execute(context.Background(), auth, cliproxyexecutor.Request{		Model:   "gpt-5.4-mini",		Payload: []byte(`{"model":"gpt-5.4-mini","messages":[{"role":"user","content":"Say ok"}]}`),	}, cliproxyexecutor.Options{		SourceFormat: sdktranslator.FromString("openai"),		Stream:       false,	})	if err != nil {		t.Fatalf("Execute error: %v", err)	} 	gotContent := gjson.GetBytes(resp.Payload, "choices.0.message.content").String()	if gotContent != "ok" {		t.Fatalf("choices.0.message.content = %q, want %q; payload=%s", gotContent, "ok", string(resp.Payload))	}} func TestCodexExecutorExecuteSurfacesTerminalStreamError(t *testing.T) {	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {		w.Header().Set("Content-Type", "text/event-stream")		_, _ = w.Write([]byte("event: response.created\n"))		_, _ = w.Write([]byte(`data: {"type":"response.created","response":{"id":"resp_1","model":"gpt-5.5"}}` + "\n\n"))		_, _ = w.Write([]byte("event: error\n"))		_, _ = w.Write([]byte(`data: {"type":"error","error":{"type":"invalid_request_error","code":"context_length_exceeded","message":"Your input exceeds the context window of this model. Please adjust your input and try again.","param":"input"},"sequence_number":2}` + "\n\n"))		_, _ = w.Write([]byte("event: response.failed\n"))		_, _ = w.Write([]byte(`data: {"type":"response.failed","response":{"id":"resp_1","status":"failed","error":{"code":"context_length_exceeded","message":"Your input exceeds the context window of this model. Please adjust your input and try again."}}}` + "\n\n"))	}))	defer server.Close() 	executor := NewCodexExecutor(&config.Config{})	auth := &cliproxyauth.Auth{Attributes: map[string]string{		"base_url": server.URL,		"api_key":  "test",	}} 	_, err := executor.Execute(context.Background(), auth, cliproxyexecutor.Request{		Model:   "gpt-5.5",		Payload: []byte(`{"model":"gpt-5.5","input":"hello"}`),	}, cliproxyexecutor.Options{		SourceFormat: sdktranslator.FromString("openai-response"),		Stream:       false,	})	if err == nil {		t.Fatal("expected terminal stream error, got nil")	}	if got := statusCodeFromTestError(t, err); got != http.StatusBadRequest {		t.Fatalf("status code = %d, want %d; err=%v", got, http.StatusBadRequest, err)	}	assertCodexErrorCode(t, err.Error(), "invalid_request_error", "context_too_large")	if !strings.Contains(err.Error(), "Your input exceeds the context window") {		t.Fatalf("error message missing upstream context text: %v", err)	}} func TestCodexExecutorExecuteStreamSurfacesTerminalStreamError(t *testing.T) {	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {		w.Header().Set("Content-Type", "text/event-stream")		_, _ = w.Write([]byte("event: response.created\n"))		_, _ = w.Write([]byte(`data: {"type":"response.created","response":{"id":"resp_1","model":"gpt-5.5"}}` + "\n\n"))		_, _ = w.Write([]byte("event: error\n"))		_, _ = w.Write([]byte(`data: {"type":"error","error":{"type":"invalid_request_error","code":"context_length_exceeded","message":"Your input exceeds the context window of this model. Please adjust your input and try again.","param":"input"},"sequence_number":2}` + "\n\n"))	}))	defer server.Close() 	executor := NewCodexExecutor(&config.Config{})	auth := &cliproxyauth.Auth{Attributes: map[string]string{		"base_url": server.URL,		"api_key":  "test",	}} 	result, err := executor.ExecuteStream(context.Background(), auth, cliproxyexecutor.Request{		Model:   "gpt-5.5",		Payload: []byte(`{"model":"gpt-5.5","input":"hello"}`),	}, cliproxyexecutor.Options{		SourceFormat: sdktranslator.FromString("openai-response"),		Stream:       true,	})	if err != nil {		t.Fatalf("ExecuteStream error: %v", err)	} 	var streamErr error	for chunk := range result.Chunks {		if chunk.Err != nil {			streamErr = chunk.Err			break		}	}	if streamErr == nil {		t.Fatal("missing stream terminal error")	}	if got := statusCodeFromTestError(t, streamErr); got != http.StatusBadRequest {		t.Fatalf("status code = %d, want %d; err=%v", got, http.StatusBadRequest, streamErr)	}	assertCodexErrorCode(t, streamErr.Error(), "invalid_request_error", "context_too_large")} func TestCodexTerminalStreamContextLengthErrFromResponseFailed(t *testing.T) {	err, ok := codexTerminalStreamContextLengthErr([]byte(`{"type":"response.failed","response":{"id":"resp_1","status":"failed","error":{"code":"context_length_exceeded","message":"Your input exceeds the context window of this model. Please adjust your input and try again."}}}`))	if !ok {		t.Fatal("expected context length terminal error")	}	if got := statusCodeFromTestError(t, err); got != http.StatusBadRequest {		t.Fatalf("status code = %d, want %d; err=%v", got, http.StatusBadRequest, err)	}	assertCodexErrorCode(t, err.Error(), "invalid_request_error", "context_too_large")} func TestCodexTerminalStreamContextLengthErrFromTopLevelError(t *testing.T) {	err, ok := codexTerminalStreamContextLengthErr([]byte(`{"type":"error","code":"context_length_exceeded","message":"Your input exceeds the context window of this model. Please adjust your input and try again.","sequence_number":2}`))	if !ok {		t.Fatal("expected top-level context length terminal error")	}	if got := statusCodeFromTestError(t, err); got != http.StatusBadRequest {		t.Fatalf("status code = %d, want %d; err=%v", got, http.StatusBadRequest, err)	}	assertCodexErrorCode(t, err.Error(), "invalid_request_error", "context_too_large")	if !strings.Contains(err.Error(), "Your input exceeds the context window") {		t.Fatalf("error message missing upstream context text: %v", err)	}} func TestCodexTerminalStreamContextLengthErrIgnoresOtherTerminalErrors(t *testing.T) {	_, ok := codexTerminalStreamContextLengthErr([]byte(`{"type":"error","error":{"type":"rate_limit_error","code":"rate_limit_exceeded","message":"Rate limit reached."}}`))	if ok {		t.Fatal("rate limit terminal error should not be handled by context length fix")	}} func TestCodexTerminalStreamErrIgnoresRateLimitTerminalErrors(t *testing.T) {	_, _, ok := codexTerminalStreamErr([]byte(`{"type":"error","error":{"type":"rate_limit_error","code":"rate_limit_exceeded","message":"Rate limit reached."}}`))	if ok {		t.Fatal("rate limit terminal error should not be handled by replay terminal error path")	}} func TestCodexTerminalStreamErrHandlesUsageLimitErrorEvent(t *testing.T) {	streamErr, _, ok := codexTerminalStreamErr([]byte(`{"type":"error","error":{"type":"usage_limit_reached","message":"You've hit your usage limit.","resets_in_seconds":300}}`))	if !ok {		t.Fatal("expected usage_limit_reached terminal error to be handled")	}	if got := statusCodeFromTestError(t, streamErr); got != http.StatusTooManyRequests {		t.Fatalf("status code = %d, want %d", got, http.StatusTooManyRequests)	}	retryAfter := streamErr.RetryAfter()	if retryAfter == nil {		t.Fatal("expected retryAfter from usage_limit_reached terminal error")	}	if *retryAfter != 300*time.Second {		t.Fatalf("retryAfter = %v, want %v", *retryAfter, 300*time.Second)	}} func TestCodexTerminalStreamErrHandlesUsageLimitResponseFailed(t *testing.T) {	streamErr, _, ok := codexTerminalStreamErr([]byte(`{"type":"response.failed","response":{"error":{"type":"usage_limit_reached","message":"usage limit reached","resets_in_seconds":60}}}`))	if !ok {		t.Fatal("expected usage_limit_reached response.failed terminal error to be handled")	}	if got := statusCodeFromTestError(t, streamErr); got != http.StatusTooManyRequests {		t.Fatalf("status code = %d, want %d", got, http.StatusTooManyRequests)	}	if streamErr.RetryAfter() == nil {		t.Fatal("expected retryAfter from usage_limit_reached response.failed terminal error")	}} func statusCodeFromTestError(t *testing.T, err error) int {	t.Helper() 	statusErr, ok := err.(interface{ StatusCode() int })	if !ok {		t.Fatalf("error %T does not expose StatusCode(): %v", err, err)	}	return statusErr.StatusCode()} func TestCodexExecutorExecuteStream_EmptyStreamCompletionOutputUsesOutputItemDone(t *testing.T) {	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {		w.Header().Set("Content-Type", "text/event-stream")		_, _ = w.Write([]byte("data: {\"type\":\"response.output_item.done\",\"item\":{\"type\":\"message\",\"role\":\"assistant\",\"content\":[{\"type\":\"output_text\",\"text\":\"ok\"}]},\"output_index\":0}\n"))		_, _ = w.Write([]byte("data: {\"type\":\"response.completed\",\"response\":{\"id\":\"resp_1\",\"object\":\"response\",\"created_at\":1775555723,\"status\":\"completed\",\"model\":\"gpt-5.4-mini-2026-03-17\",\"output\":[],\"usage\":{\"input_tokens\":8,\"output_tokens\":28,\"total_tokens\":36}}}\n\n"))	}))	defer server.Close() 	executor := NewCodexExecutor(&config.Config{})	auth := &cliproxyauth.Auth{Attributes: map[string]string{		"base_url": server.URL,		"api_key":  "test",	}} 	result, err := executor.ExecuteStream(context.Background(), auth, cliproxyexecutor.Request{		Model:   "gpt-5.4-mini",		Payload: []byte(`{"model":"gpt-5.4-mini","input":"Say ok"}`),	}, cliproxyexecutor.Options{		SourceFormat: sdktranslator.FromString("openai-response"),		Stream:       true,	})	if err != nil {		t.Fatalf("ExecuteStream error: %v", err)	} 	var completed []byte	for chunk := range result.Chunks {		if chunk.Err != nil {			t.Fatalf("stream chunk error: %v", chunk.Err)		}		payload := bytes.TrimSpace(chunk.Payload)		if !bytes.HasPrefix(payload, []byte("data:")) {			continue		}		data := bytes.TrimSpace(payload[5:])		if gjson.GetBytes(data, "type").String() == "response.completed" {			completed = append([]byte(nil), data...)		}	} 	if len(completed) == 0 {		t.Fatal("missing response.completed chunk")	} 	gotContent := gjson.GetBytes(completed, "response.output.0.content.0.text").String()	if gotContent != "ok" {		t.Fatalf("response.output[0].content[0].text = %q, want %q; completed=%s", gotContent, "ok", string(completed))	}}