Function TestChatNormalFlow

quickstart/chatwitheino/server/server_test.go:167–246

(t *testing.T)

}

func TestChatNormalFlow(t *testing.T) {
	agent := simpleReplyAgent("test response")
	srv, _, cleanup := newTestServer(t, agent)
	defer cleanup()

	sessionID := createSession(t, srv)
	sess, _ := srv.cfg.Store.GetOrCreate(sessionID)

	// Simulate handleChat: create loop, push item, run, consume events.
	ts := srv.getTurnState(sessionID)
	ts.mu.Lock()
	loop := srv.newLoop(sess, sessionID, false)
	ts.loop = loop
	ts.iterReady = make(chan iterEnvelope[*schema.Message], 1)
	ts.iterDone = make(chan iterResult[*schema.Message], 1)
	ts.mu.Unlock()

	item := &ChatItem{Query: "hello"}
	if err := sess.Append(schema.UserMessage("hello")); err != nil {
		t.Fatalf("append: %v", err)
	}
	loop.Push(item)
	loop.Run(context.Background())

	// Wait for the iterator.
	var envelope iterEnvelope[*schema.Message]
	select {
	case envelope = <-ts.iterReady:
	case <-time.After(5 * time.Second):
		t.Fatal("timeout waiting for iterReady")
	}

	// Drain the iterator.
	var content string
	for {
		event, ok := envelope.events.Next()
		if !ok {
			break
		}
		if event.Output != nil && event.Output.MessageOutput != nil {
			if m := event.Output.MessageOutput.Message; m != nil && m.Content != "" {
				content += m.Content
			}
		}
	}

	if content != "test response" {
		t.Errorf("expected 'test response', got %q", content)
	}

	// Signal done to OnAgentEvents.
	envelope.done <- iterResult[*schema.Message]{
		lastContent:   content,
		intermediates: []*schema.Message{schema.AssistantMessage(content, nil)},
	}

	// Stop the loop — no more items to process.
	loop.Stop()