// Copyright 2024 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 helpers_test import ( "bytes" "html/template" "strings" "testing" qt "github.com/frankban/quicktest" "github.com/gohugoio/hugo/config" "github.com/gohugoio/hugo/helpers" ) func TestTrimShortHTML(t *testing.T) { tests := []struct { markup string input []byte output []byte }{ {"markdown", []byte(""), []byte("")}, {"markdown", []byte("Plain text"), []byte("Plain text")}, {"markdown", []byte("

Simple paragraph

"), []byte("Simple paragraph")}, {"markdown", []byte("\n \n \t

\t Whitespace\nHTML \n\t

\n\t"), []byte("Whitespace\nHTML")}, {"markdown", []byte("

Multiple

paragraphs

"), []byte("

Multiple

paragraphs

")}, {"markdown", []byte("

Nested

paragraphs

"), []byte("

Nested

paragraphs

")}, {"markdown", []byte("

Hello

\n"), []byte("

Hello

\n")}, // Issue 11698 {"markdown", []byte("

b

\n\n

c

"), []byte("

b

\n\n

c

")}, // Issue 12369 {"markdown", []byte("
\n

foo

\n
"), []byte("
\n

foo

\n
")}, {"asciidocext", []byte("
\n

foo

\n
"), []byte("foo")}, } c := newTestContentSpec(nil) for i, test := range tests { output := c.TrimShortHTML(test.input, test.markup) if !bytes.Equal(test.output, output) { t.Errorf("Test %d failed. Expected %q got %q", i, test.output, output) } } } func BenchmarkTrimShortHTML(b *testing.B) { c := newTestContentSpec(nil) b.ResetTimer() for i := 0; i < b.N; i++ { c.TrimShortHTML([]byte("

Simple paragraph

"), "markdown") } } func TestBytesToHTML(t *testing.T) { c := qt.New(t) c.Assert(helpers.BytesToHTML([]byte("dobedobedo")), qt.Equals, template.HTML("dobedobedo")) } var benchmarkTruncateString = strings.Repeat("This is a sentence about nothing.", 20) func BenchmarkTestTruncateWordsToWholeSentence(b *testing.B) { c := newTestContentSpec(nil) b.ResetTimer() for i := 0; i < b.N; i++ { c.TruncateWordsToWholeSentence(benchmarkTruncateString) } } func TestTruncateWordsToWholeSentence(t *testing.T) { type test struct { input, expected string max int truncated bool } data := []test{ {"a b c", "a b c", 12, false}, {"a b c", "a b c", 3, false}, {"a", "a", 1, false}, {"This is a sentence.", "This is a sentence.", 5, false}, {"This is also a sentence!", "This is also a sentence!", 1, false}, {"To be. Or not to be. That's the question.", "To be.", 1, true}, {" \nThis is not a sentence\nAnd this is another", "This is not a sentence", 4, true}, {"", "", 10, false}, {"This... is a more difficult test?", "This... is a more difficult test?", 1, false}, } for i, d := range data { cfg := config.New() cfg.Set("summaryLength", d.max) c := newTestContentSpec(cfg) output, truncated := c.TruncateWordsToWholeSentence(d.input) if d.expected != output { t.Errorf("Test %d failed. Expected %q got %q", i, d.expected, output) } if d.truncated != truncated { t.Errorf("Test %d failed. Expected truncated=%t got %t", i, d.truncated, truncated) } } } func TestTruncateWordsByRune(t *testing.T) { type test struct { input, expected string max int truncated bool } data := []test{ {"", "", 1, false}, {"a b c", "a b c", 12, false}, {"a b c", "a b c", 3, false}, {"a", "a", 1, false}, {"Hello 中国", "", 0, true}, {"这是中文,全中文。", "这是中文,", 5, true}, {"Hello 中国", "Hello 中", 2, true}, {"Hello 中国", "Hello 中国", 3, false}, {"Hello中国 Good 好的", "Hello中国 Good 好", 9, true}, {"This is a sentence.", "This is", 2, true}, {"This is also a sentence!", "This", 1, true}, {"To be. Or not to be. That's the question.", "To be. Or not", 4, true}, {" \nThis is not a sentence\n ", "This is not", 3, true}, } for i, d := range data { cfg := config.New() cfg.Set("summaryLength", d.max) c := newTestContentSpec(cfg) output, truncated := c.TruncateWordsByRune(strings.Fields(d.input)) if d.expected != output { t.Errorf("Test %d failed. Expected %q got %q", i, d.expected, output) } if d.truncated != truncated { t.Errorf("Test %d failed. Expected truncated=%t got %t", i, d.truncated, truncated) } } } func TestExtractTOCNormalContent(t *testing.T) { content := []byte("