package transform import ( "bytes" "github.com/spf13/hugo/helpers" "strings" "testing" ) const H5_JS_CONTENT_DOUBLE_QUOTE = "
content foobar. Follow up
" const H5_JS_CONTENT_SINGLE_QUOTE = "
content foobar. Follow up
" const H5_JS_CONTENT_ABS_URL = "
content foobar. Follow up
" const H5_JS_CONTENT_ABS_URL_SCHEMALESS = "
content foobar. Follow up
" const CORRECT_OUTPUT_SRC_HREF_SQ = "
content foobar. Follow up
" const H5_XML_CONTENT_ABS_URL = "<p><a href="/foobar">foobar</a></p> <p>A video: <iframe src='/foo'></iframe></p>" const CORRECT_OUTPUT_SRC_HREF_IN_XML = "<p><a href="http://base/foobar">foobar</a></p> <p>A video: <iframe src='http://base/foo'></iframe></p>" const H5_XML_CONTENT_GUARDED = "<p><a href="//foobar">foobar</a></p> <p>A video: <iframe src='//foo'></iframe></p>" // additional sanity tests for replacements testing const REPLACE_1 = "No replacements." const REPLACE_2 = "ᚠᛇᚻ ᛒᛦᚦ ᚠᚱᚩᚠᚢᚱ\nᚠᛁᚱᚪ ᚷᛖᚻᚹᛦᛚᚳᚢᛗ" // Issue: 816, schemaless links combined with others const REPLACE_SCHEMALESS_HTML = `Pre. src='//schemaless' src='/normal' Schemaless. normal. Post.` const REPLACE_SCHEMALESS_HTML_CORRECT = `Pre. src='//schemaless' src='http://base/normal' Schemaless. normal. Post.` const REPLACE_SCHEMALESS_XML = `Pre. src="//schemaless" src="/normal" Schemaless. normal. Post.` const REPLACE_SCHEMALESS_XML_CORRECT = `Pre. src="//schemaless" src="http://base/normal" Schemaless. normal. Post.` var abs_url_bench_tests = []test{ {H5_JS_CONTENT_DOUBLE_QUOTE, CORRECT_OUTPUT_SRC_HREF_DQ}, {H5_JS_CONTENT_SINGLE_QUOTE, CORRECT_OUTPUT_SRC_HREF_SQ}, {H5_JS_CONTENT_ABS_URL, H5_JS_CONTENT_ABS_URL}, {H5_JS_CONTENT_ABS_URL_SCHEMALESS, H5_JS_CONTENT_ABS_URL_SCHEMALESS}, } var xml_abs_url_bench_tests = []test{ {H5_XML_CONTENT_ABS_URL, CORRECT_OUTPUT_SRC_HREF_IN_XML}, {H5_XML_CONTENT_GUARDED, H5_XML_CONTENT_GUARDED}, } var sanity_tests = []test{{REPLACE_1, REPLACE_1}, {REPLACE_2, REPLACE_2}} var extra_tests_html = []test{{REPLACE_SCHEMALESS_HTML, REPLACE_SCHEMALESS_HTML_CORRECT}} var abs_url_tests = append(abs_url_bench_tests, append(sanity_tests, extra_tests_html...)...) var extra_tests_xml = []test{{REPLACE_SCHEMALESS_XML, REPLACE_SCHEMALESS_XML_CORRECT}} var xml_abs_url_tests = append(xml_abs_url_bench_tests, append(sanity_tests, extra_tests_xml...)...) func TestChainZeroTransformers(t *testing.T) { tr := NewChain() in := new(bytes.Buffer) out := new(bytes.Buffer) if err := tr.Apply(in, out); err != nil { t.Errorf("A zero transformer chain returned an error.") } } func TestChaingMultipleTransformers(t *testing.T) { f1 := func(ct contentTransformer) { ct.Write(bytes.Replace(ct.Content(), []byte("f1"), []byte("f1r"), -1)) } f2 := func(ct contentTransformer) { ct.Write(bytes.Replace(ct.Content(), []byte("f2"), []byte("f2r"), -1)) } f3 := func(ct contentTransformer) { ct.Write(bytes.Replace(ct.Content(), []byte("f3"), []byte("f3r"), -1)) } f4 := func(ct contentTransformer) { ct.Write(bytes.Replace(ct.Content(), []byte("f4"), []byte("f4r"), -1)) } tr := NewChain(f1, f2, f3, f4) out := new(bytes.Buffer) if err := tr.Apply(out, helpers.StringToReader("Test: f4 f3 f1 f2 f1 The End.")); err != nil { t.Errorf("Multi transformer chain returned an error: %s", err) } expected := "Test: f4r f3r f1r f2r f1r The End." if string(out.Bytes()) != expected { t.Errorf("Expected %s got %s", expected, string(out.Bytes())) } } func BenchmarkAbsURL(b *testing.B) { absURL, _ := absURLFromURL("http://base") tr := NewChain(absURL...) b.ResetTimer() for i := 0; i < b.N; i++ { apply(b.Errorf, tr, abs_url_bench_tests) } } func TestAbsURL(t *testing.T) { absURL, _ := absURLFromURL("http://base") tr := NewChain(absURL...) apply(t.Errorf, tr, abs_url_tests) } func BenchmarkXMLAbsURL(b *testing.B) { absURLInXML, _ := absURLInXMLFromURL("http://base") tr := NewChain(absURLInXML...) b.ResetTimer() for i := 0; i < b.N; i++ { apply(b.Errorf, tr, xml_abs_url_bench_tests) } } func TestXMLAbsURL(t *testing.T) { absURLInXML, _ := absURLInXMLFromURL("http://base") tr := NewChain(absURLInXML...) apply(t.Errorf, tr, xml_abs_url_tests) } type errorf func(string, ...interface{}) func apply(ef errorf, tr chain, tests []test) { for _, test := range tests { out := new(bytes.Buffer) err := tr.Apply(out, strings.NewReader(test.content)) if err != nil { ef("Unexpected error: %s", err) } if test.expected != string(out.Bytes()) { ef("Expected:\n%s\nGot:\n%s", test.expected, string(out.Bytes())) } } } type test struct { content string expected string }