1 files changed, 63 insertions, 0 deletions

diff --git a/vendor/github.com/tdewolff/parse/xml/util_test.go b/vendor/github.com/tdewolff/parse/xml/util_test.go
new file mode 100644
index 0000000..65be6b8
--- /dev/null
+++ b/vendor/github.com/tdewolff/parse/xml/util_test.go
@@ -0,0 +1,63 @@
+package xml // import "github.com/tdewolff/parse/xml"
+
+import (
+	"testing"
+
+	"github.com/tdewolff/test"
+)
+
+func TestEscapeAttrVal(t *testing.T) {
+	var attrValTests = []struct {
+		attrVal  string
+		expected string
+	}{
+		{"xyz", "\"xyz\""},
+		{"", "\"\""},
+		{"x&z", "\"x&z\""},
+		{"x'z", "\"x'z\""},
+		{"x\"z", "'x\"z'"},
+		{"a'b=\"\"", "'a'b=\"\"'"},
+		{"'x'\"'z'", "\"x'"'z\""},
+		{"\"x"'"z\"", "'x\"'\"z'"},
+		{"a'b=\"\"", "'a'b=\"\"'"},
+	}
+	var buf []byte
+	for _, tt := range attrValTests {
+		t.Run(tt.attrVal, func(t *testing.T) {
+			b := []byte(tt.attrVal)
+			if len(b) > 1 && (b[0] == '"' || b[0] == '\'') && b[0] == b[len(b)-1] {
+				b = b[1 : len(b)-1]
+			}
+			val := EscapeAttrVal(&buf, []byte(b))
+			test.String(t, string(val), tt.expected)
+		})
+	}
+}
+
+func TestEscapeCDATAVal(t *testing.T) {
+	var CDATAValTests = []struct {
+		CDATAVal string
+		expected string
+	}{
+		{"<![CDATA[<b>]]>", "&lt;b>"},
+		{"<![CDATA[abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz]]>", "abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz"},
+		{"<![CDATA[ <b> ]]>", " &lt;b> "},
+		{"<![CDATA[<<<<<]]>", "<![CDATA[<<<<<]]>"},
+		{"<![CDATA[&]]>", "&amp;"},
+		{"<![CDATA[&&&&]]>", "<![CDATA[&&&&]]>"},
+		{"<![CDATA[ a ]]>", " a "},
+		{"<![CDATA[]]>", ""},
+	}
+	var buf []byte
+	for _, tt := range CDATAValTests {
+		t.Run(tt.CDATAVal, func(t *testing.T) {
+			b := []byte(tt.CDATAVal[len("<![CDATA[") : len(tt.CDATAVal)-len("]]>")])
+			data, useText := EscapeCDATAVal(&buf, b)
+			text := string(data)
+			if !useText {
+				text = "<![CDATA[" + text + "]]>"
+			}
+			test.String(t, text, tt.expected)
+		})
+	}
+}