github.com/kovansky/hugo@v0.92.3-0.20220224232819-63076e4ff19f/tpl/strings/regexp_test.go (about)

     1  // Copyright 2017 The Hugo Authors. All rights reserved.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  // http://www.apache.org/licenses/LICENSE-2.0
     7  //
     8  // Unless required by applicable law or agreed to in writing, software
     9  // distributed under the License is distributed on an "AS IS" BASIS,
    10  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    11  // See the License for the specific language governing permissions and
    12  // limitations under the License.
    13  
    14  package strings
    15  
    16  import (
    17  	"testing"
    18  
    19  	qt "github.com/frankban/quicktest"
    20  )
    21  
    22  func TestFindRE(t *testing.T) {
    23  	t.Parallel()
    24  	c := qt.New(t)
    25  
    26  	for _, test := range []struct {
    27  		expr    string
    28  		content interface{}
    29  		limit   interface{}
    30  		expect  interface{}
    31  	}{
    32  		{"[G|g]o", "Hugo is a static site generator written in Go.", 2, []string{"go", "Go"}},
    33  		{"[G|g]o", "Hugo is a static site generator written in Go.", -1, []string{"go", "Go"}},
    34  		{"[G|g]o", "Hugo is a static site generator written in Go.", 1, []string{"go"}},
    35  		{"[G|g]o", "Hugo is a static site generator written in Go.", "1", []string{"go"}},
    36  		{"[G|g]o", "Hugo is a static site generator written in Go.", nil, []string(nil)},
    37  		// errors
    38  		{"[G|go", "Hugo is a static site generator written in Go.", nil, false},
    39  		{"[G|g]o", t, nil, false},
    40  	} {
    41  		result, err := ns.FindRE(test.expr, test.content, test.limit)
    42  
    43  		if b, ok := test.expect.(bool); ok && !b {
    44  			c.Assert(err, qt.Not(qt.IsNil))
    45  			continue
    46  		}
    47  
    48  		c.Assert(err, qt.IsNil)
    49  		c.Check(result, qt.DeepEquals, test.expect)
    50  	}
    51  }
    52  
    53  func TestReplaceRE(t *testing.T) {
    54  	t.Parallel()
    55  	c := qt.New(t)
    56  
    57  	for _, test := range []struct {
    58  		pattern interface{}
    59  		repl    interface{}
    60  		s       interface{}
    61  		n       []interface{}
    62  		expect  interface{}
    63  	}{
    64  		{"^https?://([^/]+).*", "$1", "http://gohugo.io/docs", nil, "gohugo.io"},
    65  		{"^https?://([^/]+).*", "$2", "http://gohugo.io/docs", nil, ""},
    66  		{"(ab)", "AB", "aabbaab", nil, "aABbaAB"},
    67  		{"(ab)", "AB", "aabbaab", []interface{}{1}, "aABbaab"},
    68  		// errors
    69  		{"(ab", "AB", "aabb", nil, false}, // invalid re
    70  		{tstNoStringer{}, "$2", "http://gohugo.io/docs", nil, false},
    71  		{"^https?://([^/]+).*", tstNoStringer{}, "http://gohugo.io/docs", nil, false},
    72  		{"^https?://([^/]+).*", "$2", tstNoStringer{}, nil, false},
    73  	} {
    74  
    75  		var (
    76  			result string
    77  			err    error
    78  		)
    79  		if len(test.n) > 0 {
    80  			result, err = ns.ReplaceRE(test.pattern, test.repl, test.s, test.n...)
    81  		} else {
    82  			result, err = ns.ReplaceRE(test.pattern, test.repl, test.s)
    83  		}
    84  
    85  		if b, ok := test.expect.(bool); ok && !b {
    86  			c.Assert(err, qt.Not(qt.IsNil))
    87  			continue
    88  		}
    89  
    90  		c.Assert(err, qt.IsNil)
    91  		c.Check(result, qt.Equals, test.expect)
    92  	}
    93  }