2015-01-07 15:40:35 -05:00
package helpers
import (
2015-01-27 04:15:57 -05:00
"github.com/stretchr/testify/assert"
"html/template"
2015-02-05 15:44:15 -05:00
"strings"
2015-01-07 15:40:35 -05:00
"testing"
)
2015-02-06 04:00:42 -05: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-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" } ,
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 ++ {
StripHTML ( tstHtmlContent )
}
}
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 )
}
}
}
