Function TestHandler_SaveSession_V1Fallback

server/handler_test.go:563–591

(t *testing.T)

}

func TestHandler_SaveSession_V1Fallback(t *testing.T) {
	// Store does NOT implement SessionStoreV2 — should fall back to v1
	logger := zap.NewNop()
	store := &mockSessionStore{}

	history := []models.Message{
		{Role: "user", Content: "hello"},
	}
	store.On("SaveSession", "test-v1", history).Return(nil)

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

	// Even with session_data present, v1 store should use messages field
	resp, err := handler.SaveSession(context.Background(), &pb.SaveSessionRequest{
		Name:     "test-v1",
		Messages: []*pb.ChatMessage{{Role: "user", Content: "hello"}},
		SessionData: &pb.SessionDataV2{
			Version:      2,
			ChatHistory:  []*pb.ChatMessage{{Role: "user", Content: "hello"}},
			AgentHistory: []*pb.ChatMessage{{Role: "user", Content: "agent"}},
		},
	})
	assert.NoError(t, err)
	assert.True(t, resp.Success)
	store.AssertCalled(t, "SaveSession", "test-v1", history)
}

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