// 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" "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">