36 lines
1.0 KiB
Go
36 lines
1.0 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/Tencent/WeKnora/internal/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestIsSearchableChunkSkipsUnsynchronizedEdits(t *testing.T) {
|
||
|
|
service := &knowledgeBaseService{}
|
||
|
|
for _, status := range []string{"processing", "failed"} {
|
||
|
|
chunk := &types.Chunk{ChunkType: types.ChunkTypeText, IndexStatus: status, IsEnabled: true}
|
||
|
|
if service.isSearchableChunk(chunk) {
|
||
|
|
t.Fatalf("chunk with index status %q should not be searchable", status)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for _, status := range []string{"", "ready"} {
|
||
|
|
chunk := &types.Chunk{ChunkType: types.ChunkTypeText, IndexStatus: status, IsEnabled: true}
|
||
|
|
if !service.isSearchableChunk(chunk) {
|
||
|
|
t.Fatalf("chunk with index status %q should be searchable", status)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIsSearchableChunkSkipsDisabledChunk(t *testing.T) {
|
||
|
|
service := &knowledgeBaseService{}
|
||
|
|
chunk := &types.Chunk{
|
||
|
|
ChunkType: types.ChunkTypeFAQ,
|
||
|
|
IndexStatus: "ready",
|
||
|
|
IsEnabled: false,
|
||
|
|
}
|
||
|
|
if service.isSearchableChunk(chunk) {
|
||
|
|
t.Fatal("disabled FAQ chunk should never be searchable")
|
||
|
|
}
|
||
|
|
}
|