From 1274c0f1911e1c80ffd1ab3e1bdc31aa0f7084eb Mon Sep 17 00:00:00 2001
From: muscle_boy <shenao.78@163.com>
Date: Tue, 12 Mar 2019 16:42:25 +0800
Subject: [PATCH] add segwit test (#1633)

* add segwit test

* format code
---
 consensus/segwit/segwit_test.go | 121 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 121 insertions(+)
 create mode 100644 consensus/segwit/segwit_test.go

diff --git a/consensus/segwit/segwit_test.go b/consensus/segwit/segwit_test.go
new file mode 100644
index 00000000..14ec13b6
--- /dev/null
+++ b/consensus/segwit/segwit_test.go
@@ -0,0 +1,121 @@
+package segwit
+
+import (
+	"encoding/hex"
+	"testing"
+)
+
+func TestConvertProgram(t *testing.T) {
+	cases := []struct {
+		desc    string
+		program string
+		script  string
+		fun     func(prog []byte) ([]byte, error)
+	}{
+		{
+			desc:    "multi sign 2-1",
+			program: "0020e402787b2bf9749f8fcdcc132a44e86bacf36780ec5df2189a11020d590533ee",
+			script:  "76aa20e402787b2bf9749f8fcdcc132a44e86bacf36780ec5df2189a11020d590533ee8808ffffffffffffffff7c00c0",
+			fun:     ConvertP2SHProgram,
+		},
+		{
+			desc:    "multi sign 5-3",
+			program: "00200824e931fb806bd77fdcd291aad3bd0a4493443a4120062bd659e64a3e0bac66",
+			script:  "76aa200824e931fb806bd77fdcd291aad3bd0a4493443a4120062bd659e64a3e0bac668808ffffffffffffffff7c00c0",
+			fun:     ConvertP2SHProgram,
+		},
+		{
+			desc:    "single sign",
+			program: "001437e1aec83a4e6587ca9609e4e5aa728db7007449",
+			script:  "76ab1437e1aec83a4e6587ca9609e4e5aa728db700744988ae7cac",
+			fun:     ConvertP2PKHSigProgram,
+		},
+	}
+
+	for i, c := range cases {
+		progBytes, err := hex.DecodeString(c.program)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		gotScript, err := c.fun(progBytes)
+		if c.script != hex.EncodeToString(gotScript) {
+			t.Errorf("case #%d (%s) got script:%s, expect script:%s", i, c.desc, hex.EncodeToString(gotScript), c.script)
+		}
+	}
+}
+
+func TestProgramType(t *testing.T) {
+	cases := []struct {
+		desc    string
+		program string
+		fun     func(prog []byte) bool
+		yes     bool
+	}{
+		{
+			desc:    "normal P2WPKHScript",
+			program: "001437e1aec83a4e6587ca9609e4e5aa728db7007449",
+			fun:     IsP2WPKHScript,
+			yes:     true,
+		},
+		{
+			desc:    "ugly P2WPKHScript",
+			program: "00200824e931fb806bd77fdcd291aad3bd0a4493443a4120062bd659e64a3e0bac66",
+			fun:     IsP2WPKHScript,
+			yes:     false,
+		},
+		{
+			desc:    "ugly P2WPKHScript",
+			program: "51",
+			fun:     IsP2WPKHScript,
+			yes:     false,
+		},
+		{
+			desc:    "normal P2WSHScript",
+			program: "00200824e931fb806bd77fdcd291aad3bd0a4493443a4120062bd659e64a3e0bac66",
+			fun:     IsP2WSHScript,
+			yes:     true,
+		},
+		{
+			desc:    "ugly P2WSHScript",
+			program: "001437e1aec83a4e6587ca9609e4e5aa728db7007449",
+			fun:     IsP2WSHScript,
+			yes:     false,
+		},
+		{
+			desc:    "ugly P2WSHScript",
+			program: "51",
+			fun:     IsP2WSHScript,
+			yes:     false,
+		},
+		{
+			desc:    "normal IsStraightforward",
+			program: "51",
+			fun:     IsStraightforward,
+			yes:     true,
+		},
+		{
+			desc:    "ugly IsStraightforward",
+			program: "001437e1aec83a4e6587ca9609e4e5aa728db7007449",
+			fun:     IsStraightforward,
+			yes:     false,
+		},
+		{
+			desc:    "ugly IsStraightforward",
+			program: "00200824e931fb806bd77fdcd291aad3bd0a4493443a4120062bd659e64a3e0bac66",
+			fun:     IsStraightforward,
+			yes:     false,
+		},
+	}
+
+	for i, c := range cases {
+		progBytes, err := hex.DecodeString(c.program)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		if c.fun(progBytes) != c.yes {
+			t.Errorf("case #%d (%s) got %t, expect %t", i, c.desc, c.fun(progBytes), c.yes)
+		}
+	}
+}
-- 
2.11.0