mirror of
				https://github.com/gohugoio/hugo.git
				synced 2024-05-11 05:54:58 +00:00 
			
		
		
		
	This is a final rewrite that removes all the global state in Hugo, which also enables the use if `t.Parallel` in tests. Updates #2701 Fixes #3016
		
			
				
	
	
		
			203 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			203 lines
		
	
	
		
			8.6 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // 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.
 | |
| 
 | |
| package hugolib
 | |
| 
 | |
| import (
 | |
| 	"testing"
 | |
| )
 | |
| 
 | |
| type shortCodeLexerTest struct {
 | |
| 	name  string
 | |
| 	input string
 | |
| 	items []item
 | |
| }
 | |
| 
 | |
| var (
 | |
| 	tstEOF       = item{tEOF, 0, ""}
 | |
| 	tstLeftNoMD  = item{tLeftDelimScNoMarkup, 0, "{{<"}
 | |
| 	tstRightNoMD = item{tRightDelimScNoMarkup, 0, ">}}"}
 | |
| 	tstLeftMD    = item{tLeftDelimScWithMarkup, 0, "{{%"}
 | |
| 	tstRightMD   = item{tRightDelimScWithMarkup, 0, "%}}"}
 | |
| 	tstSCClose   = item{tScClose, 0, "/"}
 | |
| 	tstSC1       = item{tScName, 0, "sc1"}
 | |
| 	tstSC2       = item{tScName, 0, "sc2"}
 | |
| 	tstSC3       = item{tScName, 0, "sc3"}
 | |
| 	tstParam1    = item{tScParam, 0, "param1"}
 | |
| 	tstParam2    = item{tScParam, 0, "param2"}
 | |
| 	tstVal       = item{tScParamVal, 0, "Hello World"}
 | |
| )
 | |
| 
 | |
