2015-12-10 17:19:38 -05:00
// Copyright 2015 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.
2015-01-07 15:40:35 -05:00
package helpers
import (
2016-02-05 12:40:49 -05:00
"bytes"
2015-01-27 04:15:57 -05:00
"html/template"
2016-02-05 12:40:49 -05:00
"reflect"
2015-02-05 15:44:15 -05:00
"strings"
2015-01-07 15:40:35 -05:00
"testing"
2015-09-03 06:22:20 -04:00
2016-02-05 12:40:49 -05:00
"github.com/miekg/mmark"
"github.com/russross/blackfriday"
2015-09-03 06:22:20 -04:00
"github.com/stretchr/testify/assert"
2015-01-07 15:40:35 -05:00
)
2015-03-11 13:34:57 -04:00
const tstHTMLContent = "<!DOCTYPE html><html><head><script src=\"http://two/foobar.js\"></script></head><body><nav><ul><li hugo-nav=\"section_0\"></li><li hugo-nav=\"section_1\"></li></ul></nav><article>content <a href=\"http://two/foobar\">foobar</a>. Follow up</article><p>This is some text.<br>And some more.</p></body></html>"
2015-02-06 04:00:42 -05:00
2015-01-07 15:40:35 -05:00
func TestStripHTML ( t * testing . T ) {
type test struct {
input , expected string
}
data := [ ] test {
{ "<h1>strip h1 tag <h1>" , "strip h1 tag " } ,
{ "<p> strip p tag </p>" , " strip p tag \n" } ,
{ "</br> strip br<br>" , " strip br\n" } ,
2015-02-05 12:31:11 -05:00
{ "</br> strip br2<br />" , " strip br2\n" } ,
{ "This <strong>is</strong> a\nnewline" , "This is a newline" } ,
2016-02-05 12:40:49 -05:00
{ "No Tags" , "No Tags" } ,
2015-01-07 15:40:35 -05:00
}
for i , d := range data {
output := StripHTML ( d . input )
if d . expected != output {
t . Errorf ( "Test %d failed. Expected %q got %q" , i , d . expected , output )
}
}
}
2015-01-27 04:15:57 -05:00
2015-02-06 04:00:42 -05:00
func BenchmarkStripHTML ( b * testing . B ) {
b . ResetTimer ( )
for i := 0 ; i < b . N ; i ++ {
2015-03-11 13:34:57 -04:00
StripHTML ( tstHTMLContent )
2015-02-06 04:00:42 -05:00
}
}
2015-01-27 04:15:57 -05:00
func TestStripEmptyNav ( t * testing . T ) {
cleaned := StripEmptyNav ( [ ] byte ( "do<nav>\n</nav>\n\nbedobedo" ) )
assert . Equal ( t , [ ] byte ( "dobedobedo" ) , cleaned )
}
func TestBytesToHTML ( t * testing . T ) {
assert . Equal ( t , template . HTML ( "dobedobedo" ) , BytesToHTML ( [ ] byte ( "dobedobedo" ) ) )
}
2015-02-05 15:44:15 -05:00
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\n " , "This is not a" , 4 , true } ,
}
for i , d := range data {
output , truncated := TruncateWordsToWholeSentence ( strings . Fields ( d . input ) , d . max )
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 )
}
}
}
2015-09-03 06:22:20 -04:00
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 {
output , truncated := TruncateWordsByRune ( strings . Fields ( d . input ) , d . max )
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 )
}
}
}
2016-02-05 12:40:49 -05:00
func TestGetHTMLRendererFlags ( t * testing . T ) {
ctx := & RenderingContext { }
renderer := GetHTMLRenderer ( blackfriday . HTML_USE_XHTML , ctx )
flags := renderer . GetFlags ( )
if flags & blackfriday . HTML_USE_XHTML != blackfriday . HTML_USE_XHTML {
t . Errorf ( "Test flag: %d was not found amongs set flags:%d; Result: %d" , blackfriday . HTML_USE_XHTML , flags , flags & blackfriday . HTML_USE_XHTML )
}
}
func TestGetHTMLRendererAllFlags ( t * testing . T ) {
type data struct {
testFlag int
}
allFlags := [ ] data {
{ blackfriday . HTML_USE_XHTML } ,
{ blackfriday . HTML_FOOTNOTE_RETURN_LINKS } ,
{ blackfriday . HTML_USE_SMARTYPANTS } ,
{ blackfriday . HTML_SMARTYPANTS_ANGLED_QUOTES } ,
{ blackfriday . HTML_SMARTYPANTS_FRACTIONS } ,
{ blackfriday . HTML_HREF_TARGET_BLANK } ,
{ blackfriday . HTML_SMARTYPANTS_DASHES } ,
{ blackfriday . HTML_SMARTYPANTS_LATEX_DASHES } ,
}
defaultFlags := blackfriday . HTML_USE_XHTML
ctx := & RenderingContext { }
ctx . Config = ctx . getConfig ( )
ctx . Config . AngledQuotes = true
ctx . Config . Fractions = true
ctx . Config . HrefTargetBlank = true
ctx . Config . LatexDashes = true
ctx . Config . PlainIDAnchors = true
ctx . Config . SmartDashes = true
ctx . Config . Smartypants = true
ctx . Config . SourceRelativeLinksEval = true
renderer := GetHTMLRenderer ( defaultFlags , ctx )
actualFlags := renderer . GetFlags ( )
var expectedFlags int
//OR-ing flags together...
for _ , d := range allFlags {
expectedFlags |= d . testFlag
}
if expectedFlags != actualFlags {
t . Errorf ( "Expected flags (%d) did not equal actual (%d) flags." , expectedFlags , actualFlags )
}
}
func TestGetHTMLRendererAnchors ( t * testing . T ) {
ctx := & RenderingContext { }
ctx . DocumentID = "testid"
ctx . Config = ctx . getConfig ( )
ctx . Config . PlainIDAnchors = false
actualRenderer := GetHTMLRenderer ( 0 , ctx )
headerBuffer := & bytes . Buffer { }
footnoteBuffer := & bytes . Buffer { }
expectedFootnoteHref := [ ] byte ( "href=\"#fn:testid:href\"" )
expectedHeaderID := [ ] byte ( "<h1 id=\"id:testid\"></h1>\n" )
actualRenderer . Header ( headerBuffer , func ( ) bool { return true } , 1 , "id" )
actualRenderer . FootnoteRef ( footnoteBuffer , [ ] byte ( "href" ) , 1 )
if ! bytes . Contains ( footnoteBuffer . Bytes ( ) , expectedFootnoteHref ) {
t . Errorf ( "Footnote anchor prefix not applied. Actual:%s Expected:%s" , footnoteBuffer . String ( ) , expectedFootnoteHref )
}
if ! bytes . Equal ( headerBuffer . Bytes ( ) , expectedHeaderID ) {
t . Errorf ( "Header Id Postfix not applied. Actual:%s Expected:%s" , headerBuffer . String ( ) , expectedHeaderID )
}
}
func TestGetMmarkHtmlRenderer ( t * testing . T ) {
ctx := & RenderingContext { }
ctx . DocumentID = "testid"
ctx . Config = ctx . getConfig ( )
ctx . Config . PlainIDAnchors = false
actualRenderer := GetMmarkHtmlRenderer ( 0 , ctx )
headerBuffer := & bytes . Buffer { }
footnoteBuffer := & bytes . Buffer { }
expectedFootnoteHref := [ ] byte ( "href=\"#fn:testid:href\"" )
expectedHeaderID := [ ] byte ( "<h1 id=\"id\"></h1>" )
actualRenderer . FootnoteRef ( footnoteBuffer , [ ] byte ( "href" ) , 1 )
actualRenderer . Header ( headerBuffer , func ( ) bool { return true } , 1 , "id" )
if ! bytes . Contains ( footnoteBuffer . Bytes ( ) , expectedFootnoteHref ) {
t . Errorf ( "Footnote anchor prefix not applied. Actual:%s Expected:%s" , footnoteBuffer . String ( ) , expectedFootnoteHref )
}
if bytes . Equal ( headerBuffer . Bytes ( ) , expectedHeaderID ) {
t . Errorf ( "Header Id Postfix applied. Actual:%s Expected:%s" , headerBuffer . String ( ) , expectedHeaderID )
}
}
func TestGetMarkdownExtensionsMasksAreRemovedFromExtensions ( t * testing . T ) {
ctx := & RenderingContext { }
ctx . Config = ctx . getConfig ( )
ctx . Config . Extensions = [ ] string { "headerId" }
ctx . Config . ExtensionsMask = [ ] string { "noIntraEmphasis" }
actualFlags := getMarkdownExtensions ( ctx )
if actualFlags & blackfriday . EXTENSION_NO_INTRA_EMPHASIS == blackfriday . EXTENSION_NO_INTRA_EMPHASIS {
t . Errorf ( "Masked out flag {%v} found amongst returned extensions." , blackfriday . EXTENSION_NO_INTRA_EMPHASIS )
}
}
func TestGetMarkdownExtensionsByDefaultAllExtensionsAreEnabled ( t * testing . T ) {
type data struct {
testFlag int
}
ctx := & RenderingContext { }
ctx . Config = ctx . getConfig ( )
ctx . Config . Extensions = [ ] string { "" }
ctx . Config . ExtensionsMask = [ ] string { "" }
allExtensions := [ ] data {
{ blackfriday . EXTENSION_NO_INTRA_EMPHASIS } ,
{ blackfriday . EXTENSION_TABLES } ,
{ blackfriday . EXTENSION_FENCED_CODE } ,
{ blackfriday . EXTENSION_AUTOLINK } ,
{ blackfriday . EXTENSION_STRIKETHROUGH } ,
{ blackfriday . EXTENSION_SPACE_HEADERS } ,
{ blackfriday . EXTENSION_FOOTNOTES } ,
{ blackfriday . EXTENSION_HEADER_IDS } ,
{ blackfriday . EXTENSION_AUTO_HEADER_IDS } ,
{ blackfriday . EXTENSION_DEFINITION_LISTS } ,
}
actualFlags := getMarkdownExtensions ( ctx )
for _ , e := range allExtensions {
if actualFlags & e . testFlag != e . testFlag {
t . Errorf ( "Flag %v was not found in the list of extensions." , e )
}
}
}
func TestGetMarkdownExtensionsAddingFlagsThroughRenderingContext ( t * testing . T ) {
ctx := & RenderingContext { }
ctx . Config = ctx . getConfig ( )
ctx . Config . Extensions = [ ] string { "definitionLists" }
ctx . Config . ExtensionsMask = [ ] string { "" }
actualFlags := getMarkdownExtensions ( ctx )
if actualFlags & blackfriday . EXTENSION_DEFINITION_LISTS != blackfriday . EXTENSION_DEFINITION_LISTS {
t . Errorf ( "Masked out flag {%v} found amongst returned extensions." , blackfriday . EXTENSION_DEFINITION_LISTS )
}
}
func TestGetMarkdownRenderer ( t * testing . T ) {
ctx := & RenderingContext { }
ctx . Content = [ ] byte ( "testContent" )
ctx . Config = ctx . getConfig ( )
actualRenderedMarkdown := markdownRender ( ctx )
expectedRenderedMarkdown := [ ] byte ( "<p>testContent</p>\n" )
if ! bytes . Equal ( actualRenderedMarkdown , expectedRenderedMarkdown ) {
t . Errorf ( "Actual rendered Markdown (%s) did not match expected markdown (%s)" , actualRenderedMarkdown , expectedRenderedMarkdown )
}
}
func TestGetMarkdownRendererWithTOC ( t * testing . T ) {
ctx := & RenderingContext { }
ctx . Content = [ ] byte ( "testContent" )
ctx . Config = ctx . getConfig ( )
actualRenderedMarkdown := markdownRenderWithTOC ( ctx )
expectedRenderedMarkdown := [ ] byte ( "<nav>\n</nav>\n\n<p>testContent</p>\n" )
if ! bytes . Equal ( actualRenderedMarkdown , expectedRenderedMarkdown ) {
t . Errorf ( "Actual rendered Markdown (%s) did not match expected markdown (%s)" , actualRenderedMarkdown , expectedRenderedMarkdown )
}
}
func TestGetMmarkExtensions ( t * testing . T ) {
//TODO: This is doing the same just with different marks...
type data struct {
testFlag int
}
ctx := & RenderingContext { }
ctx . Config = ctx . getConfig ( )
ctx . Config . Extensions = [ ] string { "tables" }
ctx . Config . ExtensionsMask = [ ] string { "" }
allExtensions := [ ] data {
{ mmark . EXTENSION_TABLES } ,
{ mmark . EXTENSION_FENCED_CODE } ,
{ mmark . EXTENSION_AUTOLINK } ,
{ mmark . EXTENSION_SPACE_HEADERS } ,
{ mmark . EXTENSION_CITATION } ,
{ mmark . EXTENSION_TITLEBLOCK_TOML } ,
{ mmark . EXTENSION_HEADER_IDS } ,
{ mmark . EXTENSION_AUTO_HEADER_IDS } ,
{ mmark . EXTENSION_UNIQUE_HEADER_IDS } ,
{ mmark . EXTENSION_FOOTNOTES } ,
{ mmark . EXTENSION_SHORT_REF } ,
{ mmark . EXTENSION_NO_EMPTY_LINE_BEFORE_BLOCK } ,
{ mmark . EXTENSION_INCLUDE } ,
}
actualFlags := GetMmarkExtensions ( ctx )
for _ , e := range allExtensions {
if actualFlags & e . testFlag != e . testFlag {
t . Errorf ( "Flag %v was not found in the list of extensions." , e )
}
}
}
func TestMmarkRender ( t * testing . T ) {
ctx := & RenderingContext { }
ctx . Content = [ ] byte ( "testContent" )
ctx . Config = ctx . getConfig ( )
actualRenderedMarkdown := MmarkRender ( ctx )
expectedRenderedMarkdown := [ ] byte ( "<p>testContent</p>\n" )
if ! bytes . Equal ( actualRenderedMarkdown , expectedRenderedMarkdown ) {
t . Errorf ( "Actual rendered Markdown (%s) did not match expected markdown (%s)" , actualRenderedMarkdown , expectedRenderedMarkdown )
}
}
func TestExtractTOCNormalContent ( t * testing . T ) {
content := [ ] byte ( "<nav>\n<ul>\nTOC<li><a href=\"#" )
actualTocLessContent , actualToc := ExtractTOC ( content )
expectedTocLess := [ ] byte ( "TOC<li><a href=\"#" )
expectedToc := [ ] byte ( "<nav id=\"TableOfContents\">\n<ul>\n" )
if ! bytes . Equal ( actualTocLessContent , expectedTocLess ) {
t . Errorf ( "Actual tocless (%s) did not equal expected (%s) tocless content" , actualTocLessContent , expectedTocLess )
}
if ! bytes . Equal ( actualToc , expectedToc ) {
t . Errorf ( "Actual toc (%s) did not equal expected (%s) toc content" , actualToc , expectedToc )
}
}
func TestExtractTOCGreaterThanSeventy ( t * testing . T ) {
content := [ ] byte ( "<nav>\n<ul>\nTOC This is a very long content which will definitly be greater than seventy, I promise you that.<li><a href=\"#" )
actualTocLessContent , actualToc := ExtractTOC ( content )
//Because the start of Toc is greater than 70+startpoint of <li> content and empty TOC will be returned
expectedToc := [ ] byte ( "" )
if ! bytes . Equal ( actualTocLessContent , content ) {
t . Errorf ( "Actual tocless (%s) did not equal expected (%s) tocless content" , actualTocLessContent , content )
}
if ! bytes . Equal ( actualToc , expectedToc ) {
t . Errorf ( "Actual toc (%s) did not equal expected (%s) toc content" , actualToc , expectedToc )
}
}
func TestExtractNoTOC ( t * testing . T ) {
content := [ ] byte ( "TOC" )
actualTocLessContent , actualToc := ExtractTOC ( content )
expectedToc := [ ] byte ( "" )
if ! bytes . Equal ( actualTocLessContent , content ) {
t . Errorf ( "Actual tocless (%s) did not equal expected (%s) tocless content" , actualTocLessContent , content )
}
if ! bytes . Equal ( actualToc , expectedToc ) {
t . Errorf ( "Actual toc (%s) did not equal expected (%s) toc content" , actualToc , expectedToc )
}
}
func TestTotalWords ( t * testing . T ) {
testString := "Two, Words!"
actualWordCount := TotalWords ( testString )
if actualWordCount != 2 {
t . Errorf ( "Actual word count (%d) for test string (%s) did not match 2." , actualWordCount , testString )
}
}
func TestWordCount ( t * testing . T ) {
testString := "Two, Words!"
expectedMap := map [ string ] int { "Two," : 1 , "Words!" : 1 }
actualMap := WordCount ( testString )
if ! reflect . DeepEqual ( expectedMap , actualMap ) {
t . Errorf ( "Actual Map (%v) does not equal expected (%v)" , actualMap , expectedMap )
}
}
func TestRemoveSummaryDivider ( t * testing . T ) {
content := [ ] byte ( "This is before. <!--more-->This is after." )
actualRemovedContent := RemoveSummaryDivider ( content )
expectedRemovedContent := [ ] byte ( "This is before. This is after." )
if ! bytes . Equal ( actualRemovedContent , expectedRemovedContent ) {
t . Errorf ( "Actual removed content (%s) did not equal expected removed content (%s)" , actualRemovedContent , expectedRemovedContent )
}
}
