1 files changed, 81 insertions, 0 deletions
diff --git a/publisher/htmlElementsCollector_test.go b/publisher/htmlElementsCollector_test.go
new file mode 100644
index 000000000..3ef159d8b
--- /dev/null
+++ b/publisher/htmlElementsCollector_test.go
@@ -0,0 +1,81 @@
+// Copyright 2020 The Hugo Authors. All rights reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package publisher
+
+import (
+	"fmt"
+	"strings"
+	"testing"
+
+	qt "github.com/frankban/quicktest"
+)
+
+func TestClassCollector(t *testing.T) {
+	c := qt.New((t))
+
+	f := func(tags, classes, ids string) HTMLElements {
+		var tagss, classess, idss []string
+		if tags != "" {
+			tagss = strings.Split(tags, " ")
+		}
+		if classes != "" {
+			classess = strings.Split(classes, " ")
+		}
+		if ids != "" {
+			idss = strings.Split(ids, " ")
+		}
+		return HTMLElements{
+			Tags:    tagss,
+			Classes: classess,
+			IDs:     idss,
+		}
+	}
+
+	for _, test := range []struct {
+		name   string
+		html   string
+		expect HTMLElements
+	}{
+		{"basic", `<body class="b a"></body>`, f("body", "a b", "")},
+		{"duplicates", `<div class="b a b"></div>`, f("div", "a b", "")},
+		{"single quote", `<body class='b a'></body>`, f("body", "a b", "")},
+		{"no quote", `<body class=b id=myelement></body>`, f("body", "b", "myelement")},
+
+		{"AlpineJS bind 1", `<body>
+			<div x-bind:class="{
+        'class1': data.open,
+        'class2 class3': data.foo == 'bar'
+         }">
+			</div>
+		</body>`, f("body div", "class1 class2 class3", "")},
+
+		{"Alpine bind 2", `<div x-bind:class="{ 'bg-black':  filter.checked }"
+                        class="inline-block mr-1 mb-2 rounded  bg-gray-300 px-2 py-2">FOO</div>`,
+			f("div", "bg-black bg-gray-300 inline-block mb-2 mr-1 px-2 py-2 rounded", "")},
+
+		{"Alpine bind 3", `<div x-bind:class="{ 'text-gray-800':  !checked, 'text-white': checked }"></div>`, f("div", "text-gray-800 text-white", "")},
+		{"Alpine bind 4", `<div x-bind:class="{ 'text-gray-800':  !checked, 
+					 'text-white': checked }"></div>`, f("div", "text-gray-800 text-white", "")},
+
+		{"Vue bind", `<div v-bind:class="{ active: isActive }"></div>`, f("div", "active", "")},
+	} {
+		c.Run(test.name, func(c *qt.C) {
+			w := newHTMLElementsCollectorWriter(newHTMLElementsCollector())
+			fmt.Fprint(w, test.html)
+			got := w.collector.getHTMLElements()
+			c.Assert(got, qt.DeepEquals, test.expect)
+		})
+	}
+
+}