Function TestMessages

lib/screentracker/pty_conversation_test.go:215–571

(t *testing.T)

}

func TestMessages(t *testing.T) {
	now := time.Date(2025, 1, 1, 0, 0, 0, 0, time.UTC)

	// newConversation creates a started conversation with a mock clock and
	// testAgent. Tests that Send() messages must use sendAndAdvance.
	newConversation := func(ctx context.Context, t *testing.T, opts ...func(*st.PTYConversationConfig)) (*st.PTYConversation, *testAgent, *quartz.Mock) {
		t.Helper()

		writeCounter := 0
		agent := &testAgent{}
		// Default onWrite: each write produces a unique screen so that
		// writeStabilize can detect screen changes.
		agent.onWrite = func(data []byte) {
			writeCounter++
			agent.screen = fmt.Sprintf("__write_%d", writeCounter)
		}
		mClock := quartz.NewMock(t)
		mClock.Set(now)
		cfg := st.PTYConversationConfig{
			Clock:                 mClock,
			AgentIO:               agent,
			SnapshotInterval:      100 * time.Millisecond,
			ScreenStabilityLength: 200 * time.Millisecond,
			Logger:                slog.New(slog.NewTextHandler(io.Discard, nil)),
		}
		for _, opt := range opts {
			opt(&cfg)
		}
		if a, ok := cfg.AgentIO.(*testAgent); ok {
			agent = a
		}

		c := st.NewPTY(ctx, cfg, &testEmitter{})
		c.Start(ctx)

		return c, agent, mClock
	}

	// threshold = 3 (200ms / 100ms = 2, + 1 = 3)
	const threshold = 3
	const interval = 100 * time.Millisecond

	t.Run("messages are copied", func(t *testing.T) {
		c, _, _ := newConversation(context.Background(), t)
		messages := c.Messages()
		assertMessages(t, c, []st.ConversationMessage{
			{Id: 0, Message: "", Role: st.ConversationRoleAgent},
		})

		messages[0].Message = "modification"

		assertMessages(t, c, []st.ConversationMessage{
			{Id: 0, Message: "", Role: st.ConversationRoleAgent},
		})
	})

	t.Run("whitespace-padding", func(t *testing.T) {
		c, _, _ := newConversation(context.Background(), t)