// 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 ( "bytes" "fmt" "io" "math/rand" "strings" "testing" "time" "github.com/gohugoio/hugo/config/testconfig" "github.com/gohugoio/hugo/media" "github.com/gohugoio/hugo/minifiers" "github.com/gohugoio/hugo/output" qt "github.com/frankban/quicktest" ) func TestClassCollector(t *testing.T) { c := qt.New((t)) rnd := rand.New(rand.NewSource(time.Now().Unix())) 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, } } skipMinifyTest := map[string]bool{ "Script tags content should be skipped": true, // https://github.com/tdewolff/minify/issues/396 } for _, test := range []struct { name string html string expect HTMLElements }{ {"basic", ``, f("body", "a b", "")}, {"duplicates", `
x'`, f("div", "a b", "")}, {"single quote", ``, f("body", "a b", "")}, {"no quote", ``, f("body", "b", "myelement")}, {"short", ``, f("i", "", "")}, {"invalid", `< body class="b a">
`, f("div", "", "")}, // https://github.com/gohugoio/hugo/issues/7318 {"thead", `
`, f("table tbody td thead tr", "cl1 cl2 cl3 cl4 cl5 cl6 cl7", "")}, {"thead uppercase", `
`, 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", "")}, {"AlpineJS bind 2", `
FOO
`, f("div", "bg-black bg-gray-300 inline-block mb-2 mr-1 px-2 py-2 rounded", ""), }, {"AlpineJS bind 3", `
`, f("div", "text-gray-800 text-white", "")}, {"AlpineJS bind 4", `
`, f("div", "text-gray-800 text-white", "")}, {"AlpineJS 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", "")}, {"AlpineJS transition 1", `
`, f("div", "mobile:-translate-x-8 opacity-0 sm:-translate-y-8 transform", "")}, {"Vue bind", `
`, f("div", "active", "")}, // Issue #7746 {"Apostrophe inside attribute value", `my text
`, f("a div", "missingclass", "")}, // Issue #7567 {"Script tags content should be skipped", `
`, f("div script", "foo", "")}, {"Style tags content should be skipped", `
`, f("div style", "foo", "")}, {"Pre tags content should be skipped", `
foobar
`, f("div pre", "foo preclass", "")}, {"Textarea tags content should be skipped", `
`, f("div textarea", "foo textareaclass", "")}, {"DOCTYPE should beskipped", ``, f("", "", "")}, {"Comments should be skipped", ``, f("", "", "")}, {"Comments with elements before and after", `
`, f("div span", "", "")}, {"Self closing tag", `

`, f("div hr", "", "")}, // svg with self closing style tag. {"SVG with self closing style tag", ``, f("g path style svg", "foo", "")}, // Issue #8530 {"Comment with single quote", ``, f("i", "foo", "")}, {"Uppercase tags", `
`, f("div", "", "")}, {"Predefined tags with distinct casing", `
`, f("div script", "", "")}, // Issue #8417 {"Tabs inline", `
d
`, f("div hr", "bar foo", "a")}, {"Tabs on multiple rows", `
d
`, f("div form", "foo", "a b")}, {"Big input, multibyte runes", strings.Repeat(`神真美好 `, rnd.Intn(500)+1) + "
" + strings.Repeat(`神真美好 `, rnd.Intn(100)+1) + " 神真美好", f("div span", "foo", "神真美好")}, } { for _, variant := range []struct { minify bool }{ {minify: false}, {minify: true}, } { c.Run(fmt.Sprintf("%s--minify-%t", test.name, variant.minify), func(c *qt.C) { w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) if variant.minify { if skipMinifyTest[test.name] { c.Skip("skip minify test") } m, _ := minifiers.New(media.DefaultTypes, output.DefaultFormats, testconfig.GetTestConfig(nil, nil)) m.Minify(media.Builtin.HTMLType, w, strings.NewReader(test.html)) } else { var buff bytes.Buffer buff.WriteString(test.html) io.Copy(w, &buff) } got := w.collector.getHTMLElements() c.Assert(got, qt.DeepEquals, test.expect) }) } } } func TestEndsWithTag(t *testing.T) { c := qt.New((t)) for _, test := range []struct { name string s string tagName string expect bool }{ {"empty", "", "div", false}, {"no match", "foo", "div", false}, {"no close", "foo
", "div", false}, {"no close 2", "foo/div>", "div", false}, {"no close 2", "foo//div>", "div", false}, {"no tag", "foo", "div", false}, {"match", "foo
", "div", true}, {"match space", "foo< / div>", "div", true}, {"match space 2", "foo< / div \n>", "div", true}, {"match case", "foo
", "div", true}, {"self closing", ``, "div", false}, } { c.Run(test.name, func(c *qt.C) { got := isClosedByTag([]byte(test.s), []byte(test.tagName)) c.Assert(got, qt.Equals, test.expect) }) } } func BenchmarkElementsCollectorWriter(b *testing.B) { const benchHTML = ` title

To force
line breaks
in a text,
use the br
element.


Month Savings
January $100
February $200
$300
` for i := 0; i < b.N; i++ { w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) fmt.Fprint(w, benchHTML) } } func BenchmarkElementsCollectorWriterPre(b *testing.B) { const benchHTML = `
foobar

foo
bar
baz
qux
quux
quuz
corge
` w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) for i := 0; i < b.N; i++ { fmt.Fprint(w, benchHTML) } }