Function TestHandler_SaveAndLoadSession

server/handler_test.go:230–264

(t *testing.T)

}

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

	history := []models.Message{
		{Role: "user", Content: "hello"},
		{Role: "assistant", Content: "hi"},
	}

	store.On("SaveSession", "test", history).Return(nil)
	store.On("LoadSession", "test").Return(history, nil)

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

	// Save
	saveResp, err := handler.SaveSession(context.Background(), &pb.SaveSessionRequest{
		Name: "test",
		Messages: []*pb.ChatMessage{
			{Role: "user", Content: "hello"},
			{Role: "assistant", Content: "hi"},
		},
	})
	assert.NoError(t, err)
	assert.True(t, saveResp.Success)

	// Load
	loadResp, err := handler.LoadSession(context.Background(), &pb.LoadSessionRequest{Name: "test"})
	assert.NoError(t, err)
	assert.Len(t, loadResp.Messages, 2)
	assert.Equal(t, "user", loadResp.Messages[0].Role)
	assert.Equal(t, "hello", loadResp.Messages[0].Content)
}

func TestHandler_DeleteSession(t *testing.T) {
	logger := zap.NewNop()