71 lines
3.3 KiB
Go
71 lines
3.3 KiB
Go
package search
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"github.com/stretchr/testify/require"
|
|
"gorm.io/gorm"
|
|
"senlinai-agent/backend/internal/models"
|
|
)
|
|
|
|
func TestSearchFindsNoteBody(t *testing.T) {
|
|
database, err := gorm.Open(sqlite.Open(fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, models.AutoMigrate(database))
|
|
models.DBService = database
|
|
require.NoError(t, database.Create(&models.SenlinAgentProject{ID: 1, OwnerID: 7, Name: "支付项目"}).Error)
|
|
require.NoError(t, database.Create(&models.SenlinAgentNote{ProjectID: 1, CreatedBy: 7, Title: "接口方案", Markdown: "二维码支付回调设计"}).Error)
|
|
|
|
service := NewService()
|
|
results, err := service.Search(7, "回调")
|
|
|
|
require.NoError(t, err)
|
|
require.Len(t, results, 1)
|
|
require.Equal(t, "note", results[0].Type)
|
|
}
|
|
|
|
func TestSearchOnlyReturnsNavigableProjectTaskAndNoteResults(t *testing.T) {
|
|
database, err := gorm.Open(sqlite.Open(fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, models.AutoMigrate(database))
|
|
models.DBService = database
|
|
require.NoError(t, database.Create(&models.SenlinAgentProject{ID: 1, OwnerID: 7, Name: "webhook 支付项目"}).Error)
|
|
require.NoError(t, database.Create(&models.SenlinAgentTask{ProjectID: 1, CreatedBy: 7, Title: "回调任务", Description: "检查 webhook"}).Error)
|
|
require.NoError(t, database.Create(&models.SenlinAgentNote{ProjectID: 1, CreatedBy: 7, Title: "回调笔记", Markdown: "webhook 流程"}).Error)
|
|
require.NoError(t, database.Create(&models.SenlinAgentSource{ProjectID: 1, CreatedBy: 7, Kind: "link", Title: "支付文档", URL: "https://example.com/pay", ContentText: "webhook 签名"}).Error)
|
|
require.NoError(t, database.Create(&models.SenlinAgentInboxItem{ProjectID: 1, CreatedBy: 7, SourceType: "text", Title: "收集项", Body: "webhook 待整理"}).Error)
|
|
require.NoError(t, database.Create(&models.SenlinAgentAISession{ProjectID: 1, CreatedBy: 7, Title: "AI 分析", Context: "webhook 问答"}).Error)
|
|
|
|
service := NewService()
|
|
results, err := service.Search(7, "webhook")
|
|
|
|
require.NoError(t, err)
|
|
types := make(map[string]bool)
|
|
for _, result := range results {
|
|
types[result.Type] = true
|
|
}
|
|
require.Equal(t, map[string]bool{"project": true, "task": true, "note": true}, types)
|
|
require.Len(t, results, 3)
|
|
}
|
|
|
|
func TestSearchTreatsLikeWildcardsAsLiteralKeywords(t *testing.T) {
|
|
database, err := gorm.Open(sqlite.Open(fmt.Sprintf("file:%s?mode=memory&cache=shared", t.Name())), &gorm.Config{})
|
|
require.NoError(t, err)
|
|
require.NoError(t, models.AutoMigrate(database))
|
|
models.DBService = database
|
|
require.NoError(t, database.Create(&models.SenlinAgentProject{ID: 1, OwnerID: 7, Name: "100%_完成", Identifier: "SPECIAL"}).Error)
|
|
require.NoError(t, database.Create(&models.SenlinAgentProject{ID: 2, OwnerID: 7, Name: "普通项目", Identifier: "NORMAL"}).Error)
|
|
|
|
percentResults, err := NewService().Search(7, "%")
|
|
require.NoError(t, err)
|
|
require.Len(t, percentResults, 1)
|
|
require.Equal(t, "100%_完成", percentResults[0].Title)
|
|
|
|
underscoreResults, err := NewService().Search(7, "_")
|
|
require.NoError(t, err)
|
|
require.Len(t, underscoreResults, 1)
|
|
require.Equal(t, "100%_完成", underscoreResults[0].Title)
|
|
}
|