From 69d560c03cea944cf48114732304e2285b7cbbb5 Mon Sep 17 00:00:00 2001
From: Dave Collins <davec@conformal.com>
Date: Wed, 13 Mar 2019 01:11:32 -0500
Subject: [PATCH] txscript: Add benchmark for GetScriptClass.

---
 txscript/bench_test.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/txscript/bench_test.go b/txscript/bench_test.go
index 52965dca..aba3d123 100644
--- a/txscript/bench_test.go
+++ b/txscript/bench_test.go
@@ -423,3 +423,18 @@ func BenchmarkGetWitnessSigOpCountNested(b *testing.B) {
 		_ = GetWitnessSigOpCount(sigScript, pkScript, witness)
 	}
 }
+
+// BenchmarkGetScriptClass benchmarks how long it takes GetScriptClass to
+// analyze a very large script.
+func BenchmarkGetScriptClass(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++ {
+		_ = GetScriptClass(script)
+	}
+}