2 files changed, 83 insertions, 0 deletions
diff --git a/tests/benchmarks/ranking-throughput/main.go b/tests/benchmarks/ranking-throughput/main.go
new file mode 100644
index 0000000..0b02dc7
--- /dev/null
+++ b/tests/benchmarks/ranking-throughput/main.go
@@ -0,0 +1,7 @@
+package main
+
+import "remembering"
+
+func main() {
+	remembering.MainTest()
+}
diff --git a/tests/benchmarks/ranking-throughput/remembering.go b/tests/benchmarks/ranking-throughput/remembering.go
new file mode 100644
index 0000000..041eaeb
--- /dev/null
+++ b/tests/benchmarks/ranking-throughput/remembering.go
@@ -0,0 +1,76 @@
+package remembering
+
+import (
+	"flag"
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+
+
+var linesFlag = flag.Int(
+	"lines",
+	1000000,
+	"The number of menu lines to rank",
+)
+
+func MainTest() {
+	flag.Parse()
+	n := *linesFlag
+
+	tmp, err := os.MkdirTemp("", "remembering-bench-")
+	if err != nil {
+		panic(err)
+	}
+	defer os.RemoveAll(tmp)
+	err = os.Setenv("XDG_DATA_HOME", tmp)
+	if err != nil {
+		panic(err)
+	}
+
+	menu := strings.Builder{}
+	for i := 0; i < n; i++ {
+		fmt.Fprintf(&menu, "item-%07d\n", i)
+	}
+
+	// half the items are already known, with spread counts
+	profile := strings.Builder{}
+	for i := 0; i < n; i += 2 {
+		fmt.Fprintf(
+			&profile, "%d profile item-%07d\n",
+			i%97, i,
+		)
+	}
+	path, err := profilePath("bench")
+	if err != nil {
+		panic(err)
+	}
+	err = os.MkdirAll(filepath.Dir(path), 0755)
+	if err != nil {
+		panic(err)
+	}
+	err = os.WriteFile(
+		path, []byte(profile.String()), 0644,
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	rc := run(envT{
+		allArgs: []string{
+			"remembering", "-p", "bench",
+			"--", "tail", "-n1",
+		},
+		in:  strings.NewReader(menu.String()),
+		out: io.Discard,
+		err: os.Stderr,
+	})
+	if rc != 0 {
+		panic("ranking failed")
+	}
+
+	fmt.Printf("ranked %d lines\n", n)
+}