package main

import (
 "strings"
 "testing"
)

func TestExtractTranslationResultIncludesStopReasonAndPreview(t *testing.T) {
 t.Parallel()

 raw := []byte(`{
  "type":"agent_end",
  "messages":[
   {
    "role":"assistant",
    "stopReason":"terminated",
    "content":[
     {"type":"text","text":"provider disconnected while streaming the translation chunk"}
    ]
   }
  ]
 }`)

 _, err := extractTranslationResult(raw)
 if err == nil {
  t.Fatal("expected error")
 }
 message := err.Error()
 for _, want := range []string{
  "pi error:",
  "stopReason=terminated",
  "assistant=provider disconnected while streaming the translation chunk",
 } {
  if !strings.Contains(message, want) {
   t.Fatalf("expected %q in error, got %q", want, message)
  }
 }
}

func TestPreviewPiAssistantTextTruncatesAndFlattensWhitespace(t *testing.T) {
 t.Parallel()

 input := "line one\n\nline   two\tline three " + strings.Repeat("x", 200)
 preview := previewPiAssistantText(input)
 if strings.Contains(preview, "\n") {
  t.Fatalf("expected flattened whitespace, got %q", preview)
 }
 if !strings.HasPrefix(preview, "line one line two line three ") {
  t.Fatalf("unexpected preview prefix: %q", preview)
 }
 if !strings.HasSuffix(preview, "...") {
  t.Fatalf("expected truncation suffix, got %q", preview)
 }
}

func TestExtractTranslationResultReturnsPiErrorBeforeDecodingStructuredErrorContent(t *testing.T) {
 t.Parallel()

 raw := []byte(`{
  "type":"agent_end",
  "messages":[
   {
    "role":"assistant",
    "stopReason":"terminated",
    "content":{"type":"error","message":"provider disconnected"}
   }
  ]
 }`)

 _, err := extractTranslationResult(raw)
 if err == nil {
  t.Fatal("expected error")
 }
 message := err.Error()
 if !strings.Contains(message, "pi error:") {
  t.Fatalf("expected normalized pi error, got %q", message)
 }
 if !strings.Contains(message, "stopReason=terminated") {
  t.Fatalf("expected stopReason in error, got %q", message)
 }
 if strings.Contains(message, "cannot unmarshal") {
  t.Fatalf("expected terminal pi error before decode failure, got %q", message)
 }
}