package test import (	"context"	"strings"	"testing" 	_ "github.com/router-for-me/CLIProxyAPI/v7/internal/translator"	sdktranslator "github.com/router-for-me/CLIProxyAPI/v7/sdk/translator"	"github.com/tidwall/gjson") func TestCodexToClaudeParallelFunctionCallsHaveValidLifecycle(t *testing.T) {	chunks := [][]byte{		[]byte(`data: {"type":"response.created","response":{"id":"resp_parallel","model":"gpt-5"}}`),		[]byte(`data: {"type":"response.output_item.added","item":{"type":"function_call","call_id":"call_a","name":"Read"},"output_index":1}`),		[]byte(`data: {"type":"response.output_item.added","item":{"type":"function_call","call_id":"call_b","name":"Read"},"output_index":2}`),		[]byte(`data: {"type":"response.function_call_arguments.delta","delta":"{\"file_path\":\"a\"}","output_index":1}`),		[]byte(`data: {"type":"response.function_call_arguments.done","arguments":"{\"file_path\":\"a\"}","output_index":1}`),		[]byte(`data: {"type":"response.output_item.done","item":{"type":"function_call","call_id":"call_a","name":"Read","arguments":"{\"file_path\":\"a\"}"},"output_index":1}`),		[]byte(`data: {"type":"response.function_call_arguments.delta","delta":"{\"file_path\":\"b\"}","output_index":2}`),		[]byte(`data: {"type":"response.function_call_arguments.done","arguments":"{\"file_path\":\"b\"}","output_index":2}`),		[]byte(`data: {"type":"response.output_item.done","item":{"type":"function_call","call_id":"call_b","name":"Read","arguments":"{\"file_path\":\"b\"}"},"output_index":2}`),		[]byte(`data: {"type":"response.completed","response":{"usage":{"input_tokens":1,"output_tokens":1},"output":[{"type":"function_call","call_id":"call_a","name":"Read","arguments":"{\"file_path\":\"a\"}"},{"type":"function_call","call_id":"call_b","name":"Read","arguments":"{\"file_path\":\"b\"}"}]}}`),	} 	originalRequest := []byte(`{"stream":true,"tools":[{"name":"Read"}]}`)	var state any	open := make(map[int64]struct{})	started := make(map[int64]struct{})	toolIDs := make(map[int64]string)	arguments := make(map[int64]string)	var startIndices []int64	var stopIndices []int64	messageState := 0 	for _, chunk := range chunks {		outputs := sdktranslator.TranslateStream(			context.Background(),			sdktranslator.FormatCodex,			sdktranslator.FormatClaude,			"gpt-5",			originalRequest,			nil,			chunk,			&state,		)		for _, output := range outputs {			for _, line := range strings.Split(string(output), "\n") {				if !strings.HasPrefix(line, "data: ") {					continue				}				event := gjson.Parse(strings.TrimPrefix(line, "data: "))				if messageState == 2 {					t.Fatalf("event emitted after message_stop: %s", event.Raw)				}				index := event.Get("index").Int()				switch event.Get("type").String() {				case "content_block_start":					if messageState != 0 {						t.Fatalf("content block started after message terminal events: %s", event.Raw)					}					if len(open) != 0 {						t.Fatalf("content block start emitted while another block remains open: %v", open)					}					if _, exists := started[index]; exists {						t.Fatalf("content block index %d was reused", index)					}					open[index] = struct{}{}					started[index] = struct{}{}					startIndices = append(startIndices, index)					toolIDs[index] = event.Get("content_block.id").String()				case "content_block_delta":					if _, exists := open[index]; !exists {						t.Fatalf("content block delta targets unopened index %d", index)					}					if event.Get("delta.type").String() == "input_json_delta" {						arguments[index] += event.Get("delta.partial_json").String()					}				case "content_block_stop":					if _, exists := open[index]; !exists {						t.Fatalf("content block stop targets unopened index %d", index)					}					delete(open, index)					stopIndices = append(stopIndices, index)				case "message_delta":					if len(open) != 0 {						t.Fatalf("message_delta emitted while content blocks remain open: %v", open)					}					if messageState != 0 {						t.Fatalf("duplicate or out-of-order message_delta: %s", event.Raw)					}					messageState = 1				case "message_stop":					if len(open) != 0 {						t.Fatalf("message_stop emitted while content blocks remain open: %v", open)					}					if messageState != 1 {						t.Fatalf("message_stop emitted before message_delta: %s", event.Raw)					}					messageState = 2				}			}		}	} 	if len(open) != 0 {		t.Fatalf("content blocks remain open: %v", open)	}	if messageState != 2 {		t.Fatalf("terminal message event state = %d, want message_delta followed by message_stop", messageState)	}	if len(startIndices) != 2 || startIndices[0] != 0 || startIndices[1] != 1 {		t.Fatalf("start indices = %v, want [0 1]", startIndices)	}	if len(stopIndices) != 2 || stopIndices[0] != 0 || stopIndices[1] != 1 {		t.Fatalf("stop indices = %v, want [0 1]", stopIndices)	}	if toolIDs[0] != "call_a" || toolIDs[1] != "call_b" {		t.Fatalf("tool IDs = %v, want call_a and call_b", toolIDs)	}	if arguments[0] != `{"file_path":"a"}` || arguments[1] != `{"file_path":"b"}` {		t.Fatalf("tool arguments = %v", arguments)	}}