mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-30 01:23:34 -05:00
33ae10b6ad
Fixes #3040
234 lines
5.1 KiB
Go
234 lines
5.1 KiB
Go
// Copyright 2017 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 transform
|
|
|
|
import (
|
|
"fmt"
|
|
"html/template"
|
|
"testing"
|
|
|
|
"github.com/gohugoio/hugo/config"
|
|
"github.com/gohugoio/hugo/deps"
|
|
"github.com/gohugoio/hugo/helpers"
|
|
"github.com/gohugoio/hugo/hugofs"
|
|
"github.com/spf13/viper"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type tstNoStringer struct{}
|
|
|
|
func TestEmojify(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ns := New(newDeps(viper.New()))
|
|
|
|
for i, test := range []struct {
|
|
s interface{}
|
|
expect interface{}
|
|
}{
|
|
{":notamoji:", template.HTML(":notamoji:")},
|
|
{"I :heart: Hugo", template.HTML("I ❤️ Hugo")},
|
|
// errors
|
|
{tstNoStringer{}, false},
|
|
} {
|
|
errMsg := fmt.Sprintf("[%d] %s", i, test.s)
|
|
|
|
result, err := ns.Emojify(test.s)
|
|
|
|
if b, ok := test.expect.(bool); ok && !b {
|
|
require.Error(t, err, errMsg)
|
|
continue
|
|
}
|
|
|
|
require.NoError(t, err, errMsg)
|
|
assert.Equal(t, test.expect, result, errMsg)
|
|
}
|
|
}
|
|
|
|
func TestHighlight(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ns := New(newDeps(viper.New()))
|
|
|
|
for i, test := range []struct {
|
|
s interface{}
|
|
lang string
|
|
opts string
|
|
expect interface{}
|
|
}{
|
|
{"func boo() {}", "go", "", "boo"},
|
|
{tstNoStringer{}, "go", "", false},
|
|
} {
|
|
errMsg := fmt.Sprintf("[%d]", i)
|
|
|
|
result, err := ns.Highlight(test.s, test.lang, test.opts)
|
|
|
|
if b, ok := test.expect.(bool); ok && !b {
|
|
require.Error(t, err, errMsg)
|
|
continue
|
|
}
|
|
|
|
require.NoError(t, err, errMsg)
|
|
assert.Contains(t, result, "boo", errMsg)
|
|
}
|
|
}
|
|
|
|
func TestHTMLEscape(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ns := New(newDeps(viper.New()))
|
|
|
|
for i, test := range []struct {
|
|
s interface{}
|
|
expect interface{}
|
|
}{
|
|
{`"Foo & Bar's Diner" <y@z>`, `"Foo & Bar's Diner" <y@z>`},
|
|
{"Hugo & Caddy > Wordpress & Apache", "Hugo & Caddy > Wordpress & Apache"},
|
|
// errors
|
|
{tstNoStringer{}, false},
|
|
} {
|
|
errMsg := fmt.Sprintf("[%d] %s", i, test.s)
|
|
|
|
result, err := ns.HTMLEscape(test.s)
|
|
|
|
if b, ok := test.expect.(bool); ok && !b {
|
|
require.Error(t, err, errMsg)
|
|
continue
|
|
}
|
|
|
|
require.NoError(t, err, errMsg)
|
|
assert.Equal(t, test.expect, result, errMsg)
|
|
}
|
|
}
|
|
|
|
func TestHTMLUnescape(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ns := New(newDeps(viper.New()))
|
|
|
|
for i, test := range []struct {
|
|
s interface{}
|
|
expect interface{}
|
|
}{
|
|
{`"Foo & Bar's Diner" <y@z>`, `"Foo & Bar's Diner" <y@z>`},
|
|
{"Hugo & Caddy > Wordpress & Apache", "Hugo & Caddy > Wordpress & Apache"},
|
|
// errors
|
|
{tstNoStringer{}, false},
|
|
} {
|
|
errMsg := fmt.Sprintf("[%d] %s", i, test.s)
|
|
|
|
result, err := ns.HTMLUnescape(test.s)
|
|
|
|
if b, ok := test.expect.(bool); ok && !b {
|
|
require.Error(t, err, errMsg)
|
|
continue
|
|
}
|
|
|
|
require.NoError(t, err, errMsg)
|
|
assert.Equal(t, test.expect, result, errMsg)
|
|
}
|
|
}
|
|
|
|
func TestMarkdownify(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ns := New(newDeps(viper.New()))
|
|
|
|
for i, test := range []struct {
|
|
s interface{}
|
|
expect interface{}
|
|
}{
|
|
{"Hello **World!**", template.HTML("Hello <strong>World!</strong>")},
|
|
{[]byte("Hello Bytes **World!**"), template.HTML("Hello Bytes <strong>World!</strong>")},
|
|
{tstNoStringer{}, false},
|
|
} {
|
|
errMsg := fmt.Sprintf("[%d] %s", i, test.s)
|
|
|
|
result, err := ns.Markdownify(test.s)
|
|
|
|
if b, ok := test.expect.(bool); ok && !b {
|
|
require.Error(t, err, errMsg)
|
|
continue
|
|
}
|
|
|
|
require.NoError(t, err, errMsg)
|
|
assert.Equal(t, test.expect, result, errMsg)
|
|
}
|
|
}
|
|
|
|
// Issue #3040
|
|
func TestMarkdownifyBlocksOfText(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
assert := require.New(t)
|
|
|
|
ns := New(newDeps(viper.New()))
|
|
|
|
text := `
|
|
#First
|
|
|
|
This is some *bold* text.
|
|
|
|
## Second
|
|
|
|
This is some more text.
|
|
|
|
And then some.
|
|
`
|
|
|
|
result, err := ns.Markdownify(text)
|
|
assert.NoError(err)
|
|
assert.Equal(template.HTML(
|
|
"<p>#First</p>\n\n<p>This is some <em>bold</em> text.</p>\n\n<h2 id=\"second\">Second</h2>\n\n<p>This is some more text.</p>\n\n<p>And then some.</p>\n"),
|
|
result)
|
|
|
|
}
|
|
|
|
func TestPlainify(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ns := New(newDeps(viper.New()))
|
|
|
|
for i, test := range []struct {
|
|
s interface{}
|
|
expect interface{}
|
|
}{
|
|
{"<em>Note:</em> blah <b>blah</b>", "Note: blah blah"},
|
|
// errors
|
|
{tstNoStringer{}, false},
|
|
} {
|
|
errMsg := fmt.Sprintf("[%d] %s", i, test.s)
|
|
|
|
result, err := ns.Plainify(test.s)
|
|
|
|
if b, ok := test.expect.(bool); ok && !b {
|
|
require.Error(t, err, errMsg)
|
|
continue
|
|
}
|
|
|
|
require.NoError(t, err, errMsg)
|
|
assert.Equal(t, test.expect, result, errMsg)
|
|
}
|
|
}
|
|
|
|
func newDeps(cfg config.Provider) *deps.Deps {
|
|
l := helpers.NewLanguage("en", cfg)
|
|
l.Set("i18nDir", "i18n")
|
|
return &deps.Deps{
|
|
Cfg: cfg,
|
|
Fs: hugofs.NewMem(l),
|
|
ContentSpec: helpers.NewContentSpec(l),
|
|
}
|
|
}
|