mirror of
https://github.com/russross/blackfriday.git
synced 2024-03-22 13:40:34 +08:00
461803619b
Simplify and optimize escapeHTML as per @Ambrevar's suggestion: lean on the fact that we're dealing with bytes: declare a 256-element array with most of it filled with zeros, except for the few slots that need escaping. This avoids some conditionals in a tight loop. Also, uncomment it's benchmark.
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package blackfriday
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
)
|
|
|
|
func TestEsc(t *testing.T) {
|
|
tests := []string{
|
|
"abc", "abc",
|
|
"a&c", "a&c",
|
|
"<", "<",
|
|
"[]:<", "[]:<",
|
|
"Hello <!--", "Hello <!--",
|
|
}
|
|
for i := 0; i < len(tests); i += 2 {
|
|
var b bytes.Buffer
|
|
escapeHTML(&b, []byte(tests[i]))
|
|
if !bytes.Equal(b.Bytes(), []byte(tests[i+1])) {
|
|
t.Errorf("\nInput [%#v]\nExpected[%#v]\nActual [%#v]",
|
|
tests[i], tests[i+1], b.String())
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkEscapeHTML(b *testing.B) {
|
|
tests := [][]byte{
|
|
[]byte(""),
|
|
[]byte("AT&T has an ampersand in their name."),
|
|
[]byte("AT&T is another way to write it."),
|
|
[]byte("This & that."),
|
|
[]byte("4 < 5."),
|
|
[]byte("6 > 5."),
|
|
[]byte("Here's a [link] [1] with an ampersand in the URL."),
|
|
[]byte("Here's a link with an ampersand in the link text: [AT&T] [2]."),
|
|
[]byte("Here's an inline [link](/script?foo=1&bar=2)."),
|
|
[]byte("Here's an inline [link](</script?foo=1&bar=2>)."),
|
|
[]byte("[1]: http://example.com/?foo=1&bar=2"),
|
|
[]byte("[2]: http://att.com/ \"AT&T\""),
|
|
}
|
|
var buff bytes.Buffer
|
|
for n := 0; n < b.N; n++ {
|
|
for _, t := range tests {
|
|
escapeHTML(&buff, t)
|
|
buff.Reset()
|
|
}
|
|
}
|
|
}
|