// 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" "strings" "testing" "github.com/gohugoio/hugo/media" "github.com/gohugoio/hugo/minifiers" "github.com/gohugoio/hugo/output" qt "github.com/frankban/quicktest" "github.com/spf13/viper" ) 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, } } 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", `
`, f("div", "a b", "")}, {"single quote", ``, f("body", "a b", "")}, {"no quote", ``, f("body", "b", "myelement")}, // https://github.com/gohugoio/hugo/issues/7318 {"thead", `
`, 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("", "", "")}, // Issue #8417 {"Tabs inline", `
d
`, f("div hr", "bar foo", "a")}, {"Tabs on multiple rows", `
d
`, f("div form", "foo", "a b")}, } { for _, minify := range []bool{false, true} { c.Run(fmt.Sprintf("%s--minify-%t", test.name, minify), func(c *qt.C) { w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) if minify { if skipMinifyTest[test.name] { c.Skip("skip minify test") } v := viper.New() m, _ := minifiers.New(media.DefaultTypes, output.DefaultFormats, v) m.Minify(media.HTMLType, w, strings.NewReader(test.html)) } else { 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) } } const benchHTML = ` title

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


Month Savings
January $100
February $200
$300
` func BenchmarkElementsCollectorWriter(b *testing.B) { b.ReportAllocs() for i := 0; i < b.N; i++ { w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) fmt.Fprint(w, benchHTML) } } func BenchmarkElementsCollectorWriterMinified(b *testing.B) { b.ReportAllocs() v := viper.New() m, _ := minifiers.New(media.DefaultTypes, output.DefaultFormats, v) var buf bytes.Buffer m.Minify(media.HTMLType, &buf, strings.NewReader(benchHTML)) b.ResetTimer() for i := 0; i < b.N; i++ { w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) fmt.Fprint(w, buf.String()) } } func BenchmarkElementsCollectorWriterWithMinifyStream(b *testing.B) { b.ReportAllocs() v := viper.New() m, _ := minifiers.New(media.DefaultTypes, output.DefaultFormats, v) b.ResetTimer() for i := 0; i < b.N; i++ { w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) m.Minify(media.HTMLType, w, strings.NewReader(benchHTML)) } } func BenchmarkElementsCollectorWriterWithMinifyString(b *testing.B) { b.ReportAllocs() v := viper.New() m, _ := minifiers.New(media.DefaultTypes, output.DefaultFormats, v) b.ResetTimer() for i := 0; i < b.N; i++ { var buf bytes.Buffer m.Minify(media.HTMLType, &buf, strings.NewReader(benchHTML)) w := newHTMLElementsCollectorWriter(newHTMLElementsCollector()) fmt.Fprint(w, buf.String()) } }