txscript: Add benchmark for IsPushOnlyScript.

btcsuite · Feb 5, 2021 · 1f2afc1 · 1f2afc1
1 parent 3cfdeca
commit 1f2afc1
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/txscript/bench_test.go b/txscript/bench_test.go
@@ -267,3 +267,18 @@ func BenchmarkIsMultisigSigScript(b *testing.B) {
 		}
 	}
 }
+
+// BenchmarkIsPushOnlyScript benchmarks how long it takes IsPushOnlyScript to
+// analyze a very large script.
+func BenchmarkIsPushOnlyScript(b *testing.B) {
+	script, err := genComplexScript()
+	if err != nil {
+		b.Fatalf("failed to create benchmark script: %v", err)
+	}
+
+	b.ResetTimer()
+	b.ReportAllocs()
+	for i := 0; i < b.N; i++ {
+		_ = IsPushOnlyScript(script)
+	}
+}