transform_test.go (1740B)
1 // Copyright 2019 The Hugo Authors. All rights reserved.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 // http://www.apache.org/licenses/LICENSE-2.0
7 //
8 // Unless required by applicable law or agreed to in writing, software
9 // distributed under the License is distributed on an "AS IS" BASIS,
10 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 // See the License for the specific language governing permissions and
12 // limitations under the License.
13
14 package text
15
16 import (
17 "testing"
18
19 qt "github.com/frankban/quicktest"
20 )
21
22 func TestRemoveAccents(t *testing.T) {
23 c := qt.New(t)
24
25 c.Assert(string(RemoveAccents([]byte("Resumé"))), qt.Equals, "Resume")
26 c.Assert(string(RemoveAccents([]byte("Hugo Rocks!"))), qt.Equals, "Hugo Rocks!")
27 c.Assert(string(RemoveAccentsString("Resumé")), qt.Equals, "Resume")
28 }
29
30 func TestChomp(t *testing.T) {
31 c := qt.New(t)
32
33 c.Assert(Chomp("\nA\n"), qt.Equals, "\nA")
34 c.Assert(Chomp("A\r\n"), qt.Equals, "A")
35 }
36
37 func TestPuts(t *testing.T) {
38 c := qt.New(t)
39
40 c.Assert(Puts("A"), qt.Equals, "A\n")
41 c.Assert(Puts("\nA\n"), qt.Equals, "\nA\n")
42 c.Assert(Puts(""), qt.Equals, "")
43 }
44
45 func TestVisitLinesAfter(t *testing.T) {
46 const lines = `line 1
47 line 2
48
49 line 3`
50
51 var collected []string
52
53 VisitLinesAfter(lines, func(s string) {
54 collected = append(collected, s)
55 })
56
57 c := qt.New(t)
58
59 c.Assert(collected, qt.DeepEquals, []string{"line 1\n", "line 2\n", "\n", "line 3"})
60
61 }
62
63 func BenchmarkVisitLinesAfter(b *testing.B) {
64 const lines = `line 1
65 line 2
66
67 line 3`
68
69 for i := 0; i < b.N; i++ {
70 VisitLinesAfter(lines, func(s string) {
71
72 })
73
74 }
75
76 }