Function TestHandler_LoadSession_V2

server/handler_test.go:523–561

(t *testing.T)

}

func TestHandler_LoadSession_V2(t *testing.T) {
	logger := zap.NewNop()
	store := &mockSessionStoreV2{}

	sd := &models.SessionData{
		Version: 2,
		ChatHistory: []models.Message{
			{Role: "user", Content: "hello"},
			{Role: "assistant", Content: "hi"},
		},
		AgentHistory: []models.Message{
			{Role: "user", Content: "agent task"},
		},
		CoderHistory: []models.Message{
			{Role: "user", Content: "coder task"},
		},
	}

	store.On("LoadSessionV2", "test-v2").Return(sd, nil)

	handler := &Handler{
		sessionManager: store,
		logger:         logger,
	}

	resp, err := handler.LoadSession(context.Background(), &pb.LoadSessionRequest{Name: "test-v2"})
	assert.NoError(t, err)

	// V1 compat: messages field should contain chat history
	assert.Len(t, resp.Messages, 2)
	assert.Equal(t, "user", resp.Messages[0].Role)

	// V2: session_data should contain all scoped histories
	assert.NotNil(t, resp.SessionData)
	assert.Equal(t, int32(2), resp.SessionData.Version)
	assert.Len(t, resp.SessionData.ChatHistory, 2)
	assert.Len(t, resp.SessionData.AgentHistory, 1)
	assert.Len(t, resp.SessionData.CoderHistory, 1)
}

func TestHandler_SaveSession_V1Fallback(t *testing.T) {
	// Store does NOT implement SessionStoreV2 — should fall back to v1