Function TestInitialPromptSent

lib/screentracker/pty_conversation_test.go:1361–1741

(t *testing.T)

}

func TestInitialPromptSent(t *testing.T) {
	discardLogger := slog.New(slog.NewTextHandler(io.Discard, nil))

	t.Run("initialPromptSent is set when initial prompt is sent", func(t *testing.T) {
		ctx, cancel := context.WithTimeout(context.Background(), testTimeout)
		t.Cleanup(cancel)

		tmpDir := t.TempDir()
		stateFile := tmpDir + "/state.json"

		mClock := quartz.NewMock(t)
		agent := &testAgent{screen: "loading..."}
		writeCounter := 0
		agent.onWrite = func(data []byte) {
			writeCounter++
			agent.screen = fmt.Sprintf("__write_%d", writeCounter)
		}

		cfg := st.PTYConversationConfig{
			Clock:                 mClock,
			SnapshotInterval:      1 * time.Second,
			ScreenStabilityLength: 0,
			AgentIO:               agent,
			ReadyForInitialPrompt: func(message string) bool {
				return message == "ready"
			},
			InitialPrompt: []st.MessagePart{st.MessagePartText{Content: "test prompt"}},
			Logger:        discardLogger,
			StatePersistenceConfig: st.StatePersistenceConfig{
				StateFile: stateFile,
				LoadState: false,
				SaveState: true,
			},
		}

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

		// Agent becomes ready and initial prompt is sent
		agent.setScreen("ready")
		advanceUntil(ctx, t, mClock, func() bool {
			return len(c.Messages()) >= 2
		})

		// Save state and verify initialPromptSent is persisted
		agent.setScreen("response")
		advanceFor(ctx, t, mClock, 2*time.Second)

		err := c.SaveState()
		require.NoError(t, err)

		data, err := os.ReadFile(stateFile)
		require.NoError(t, err)

		var agentState st.AgentState
		err = json.Unmarshal(data, &agentState)
		require.NoError(t, err)