| var shortCodeLexerTests = []shortCodeLexerTest{
 | |
| 	{"empty", "", []item{tstEOF}},
 | |
| 	{"spaces", " \t\n", []item{{tText, 0, " \t\n"}, tstEOF}},
 | |
| 	{"text", `to be or not`, []item{{tText, 0, "to be or not"}, tstEOF}},
 | |
| 	{"no markup", `{{< sc1 >}}`, []item{tstLeftNoMD, tstSC1, tstRightNoMD, tstEOF}},
 | |
| 	{"with EOL", "{{< sc1 \n >}}", []item{tstLeftNoMD, tstSC1, tstRightNoMD, tstEOF}},
 | |
| 
 | |
| 	{"simple with markup", `{{% sc1 %}}`, []item{tstLeftMD, tstSC1, tstRightMD, tstEOF}},
 | |
| 	{"with spaces", `{{<     sc1     >}}`, []item{tstLeftNoMD, tstSC1, tstRightNoMD, tstEOF}},
 | |
| 	{"mismatched rightDelim", `{{< sc1 %}}`, []item{tstLeftNoMD, tstSC1,
 | |
| 		{tError, 0, "unrecognized character in shortcode action: U+0025 '%'. Note: Parameters with non-alphanumeric args must be quoted"}}},
 | |
| 	{"inner, markup", `{{% sc1 %}} inner {{% /sc1 %}}`, []item{
 | |
| 		tstLeftMD,
 | |
| 		tstSC1,
 | |
| 		tstRightMD,
 | |
| 		{tText, 0, " inner "},
 | |
| 		tstLeftMD,
 | |
| 		tstSCClose,
 | |
| 		tstSC1,
 | |
| 		tstRightMD,
 | |
| 		tstEOF,
 | |
| 	}},
 | |
| 	{"close, but no open", `{{< /sc1 >}}`, []item{
 | |
| 		tstLeftNoMD, {tError, 0, "got closing shortcode, but none is open"}}},
 | |
| 	{"close wrong", `{{< sc1 >}}{{< /another >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstRightNoMD, tstLeftNoMD, tstSCClose,
 | |
| 		{tError, 0, "closing tag for shortcode 'another' does not match start tag"}}},
 | |
| 	{"close, but no open, more", `{{< sc1 >}}{{< /sc1 >}}{{< /another >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstRightNoMD, tstLeftNoMD, tstSCClose, tstSC1, tstRightNoMD, tstLeftNoMD, tstSCClose,
 | |
| 		{tError, 0, "closing tag for shortcode 'another' does not match start tag"}}},
 | |
| 	{"close with extra keyword", `{{< sc1 >}}{{< /sc1 keyword>}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstRightNoMD, tstLeftNoMD, tstSCClose, tstSC1,
 | |
| 		{tError, 0, "unclosed shortcode"}}},
 | |
| 	{"Youtube id", `{{< sc1 -ziL-Q_456igdO-4 >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, {tScParam, 0, "-ziL-Q_456igdO-4"}, tstRightNoMD, tstEOF}},
 | |
| 	{"non-alphanumerics param quoted", `{{< sc1 "-ziL-.%QigdO-4" >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, {tScParam, 0, "-ziL-.%QigdO-4"}, tstRightNoMD, tstEOF}},
 | |
| 
 | |
| 	{"two params", `{{< sc1 param1   param2 >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstParam2, tstRightNoMD, tstEOF}},
 | |
| 	// issue #934
 | |
| 	{"self-closing", `{{< sc1 />}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstSCClose, tstRightNoMD, tstEOF}},
 | |
| 	// Issue 2498
 | |
| 	{"multiple self-closing", `{{< sc1 />}}{{< sc1 />}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstSCClose, tstRightNoMD,
 | |
| 		tstLeftNoMD, tstSC1, tstSCClose, tstRightNoMD, tstEOF}},
 | |
| 	{"self-closing with param", `{{< sc1 param1 />}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstSCClose, tstRightNoMD, tstEOF}},
 | |
| 	{"multiple self-closing with param", `{{< sc1 param1 />}}{{< sc1 param1 />}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstSCClose, tstRightNoMD,
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstSCClose, tstRightNoMD, tstEOF}},
 | |
| 	{"multiple different self-closing with param", `{{< sc1 param1 />}}{{< sc2 param1 />}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstSCClose, tstRightNoMD,
 | |
| 		tstLeftNoMD, tstSC2, tstParam1, tstSCClose, tstRightNoMD, tstEOF}},
 | |
| 	{"nested simple", `{{< sc1 >}}{{< sc2 >}}{{< /sc1 >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstRightNoMD,
 | |
| 		tstLeftNoMD, tstSC2, tstRightNoMD,
 | |
| 		tstLeftNoMD, tstSCClose, tstSC1, tstRightNoMD, tstEOF}},
 | |
| 	{"nested complex", `{{< sc1 >}}ab{{% sc2 param1 %}}cd{{< sc3 >}}ef{{< /sc3 >}}gh{{% /sc2 %}}ij{{< /sc1 >}}kl`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstRightNoMD,
 | |
| 		{tText, 0, "ab"},
 | |
| 		tstLeftMD, tstSC2, tstParam1, tstRightMD,
 | |
| 		{tText, 0, "cd"},
 | |
| 		tstLeftNoMD, tstSC3, tstRightNoMD,
 | |
| 		{tText, 0, "ef"},
 | |
| 		tstLeftNoMD, tstSCClose, tstSC3, tstRightNoMD,
 | |
| 		{tText, 0, "gh"},
 | |
| 		tstLeftMD, tstSCClose, tstSC2, tstRightMD,
 | |
| 		{tText, 0, "ij"},
 | |
| 		tstLeftNoMD, tstSCClose, tstSC1, tstRightNoMD,
 | |
| 		{tText, 0, "kl"}, tstEOF,
 | |
| 	}},
 | |
| 
 | |
| 	{"two quoted params", `{{< sc1 "param nr. 1" "param nr. 2" >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, {tScParam, 0, "param nr. 1"}, {tScParam, 0, "param nr. 2"}, tstRightNoMD, tstEOF}},
 | |
| 	{"two named params", `{{< sc1 param1="Hello World" param2="p2Val">}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstVal, tstParam2, {tScParamVal, 0, "p2Val"}, tstRightNoMD, tstEOF}},
 | |
| 	{"escaped quotes", `{{< sc1 param1=\"Hello World\"  >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstVal, tstRightNoMD, tstEOF}},
 | |
| 	{"escaped quotes, positional param", `{{< sc1 \"param1\"  >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstRightNoMD, tstEOF}},
 | |
| 	{"escaped quotes inside escaped quotes", `{{< sc1 param1=\"Hello \"escaped\" World\"  >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1,
 | |
| 		{tScParamVal, 0, `Hello `}, {tError, 0, `got positional parameter 'escaped'. Cannot mix named and positional parameters`}}},
 | |
| 	{"escaped quotes inside nonescaped quotes",
 | |
| 		`{{< sc1 param1="Hello \"escaped\" World"  >}}`, []item{
 | |
| 			tstLeftNoMD, tstSC1, tstParam1, {tScParamVal, 0, `Hello "escaped" World`}, tstRightNoMD, tstEOF}},
 | |
| 	{"escaped quotes inside nonescaped quotes in positional param",
 | |
| 		`{{< sc1 "Hello \"escaped\" World"  >}}`, []item{
 | |
| 			tstLeftNoMD, tstSC1, {tScParam, 0, `Hello "escaped" World`}, tstRightNoMD, tstEOF}},
 | |
| 	{"unterminated quote", `{{< sc1 param2="Hello World>}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam2, {tError, 0, "unterminated quoted string in shortcode parameter-argument: 'Hello World>}}'"}}},
 | |
| 	{"one named param, one not", `{{< sc1 param1="Hello World" p2 >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstVal,
 | |
| 		{tError, 0, "got positional parameter 'p2'. Cannot mix named and positional parameters"}}},
 | |
| 	{"one named param, one quoted positional param", `{{< sc1 param1="Hello World" "And Universe" >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1, tstVal,
 | |
| 		{tError, 0, "got quoted positional parameter. Cannot mix named and positional parameters"}}},
 | |
| 	{"one quoted positional param, one named param", `{{< sc1 "param1" param2="And Universe" >}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1,
 | |
| 		{tError, 0, "got named parameter 'param2'. Cannot mix named and positional parameters"}}},
 | |
| 	{"ono positional param, one not", `{{< sc1 param1 param2="Hello World">}}`, []item{
 | |
| 		tstLeftNoMD, tstSC1, tstParam1,
 | |
| 		{tError, 0, "got named parameter 'param2'. Cannot mix named and positional parameters"}}},
 | |
| 	{"commented out", `{{</* sc1 */>}}`, []item{
 | |
| 		{tText, 0, "{{<"}, {tText, 0, " sc1 "}, {tText, 0, ">}}"}, tstEOF}},
 | |
| 	{"commented out, missing close", `{{</* sc1 >}}`, []item{
 | |
| 		{tError, 0, "comment must be closed"}}},
 | |
| 	{"commented out, misplaced close", `{{</* sc1 >}}*/`, []item{
 | |
| 		{tText, 0, "{{<"}, {tText, 0, " sc1 >}}"}, {tError, 0, "comment ends before the right shortcode delimiter"}}},
 | |
| }
 | |
| 
 | |
| func TestShortcodeLexer(t *testing.T) {
 | |
| 	t.Parallel()
 | |
| 	for i, test := range shortCodeLexerTests {
 | |
| 		items := collect(&test)
 | |
| 		if !equal(items, test.items) {
 | |
| 			t.Errorf("[%d] %s: got\n\t%v\nexpected\n\t%v", i, test.name, items, test.items)
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func BenchmarkShortcodeLexer(b *testing.B) {
 | |
| 	b.ResetTimer()
 | |
| 	for i := 0; i < b.N; i++ {
 | |
| 		for _, test := range shortCodeLexerTests {
 | |
| 			items := collect(&test)
 | |
| 			if !equal(items, test.items) {
 | |
| 				b.Errorf("%s: got\n\t%v\nexpected\n\t%v", test.name, items, test.items)
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func collect(t *shortCodeLexerTest) (items []item) {
 | |
| 	l := newShortcodeLexer(t.name, t.input, 0)
 | |
| 	for {
 | |
| 		item := l.nextItem()
 | |
| 		items = append(items, item)
 | |
| 		if item.typ == tEOF || item.typ == tError {
 | |
| 			break
 | |
| 		}
 | |
| 	}
 | |
| 	return
 | |
| }
 | |
| 
 | |
| // no positional checking, for now ...
 | |
| func equal(i1, i2 []item) bool {
 | |
| 	if len(i1) != len(i2) {
 | |
| 		return false
 | |
| 	}
 | |
| 	for k := range i1 {
 | |
| 		if i1[k].typ != i2[k].typ {
 | |
| 			return false
 | |
| 		}
 | |
| 		if i1[k].val != i2[k].val {
 | |
| 			return false
 | |
| 		}
 | |
| 	}
 | |
| 	return true
 | |
| }
 |