1 files changed, 62 insertions, 0 deletions
diff --git a/lib/diskio/kmp_test.go b/lib/diskio/kmp_test.go
new file mode 100644
index 0000000..6c6ef78
--- /dev/null
+++ b/lib/diskio/kmp_test.go
@@ -0,0 +1,62 @@
+// Copyright (C) 2022  Luke Shumaker <lukeshu@lukeshu.com>
+//
+// SPDX-License-Identifier: GPL-2.0-or-later
+
+package diskio
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBuildKMPTable(t *testing.T) {
+	substr := []byte("ababaa")
+	table := buildKMPTable(substr)
+	assert.Equal(t,
+		[]int{0, 0, 1, 2, 3, 1},
+		table)
+	for j, val := range table {
+		matchLen := j + 1
+		assert.Equalf(t, substr[:val], substr[matchLen-val:matchLen],
+			"for table[%d]=%d", j, val)
+	}
+}
+
+func FuzzBuildKMPTable(f *testing.F) {
+	f.Add([]byte("ababaa"))
+	f.Fuzz(func(t *testing.T, substr []byte) {
+		table := buildKMPTable(substr)
+		assert.Equal(t, len(substr), len(table), "length")
+		for j, val := range table {
+			matchLen := j + 1
+			assert.Equalf(t, substr[:val], substr[matchLen-val:matchLen],
+				"for table[%d]=%d", j, val)
+		}
+	})
+}
+
+func NaiveFindAll(str, substr []byte) []int64 {
+	var matches []int64
+	for i := range str {
+		if bytes.HasPrefix(str[i:], substr) {
+			matches = append(matches, int64(i))
+		}
+	}
+	return matches
+}
+
+func FuzzFindAll(f *testing.F) {
+	f.Fuzz(func(t *testing.T, str, substr []byte) {
+		if len(substr) == 0 {
+			t.Skip()
+		}
+		t.Logf("str   =%q", str)
+		t.Logf("substr=%q", substr)
+		exp := NaiveFindAll(str, substr)
+		act, err := FindAll(bytes.NewReader(str), substr)
+		assert.NoError(t, err)
+		assert.Equal(t, exp, act)
+	})
+}