diff --git a/morty_test.go b/morty_test.go
index 316e7e7..dee5e56 100644
--- a/morty_test.go
+++ b/morty_test.go
@@ -37,10 +37,10 @@ var attrTestData []*AttrTestCase = []*AttrTestCase{
func TestAttrSanitizer(t *testing.T) {
u, _ := url.Parse("http://127.0.0.1/")
- rc := &RequestConfig{nil, u}
+ rc := &RequestConfig{BaseURL: u}
for _, testCase := range attrTestData {
out := bytes.NewBuffer(nil)
- sanitizeAttr(rc, out, testCase.AttrName, testCase.AttrValue)
+ sanitizeAttr(rc, out, testCase.AttrName, testCase.AttrValue, testCase.AttrValue)
res, _ := out.ReadBytes(byte(0))
if !bytes.Equal(res, testCase.ExpectedOutput) {
t.Errorf(
@@ -53,3 +53,53 @@ func TestAttrSanitizer(t *testing.T) {
}
}
}
+
+var BENCH_SIMPLE_HTML []byte = []byte(`
+
+
+ test
+
+
+ Test heading
+
+`)
+
+func BenchmarkSanitizeSimpleHTML(b *testing.B) {
+ u, _ := url.Parse("http://127.0.0.1/")
+ rc := &RequestConfig{BaseURL: u}
+ b.ResetTimer()
+ for i := 0; i < b.N; i++ {
+ out := bytes.NewBuffer(nil)
+ sanitizeHTML(rc, out, BENCH_SIMPLE_HTML)
+ }
+}
+
+var BENCH_COMPLEX_HTML []byte = []byte(`
+
+
+
+ test 2
+
+
+
+
+ Test heading
+ 
+
+
+`)
+
+func BenchmarkSanitizeComplexHTML(b *testing.B) {
+ u, _ := url.Parse("http://127.0.0.1/")
+ rc := &RequestConfig{BaseURL: u}
+ b.ResetTimer()
+ for i := 0; i < b.N; i++ {
+ out := bytes.NewBuffer(nil)
+ sanitizeHTML(rc, out, BENCH_COMPLEX_HTML)
+ }
+}