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]) } } }