badprofanity/profanity_test.go

29 lines
474 B
Go

package profanity
import (
"os"
"testing"
)
func init() {
}
func TestWordFilter(t *testing.T) {
filter := New()
filter.MustAddFile(os.Open("./data/wordlist.txt"))
cases := [][2]string{
{
"That wh0re gave m3 a very good H@nD j0b.",
"That ***** gave m3 a very good *********",
},
}
for _, tc := range cases {
filtered := filter.CensorSentenceToN(tc[0], "*", 6)
if filtered != tc[1] {
t.Errorf("expected '%s' to be '%s'", filtered, tc[1])
}
}
}