Function TestReplayCurrent_InferenceSuccess

internal/replay/replay_test.go:220–254

(t *testing.T)

}

func TestReplayCurrent_InferenceSuccess(t *testing.T) {
	cfg := replayTestConfig()
	st := requests.NewMemoryStore()
	req := types.AIRequest{
		TenantID: "team-a",
		TaskType: "chat",
		Priority: "high",
		Input:    map[string]any{"text": "replay hi"},
		Options:  types.RequestOptions{Stream: false, MaxTokens: 64},
	}
	full, _ := json.Marshal(req)
	_ = st.CreateRequest(context.Background(), requests.RequestRow{
		RequestID:     "rid3",
		TenantID:      req.TenantID,
		TaskType:      req.TaskType,
		Priority:      req.Priority,
		AIRequestJSON: full,
		CreatedAt:     time.Now(),
		UpdatedAt:     time.Now(),
	})
	ad := mock.New(cfg.Backends[0])
	deps := NewDependenciesFromConfig(cfg, map[string]interfaces.BackendAdapter{
		"small-model": ad,
	}, nil)
	resp, err := Replay(context.Background(), cfg, st, "rid3", ModeCurrent, deps)
	if err != nil {
		t.Fatalf("replay: %v", err)
	}
	if resp.Status != "success" {
		t.Fatalf("status=%q", resp.Status)
	}
	if resp.SelectedBackend != "small-model" {
		t.Fatalf("backend=%q", resp.SelectedBackend)
	}
}