From f9935daa8376572e8fbc686f5c37a40317f479e2 Mon Sep 17 00:00:00 2001
From: Wei Yang <richard.weiyang@gmail.com>
Date: Mon, 16 Oct 2017 21:50:56 +0800
Subject: [PATCH] base58_test: add test case for Base58

---
 base58_test.go | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)
 create mode 100644 base58_test.go

diff --git a/base58_test.go b/base58_test.go
new file mode 100644
index 0000000..5221340
--- /dev/null
+++ b/base58_test.go
@@ -0,0 +1,27 @@
+package main
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestBase58(t *testing.T) {
+	for i := 0; i < 100; i++ {
+		_, public := newKeyPair()
+		pubKeyHash := HashPubKey(public)
+
+		versionedPayload := append([]byte{version}, pubKeyHash...)
+		checksum := checksum(versionedPayload)
+
+		fullPayload := append(versionedPayload, checksum...)
+		address := Base58Encode(fullPayload)
+
+		assert.Equal(
+			t,
+			ValidateAddress(string(address[:])),
+			true,
+			"Address: %s is invalid", address,
+		)
+	}
+}