Merge pull request #29 from phantom5099/fork-pr-337-1776410957

test: 提高 PR #337 覆盖率

Author: phantom5099

internal/session/input_preparer_test.go

@@ -308,6 +308,31 @@ func TestInputPreparerPrepareImagePathAndMimeValidation(t *testing.T) {
 	})
 }
 
+func TestAssetSaveErrorMethods(t *testing.T) {
+	t.Parallel()
+
+	if err := (*AssetSaveError)(nil).Unwrap(); err != nil {
+		t.Fatalf("expected nil asset save error unwrap to return nil, got %v", err)
+	}
+	if msg := (*AssetSaveError)(nil).Error(); msg != "session: asset save failed" {
+		t.Fatalf("unexpected nil asset save error message: %q", msg)
+	}
+
+	inner := errors.New("boom")
+	assetErr := &AssetSaveError{
+		SessionID: "session-1",
+		Index:     2,
+		Path:      "/tmp/image.png",
+		Err:       inner,
+	}
+	if !errors.Is(assetErr, inner) {
+		t.Fatalf("expected asset save error to unwrap inner error")
+	}
+	if !strings.Contains(assetErr.Error(), "image.png") || !strings.Contains(assetErr.Error(), "index 2") {
+		t.Fatalf("unexpected asset save error message: %q", assetErr.Error())
+	}
+}
+
 func minimalPNGBytes() []byte {
 	return []byte{
 		0x89, 0x50, 0x4e, 0x47, 0x0d, 0x0a, 0x1a, 0x0a,

internal/tui/core/app/input_features_test.go

@@ -301,6 +301,53 @@ func TestAbsorbInlineImageReferencesSupportsQuotedPathWithSpaces(t *testing.T) {
 	}
 }
 
+func TestParseInlineImagePathToken(t *testing.T) {
+	app, _ := newTestApp(t)
+	root := t.TempDir()
+	app.state.CurrentWorkdir = root
+
+	relative, ok := app.parseInlineImagePathToken(`@image:"charts/sales q1.png"`)
+	if !ok {
+		t.Fatalf("expected quoted relative token to parse")
+	}
+	if relative != filepath.Join(root, filepath.FromSlash("charts/sales q1.png")) {
+		t.Fatalf("unexpected resolved path: %q", relative)
+	}
+
+	absolutePath := filepath.Join(root, "abs.png")
+	absolute, ok := app.parseInlineImagePathToken("@image:" + absolutePath)
+	if !ok || absolute != absolutePath {
+		t.Fatalf("expected absolute token to pass through, got %q ok=%v", absolute, ok)
+	}
+
+	if _, ok := app.parseInlineImagePathToken("@image:notes.txt"); ok {
+		t.Fatalf("expected non-image token to be rejected")
+	}
+	app.state.CurrentWorkdir = ""
+	if _, ok := app.parseInlineImagePathToken("@image:relative.png"); ok {
+		t.Fatalf("expected missing workdir to reject relative token")
+	}
+	if _, ok := app.parseInlineImagePathToken("not-image-token"); ok {
+		t.Fatalf("expected invalid token to be rejected")
+	}
+}
+
+func TestParseInlineImageReferenceAtBranches(t *testing.T) {
+	if _, _, ok := parseInlineImageReferenceAt("x@image:a.png", 1); ok {
+		t.Fatalf("expected token without boundary whitespace to be rejected")
+	}
+	path, end, ok := parseInlineImageReferenceAt(`@image:folder\ with\ space.png next`, 0)
+	if !ok {
+		t.Fatalf("expected escaped-space token to parse")
+	}
+	if path != "folder with space.png" || end <= 0 {
+		t.Fatalf("unexpected escaped path parse result path=%q end=%d", path, end)
+	}
+	if _, _, ok := parseInlineImageReferenceAt(`@image:""`, 0); ok {
+		t.Fatalf("expected empty quoted token to fail")
+	}
+}
+
 func TestGetAndClearImageAttachments(t *testing.T) {
 	app, _ := newTestApp(t)
 	app.pendingImageAttachments = []pendingImageAttachment{

internal/tui/core/app/update_runtime_events_test.go

@@ -1,6 +1,7 @@
 package tui
 
 import (
+	"strings"
 	"testing"
 
 	agentruntime "neo-code/internal/runtime"
@@ -161,3 +162,92 @@ func TestShouldHandleRuntimeEventFiltersBySessionAndRun(t *testing.T) {
 		t.Fatalf("expected matched event to be handled")
 	}
 }
+
+func TestRuntimeEventMultimodalHandlers(t *testing.T) {
+	t.Parallel()
+
+	app, _ := newTestApp(t)
+
+	if handled := runtimeEventInputNormalizedHandler(&app, agentruntime.RuntimeEvent{Payload: "bad"}); handled {
+		t.Fatalf("expected invalid normalized payload to return false")
+	}
+	runtimeEventInputNormalizedHandler(&app, agentruntime.RuntimeEvent{
+		RunID: "run-1",
+		Payload: agentruntime.InputNormalizedPayload{
+			TextLength: 12,
+			ImageCount: 2,
+		},
+	})
+	if app.state.ActiveRunID != "run-1" {
+		t.Fatalf("expected active run id to be updated, got %q", app.state.ActiveRunID)
+	}
+	if len(app.activities) == 0 {
+		t.Fatalf("expected input normalized activity to be appended")
+	}
+	last := app.activities[len(app.activities)-1]
+	if last.Title != "Input normalized" || !strings.Contains(last.Detail, "images=2") {
+		t.Fatalf("unexpected normalized activity: %+v", last)
+	}
+
+	before := len(app.activities)
+	runtimeEventAssetSavedHandler(&app, agentruntime.RuntimeEvent{
+		Payload: agentruntime.AssetSavedPayload{
+			AssetID: "asset-1",
+			Path:    "/tmp/chart.png",
+		},
+	})
+	if len(app.activities) != before+1 {
+		t.Fatalf("expected saved attachment activity appended")
+	}
+	last = app.activities[len(app.activities)-1]
+	if last.Title != "Saved attachment" || !strings.Contains(last.Detail, "chart.png") {
+		t.Fatalf("unexpected asset saved activity: %+v", last)
+	}
+	if handled := runtimeEventAssetSavedHandler(&app, agentruntime.RuntimeEvent{Payload: 123}); handled {
+		t.Fatalf("expected invalid asset_saved payload to return false")
+	}
+
+	runtimeEventAssetSaveFailedHandler(&app, agentruntime.RuntimeEvent{
+		Payload: agentruntime.AssetSaveFailedPayload{Message: " failed "},
+	})
+	if app.state.ExecutionError != "failed" || app.state.StatusText != "failed" {
+		t.Fatalf("expected failed status to be surfaced, got status=%q err=%q", app.state.StatusText, app.state.ExecutionError)
+	}
+	last = app.activities[len(app.activities)-1]
+	if !last.IsError || last.Title != "Failed to save attachment" {
+		t.Fatalf("unexpected asset save failed activity: %+v", last)
+	}
+	runtimeEventAssetSaveFailedHandler(&app, agentruntime.RuntimeEvent{
+		Payload: agentruntime.AssetSaveFailedPayload{},
+	})
+	if app.state.ExecutionError != "failed to save attachment" || app.state.StatusText != "failed to save attachment" {
+		t.Fatalf("expected default failed message, got status=%q err=%q", app.state.StatusText, app.state.ExecutionError)
+	}
+	if handled := runtimeEventAssetSaveFailedHandler(&app, agentruntime.RuntimeEvent{Payload: true}); handled {
+		t.Fatalf("expected invalid asset_save_failed payload to return false")
+	}
+}
+
+func TestHandleRuntimeEventSetsSessionAndRoutesByRegistry(t *testing.T) {
+	t.Parallel()
+
+	app, _ := newTestApp(t)
+	handled := app.handleRuntimeEvent(agentruntime.RuntimeEvent{
+		Type:      agentruntime.EventAssetSaved,
+		SessionID: "session-1",
+		Payload:   agentruntime.AssetSavedPayload{AssetID: "asset-1"},
+	})
+	if handled {
+		t.Fatalf("expected asset_saved handler to return false")
+	}
+	if app.state.ActiveSessionID != "session-1" {
+		t.Fatalf("expected active session to be set from event, got %q", app.state.ActiveSessionID)
+	}
+	if len(app.activities) == 0 || app.activities[len(app.activities)-1].Title != "Saved attachment" {
+		t.Fatalf("expected saved attachment activity")
+	}
+
+	if app.handleRuntimeEvent(agentruntime.RuntimeEvent{Type: "unknown_event", SessionID: "session-1"}) {
+		t.Fatalf("expected unknown event handler result to be false")
+	}
+}