diff options
Diffstat (limited to 'markup')
-rw-r--r-- | markup/goldmark/hugocontext/hugocontext.go | 4 | ||||
-rw-r--r-- | markup/goldmark/hugocontext/hugocontext_test.go | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/markup/goldmark/hugocontext/hugocontext.go b/markup/goldmark/hugocontext/hugocontext.go index ed62bb8c6..b9c548dac 100644 --- a/markup/goldmark/hugocontext/hugocontext.go +++ b/markup/goldmark/hugocontext/hugocontext.go @@ -34,7 +34,7 @@ func New() goldmark.Extender { // Wrap wraps the given byte slice in a Hugo context that used to determine the correct Page // in .RenderShortcodes. -func Wrap(b []byte, pid uint64) []byte { +func Wrap(b []byte, pid uint64) string { buf := bufferpool.GetBuffer() defer bufferpool.PutBuffer(buf) buf.Write(prefix) @@ -45,7 +45,7 @@ func Wrap(b []byte, pid uint64) []byte { buf.Write(b) buf.Write(prefix) buf.Write(closingDelimAndNewline) - return buf.Bytes() + return buf.String() } var kindHugoContext = ast.NewNodeKind("HugoContext") diff --git a/markup/goldmark/hugocontext/hugocontext_test.go b/markup/goldmark/hugocontext/hugocontext_test.go index da96cc339..4a6eb80f5 100644 --- a/markup/goldmark/hugocontext/hugocontext_test.go +++ b/markup/goldmark/hugocontext/hugocontext_test.go @@ -24,7 +24,7 @@ func TestWrap(t *testing.T) { b := []byte("test") - c.Assert(string(Wrap(b, 42)), qt.Equals, "{{__hugo_ctx pid=42}}\ntest{{__hugo_ctx/}}\n") + c.Assert(Wrap(b, 42), qt.Equals, "{{__hugo_ctx pid=42}}\ntest{{__hugo_ctx/}}\n") } func BenchmarkWrap(b *testing.B) { |