// 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", ``, f("body", "a b", "")}, {"duplicates", `

`, f("div", "a b", "")}, {"single quote", ``, f("body", "a b", "")}, {"no quote", ``, f("body", "b", "myelement")}, {"thead", ` https://github.com/gohugoio/hugo/issues/7318

`, f("table tbody td thead tr", "cl1 cl2 cl3 cl4 cl5 cl6 cl7", "")}, // https://github.com/gohugoio/hugo/issues/7161 {"minified a href", ``, f("a", "a b", "")}, {"AlpineJS bind 1", `

`, f("body div", "class1 class2 class3", "")}, { "Alpine bind 2", `

FOO

`, f("div", "bg-black bg-gray-300 inline-block mb-2 mr-1 px-2 py-2 rounded", ""), }, {"Alpine bind 3", `

`, f("div", "text-gray-800 text-white", "")}, {"Alpine bind 4", `

`, f("div", "text-gray-800 text-white", "")}, {"Alpine bind 5", ``, f("a", "block capitalize cursor-pointer no-underline pl-2 pl-3 pr-3 text-a text-b text-gray-600 w-36", "")}, {"Alpine transition 1", `

`, f("div", "mobile:-translate-x-8 opacity-0 sm:-translate-y-8 transform", "")}, {"Vue bind", `

`, f("div", "active", "")}, // https://github.com/gohugoio/hugo/issues/7746 {"Apostrophe inside attribute value", `my text

`, f("a div", "missingclass", "")}, } { 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) }) } } func BenchmarkClassCollectorWriter(b *testing.B) { const benchHTML = `

` for i := 0; i < b.N; i++ { w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) fmt.Fprint(w, benchHTML) } }