package loop

import (
	"context"
	"testing"

	"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/model"
)

func TestRun_ImageResolverFoldsToText(t *testing.T) {
	t.Parallel()
	ad := &mockAdapter{chat: []mgResp{{resp: model.Response{Model: "m", Content: "Done."}}}}
	a := newTestAgent(t, ad, nil)
	l := New(Options{
		Agent:     a,
		MaxRounds: 8,
		Vision: ImageResolveFunc(func(_ context.Context, _ model.Message) (model.Message, error) {
			return model.TextMessage(model.RoleUser, "图中是一只猫"), nil
		}),
	})

	in := []model.Message{model.ImageMessage(model.RoleUser, "", "data:image/png;base64,AAAA")}
	if _, err := l.Run(context.Background(), "sess-1", in); err != nil {
		t.Fatalf("Run: %v", err)
	}
	if len(ad.reqs) == 0 {
		t.Fatalf("no model request recorded")
	}
	var userSeen bool
	for _, m := range ad.reqs[0].Messages {
		if m.Role == model.RoleUser {
			userSeen = true
			if messageHasImage(m) {
				t.Fatalf("image still reached the model: %+v", m)
			}
		}
	}
	if !userSeen {
		t.Fatalf("no user message seen by the model")
	}
}

func TestRun_ImageKeptWithoutResolver(t *testing.T) {
	t.Parallel()
	ad := &mockAdapter{chat: []mgResp{{resp: model.Response{Model: "m", Content: "Done."}}}}
	a := newTestAgent(t, ad, nil)
	l := New(Options{Agent: a, MaxRounds: 8})

	in := []model.Message{model.ImageMessage(model.RoleUser, "look", "data:image/png;base64,AAAA")}
	if _, err := l.Run(context.Background(), "sess-1", in); err != nil {
		t.Fatalf("Run: %v", err)
	}
	if len(ad.reqs) == 0 {
		t.Fatalf("no model request recorded")
	}
	var sawImage bool
	for _, m := range ad.reqs[0].Messages {
		if m.Role == model.RoleUser && messageHasImage(m) {
			sawImage = true
		}
	}
	if !sawImage {
		t.Fatalf("expected the image part to pass through without a resolver")
	}
}