new-api/service/sensitive_test.go

package service

import (
	"testing"

	"github.com/QuantumNous/new-api/setting"
	"github.com/stretchr/testify/require"
)

func TestSensitiveWordReplaceMasksChineseWordAfterASCII(t *testing.T) {
	oldWords := setting.SensitiveWords
	setting.SensitiveWords = []string{"秘密"}
	t.Cleanup(func() { setting.SensitiveWords = oldWords })

	found, words, replaced := SensitiveWordReplace("prefix秘密suffix", false)

	require.True(t, found)
	require.Equal(t, []string{"秘密"}, words)
	require.Equal(t, "prefix**###**suffix", replaced)
}