37 lines
1.8 KiB
Go
37 lines
1.8 KiB
Go
|
package service
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/Tencent/WeKnora/internal/types"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWikiCitationDeclarationOptionalDetailsRetainsEvidenceRequirements(t *testing.T) {
|
||
|
|
batch := splitChunksIntoCitationBatches([]*types.Chunk{
|
||
|
|
{ID: "first", Content: "Alpha requires condition X."},
|
||
|
|
{ID: "second", Content: "Alpha excludes condition Y.", ChunkIndex: 1},
|
||
|
|
})[0]
|
||
|
|
original := citationBatchResult{Citations: map[string][]string{"concept/alpha": {"first", "second"}}}
|
||
|
|
missing := map[string][]string{"concept/alpha": {"c000", "c001"}}
|
||
|
|
raw := `{"cannot_repair":false,"new_slugs":[{"type":"concept","slug":"concept/alpha","name":"Alpha","aliases":[],"description":"Alpha conditions","details":"","source_chunks":["c000","c001"],"evidence_quotes":[{"chunk":"c000","quote":"Alpha requires condition X."},{"chunk":"c001","quote":"Alpha excludes condition Y."}]}]}`
|
||
|
|
got, err := applyWikiCitationDeclarations(raw, original, missing, batch, nil)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Len(t, got.NewSlugs, 1)
|
||
|
|
require.Empty(t, got.NewSlugs[0].Details)
|
||
|
|
require.Equal(t, original.Citations, got.Citations)
|
||
|
|
require.ElementsMatch(t, []string{"first", "second"}, got.NewSlugs[0].SourceChunks)
|
||
|
|
for name, invalid := range map[string]string{
|
||
|
|
"missing restriction source": strings.Replace(raw, `"source_chunks":["c000","c001"]`, `"source_chunks":["c000"]`, 1),
|
||
|
|
"invented restriction quote": strings.Replace(raw, `"quote":"Alpha excludes condition Y."`, `"quote":"Alpha permits condition Y."`, 1),
|
||
|
|
"missing identity": strings.Replace(raw, `"name":"Alpha"`, `"name":""`, 1),
|
||
|
|
} {
|
||
|
|
t.Run(name, func(t *testing.T) {
|
||
|
|
_, err := applyWikiCitationDeclarations(invalid, original, missing, batch, nil)
|
||
|
|
require.Error(t, err)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
require.Empty(t, original.NewSlugs)
|
||
|
|
}
|