go-zero/core/stringx/node_fuzz_test.go

package stringx

import (
	"fmt"
	"math/rand"
	"strings"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func FuzzNodeFind(f *testing.F) {
	rand.Seed(time.Now().UnixNano())

	f.Add(10)
	f.Fuzz(func(t *testing.T, keys int) {
		str := Randn(rand.Intn(100) + 50)
		keywords := make(map[string]struct{})
		for i := 0; i < keys; i++ {
			keyword := Randn(rand.Intn(10) + 5)
			if !strings.Contains(str, keyword) {
				keywords[keyword] = struct{}{}
			}
		}

		size := len(str)
		var scopes []scope
		var n node
		for i := 0; i < size%20; i++ {
			start := rand.Intn(size)
			stop := start + rand.Intn(20) + 1
			if stop > size {
				stop = size
			}
			if start == stop {
				continue
			}

			keyword := str[start:stop]
			if _, ok := keywords[keyword]; ok {
				continue
			}

			keywords[keyword] = struct{}{}
			var pos int
			for pos <= len(str)-len(keyword) {
				val := str[pos:]
				p := strings.Index(val, keyword)
				if p < 0 {
					break
				}

				scopes = append(scopes, scope{
					start: pos + p,
					stop:  pos + p + len(keyword),
				})
				pos += p + 1
			}
		}

		for keyword := range keywords {
			n.add(keyword)
		}
		n.build()

		var buf strings.Builder
		buf.WriteString("keywords:\n")
		for key := range keywords {
			fmt.Fprintf(&buf, "\t%q,\n", key)
		}
		buf.WriteString("scopes:\n")
		for _, scp := range scopes {
			fmt.Fprintf(&buf, "\t{%d, %d},\n", scp.start, scp.stop)
		}
		fmt.Fprintf(&buf, "text:\n\t%s\n", str)
		defer func() {
			if r := recover(); r != nil {
				t.Errorf(buf.String())
			}
		}()
		assert.ElementsMatchf(t, scopes, n.find([]rune(str)), buf.String())
	})
}