Function TestStatePersistence

lib/screentracker/pty_conversation_test.go:573–1194

(t *testing.T)

}

func TestStatePersistence(t *testing.T) {
	t.Run("SaveState creates file with correct structure", func(t *testing.T) {
		ctx, cancel := context.WithTimeout(context.Background(), testTimeout)
		t.Cleanup(cancel)

		// Create temp directory for state file
		tmpDir := t.TempDir()
		stateFile := tmpDir + "/state.json"

		mClock := quartz.NewMock(t)
		agent := &testAgent{screen: "initial"}
		cfg := st.PTYConversationConfig{
			Clock:                 mClock,
			SnapshotInterval:      100 * time.Millisecond,
			ScreenStabilityLength: 200 * time.Millisecond,
			AgentIO:               agent,
			Logger:                slog.New(slog.NewTextHandler(io.Discard, nil)),
			StatePersistenceConfig: st.StatePersistenceConfig{
				StateFile: stateFile,
				LoadState: false,
				SaveState: true,
			},
			InitialPrompt: []st.MessagePart{st.MessagePartText{Content: "test prompt"}},
		}

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

		// Generate some conversation
		agent.setScreen("hello")
		advanceFor(ctx, t, mClock, 300*time.Millisecond)

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

		// Read and verify the saved file
		data, err := os.ReadFile(stateFile)
		require.NoError(t, err)

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

		assert.Equal(t, 1, agentState.Version)
		assert.Equal(t, "test prompt", agentState.InitialPrompt)
		assert.NotEmpty(t, agentState.Messages)
	})

	t.Run("SaveState creates valid JSON", 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)
		fixedTime := time.Date(2025, 1, 1, 0, 0, 0, 0, time.UTC)