From 77660b7fb0e2de0fd4d159fd09e29bf4025690df Mon Sep 17 00:00:00 2001
From: Dave Collins <davec@conformal.com>
Date: Wed, 13 Mar 2019 01:12:20 -0500
Subject: [PATCH] txscript: Add benchmark for IsUnspendable.

---
 txscript/bench_test.go  | 14 ++++++++++++++
 txscript/script_test.go | 13 -------------
 2 files changed, 14 insertions(+), 13 deletions(-)

diff --git a/txscript/bench_test.go b/txscript/bench_test.go
index fda83771..c8f7d0f7 100644
--- a/txscript/bench_test.go
+++ b/txscript/bench_test.go
@@ -327,3 +327,17 @@ func BenchmarkIsNullDataScript(b *testing.B) {
 		_ = IsNullData(script)
 	}
 }
+
+// BenchmarkIsUnspendable benchmarks how long it takes IsUnspendable to analyze
+// a very large script.
+func BenchmarkIsUnspendable(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++ {
+		_ = IsUnspendable(script)
+	}
+}
diff --git a/txscript/script_test.go b/txscript/script_test.go
index 34c8ef97..665d9ef6 100644
--- a/txscript/script_test.go
+++ b/txscript/script_test.go
@@ -4334,16 +4334,3 @@ func TestIsUnspendable(t *testing.T) {
 		}
 	}
 }
-
-// BenchmarkIsUnspendable adds a benchmark to compare the time and allocations
-// necessary for the IsUnspendable function.
-func BenchmarkIsUnspendable(b *testing.B) {
-	pkScriptToUse := []byte{0xa9, 0x14, 0x82, 0x1d, 0xba, 0x94, 0xbc, 0xfb, 0xa2, 0x57, 0x36, 0xa3, 0x9e, 0x5d, 0x14, 0x5d, 0x69, 0x75, 0xba, 0x8c, 0x0b, 0x42, 0x87}
-	var res bool = false
-	for i := 0; i < b.N; i++ {
-		res = IsUnspendable(pkScriptToUse)
-	}
-	if res {
-		b.Fatalf("Benchmark should never have res be %t\n", res)
-	}
-}