// Copyright 2016 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 strings import ( "html/template" "reflect" "strings" "testing" ) func TestTruncate(t *testing.T) { t.Parallel() var err error cases := []struct { v1 interface{} v2 interface{} v3 interface{} want interface{} isErr bool }{ {10, "I am a test sentence", nil, template.HTML("I am a …"), false}, {10, "", "I am a test sentence", template.HTML("I am a"), false}, {10, "", "a b c d e f g h i j k", template.HTML("a b c d e"), false}, {12, "", "Should be escaped", template.HTML("<b>Should be"), false}, {10, template.HTML(" Read more"), "I am a test sentence", template.HTML("I am a Read more"), false}, {20, template.HTML("I have a Markdown link inside."), nil, template.HTML("I have a Markdown …"), false}, {10, "IamanextremelylongwordthatjustgoesonandonandonjusttoannoyyoualmostasifIwaswritteninGermanActuallyIbettheresagermanwordforthis", nil, template.HTML("Iamanextre …"), false}, {10, template.HTML("

IamanextremelylongwordthatjustgoesonandonandonjusttoannoyyoualmostasifIwaswritteninGermanActuallyIbettheresagermanwordforthis

"), nil, template.HTML("

Iamanextre …

"), false}, {13, template.HTML("With Markdown inside."), nil, template.HTML("With Markdown …"), false}, {14, "Hello中国 Good 好的", nil, template.HTML("Hello中国 Good 好 …"), false}, {15, "", template.HTML("A
tag that's not closed"), template.HTML("A
tag that's"), false}, {14, template.HTML("

Hello中国 Good 好的

"), nil, template.HTML("

Hello中国 Good 好 …

"), false}, {2, template.HTML("

P1

P2

"), nil, template.HTML("

P1 …

"), false}, {3, template.HTML(strings.Repeat("

P

", 20)), nil, template.HTML("

P

P

P …

"), false}, {18, template.HTML("

test hello test something

"), nil, template.HTML("

test hello test …

"), false}, {4, template.HTML("

abc d e

"), nil, template.HTML("

abc …

"), false}, {10, nil, nil, template.HTML(""), true}, {nil, nil, nil, template.HTML(""), true}, } for i, c := range cases { var result template.HTML if c.v2 == nil { result, err = ns.Truncate(c.v1) } else if c.v3 == nil { result, err = ns.Truncate(c.v1, c.v2) } else { result, err = ns.Truncate(c.v1, c.v2, c.v3) } if c.isErr { if err == nil { t.Errorf("[%d] Slice didn't return an expected error", i) } } else { if err != nil { t.Errorf("[%d] failed: %s", i, err) continue } if !reflect.DeepEqual(result, c.want) { t.Errorf("[%d] got '%s' but expected '%s'", i, result, c.want) } } } // Too many arguments _, err = ns.Truncate(10, " ...", "I am a test sentence", "wrong") if err == nil { t.Errorf("Should have errored") } }