Hulk did something

[bytom/vapor.git] / testutil / hex.go

diff --git a/testutil/hex.go b/testutil/hex.go
new file mode 100644 (file)
index 0000000..84fbdc9
--- /dev/null
+++ b/testutil/hex.go
@@ -0,0 +1,40 @@
+package testutil
+
+import (
+       "bytes"
+       "encoding/hex"
+       "io"
+       "testing"
+
+       "github.com/vapor/protocol/bc"
+)
+
+func MustDecodeHash(s string) (h bc.Hash) {
+       if err := h.UnmarshalText([]byte(s)); err != nil {
+               panic(err)
+       }
+       return h
+}
+
+func MustDecodeHexString(s string) []byte {
+       bytes, err := hex.DecodeString(s)
+       if err != nil {
+               panic(err)
+       }
+       return bytes
+}
+
+func MustDecodeAsset(s string) (h bc.AssetID) {
+       if err := h.UnmarshalText([]byte(s)); err != nil {
+               panic(err)
+       }
+       return h
+}
+
+func Serialize(t *testing.T, wt io.WriterTo) []byte {
+       var b bytes.Buffer
+       if _, err := wt.WriteTo(&b); err != nil {
+               t.Fatal(err)
+       }
+       return b.Bytes()
+}