addrgen_test.go

package addrgen

import (
	"reflect"
	"testing"
)

const (
	zpub     = "zpub6mppVZRwQ6zgSK6mpetXfnc9y3VQcuUb1eLoRCrZQGZDbFSNrrmNB3khX3AFYaQxRzSGU7hTjPoftt9D25VfuM9vpDfSdk5XopngBQWmKyG"
	ypub     = "ypub6X1q8RUQjFhN5faTnwmZ7ANUcHWXxEE2DWTDf8dtzJFo4gCioocNo2fAyjk13skADsx9FzRNQuxXQAdK2WWNRB5Cifzvps1h7G4z4NyurDH"
	xpub     = "xpub661MyMwAqRbcEgw6XLyjPzE4qq3ACgWapJBEzRoBGwHdtxdbTxKn5WUVNx4zpmt6RTqTXQ6XYA6AAEJJf5RNytMmfJgmaWef5XBftAr48ut"
	xpubETH  = "xpub6Do3SF4NSy8AHDEpJ5WLP2jpgGwMw7wtrL8KoFvYXp7VCUEbL39XWcefzpNVLoz7DmDKch9nFi8xpfrmyH4yitmScrc2fWseytKDfXJrKYK"
	xpubTron = "xpub6F7xFHbo5T4PAT4iQjuU3stbhbnTaY9M13hkdTi2PkyqeM5AchV1d4y3MNWZtgTUn1CsP6AS2U1qnMb8BNwNmN5ypFYFVSocyUpVjM5M7KA"

	xpubDash = "xpub6CA5bpUeh1Qx8G9vHi1GdH7ZXqAfUwvquNHYKmDd8ewWrsWPuyNfBDPRxR5GgZyF1nDJMZ1ZdBNLvx5b2e3Lj1CzUXZ6avVv4gmr5womkwD"
	tpubDash = "tpubDCxj2nRzABz1oMykSf9mKZqvbXKFR3bjkjndVLz9RF5jbnhuUQtvchH8rhXirN3nEJmTnpAo6XZUiA4cSDWxdU4kvmaCjtUMykPCskTbVD3"

	tpub1 = "tpubDCTeyz8KLiDVwexXoYmfDSMRmooGoMrKKrrhXJMhZxtWqm63Y6dbaDaYaEd99dgp6w2b9miDEK6Z7f1qcmbCshEkx7WMgJGkVJtDCdiEarh"
	tpub2 = "tpubD6NzVbkrYhZ4WV7wyXxpbuZSAx8pC51eMvmYLyrKcr3XeFJJYBAsDx9Kcz9nMGqLDNNN8ycQ5FvCrQnv3wGcsKp4iuS5G3JtfUn5e9Uujfz"
	tpub  = "tpubDCFUZ43iCUBBTJnMv2nHT4wTuNpDwguV4pbCgDKxbRijNF4N9fWGXNARD22w2AcjHDnzs9SkriSwHS5piRVm91tNMtJywpJwuEY1pt2ioFD"
	upub  = "upub5DP47pDH39bHdd29xmSKctKb4MHE8SRvmrSEyUbA7VqCTewyKUAVWjXAuN5PVwhjGPug199CwdZEwPkXAUQUrsPGhGUQWZCFW6kfu6aNZb1"
	vpub  = "vpub5Vm8JiyeMgCWT2SqgFkoJyaovNQH8RCF3wAUKCrFAfRdVujdYubBrYUGtggtabj71XxvUQuS5r9AgT4VhGvax9gXEpdi9XBg7jHnvm1WDii"
	vpub1 = "vpub5VuZzHLfdqMTX3uRj42fDUnhxey8dQqWfETPw5DTj1tg4DmQjD7VpUTsJCNAkVQVWpozJQoHuxmDuQ1JjrviGgJDtUZEzqDKZFeEmKhUXvG"
)

func TestGenerate(t *testing.T) {

	tests := []struct {
		name    string
		pubKey  string
		index   int
		result  string
		network Network
	}{
		{"xpub", xpub, 0, "1AGRgsh1Hu7ATwiPDUibdj933gz8P2pMho", "BTC"},
		{"ypub", ypub, 0, "3FcBUBCSataU7tSZYQqYfran6jjWtLWUYr", "BTC"},
		{"zpub", zpub, 0, "bc1qswkdw6e7a7ydr8rcv7duucxaugf82mhghj3a6n", "BTC"},
		{"tpub1", tpub1, 0, "mkKNecT6fGfzb8iL9YmjMDHbaL87eLjFFz", "BTC"},
		{"tpub2", tpub2, 0, "mpnNyvmz6vYRF4Bzw3gyTeMMugaqJPSY5w", "BTC"},
		{"tpub", tpub, 0, "mmVn5JSRUZcWgxJJnAVSX7rGxt1fwbaXb3", "BTC"},
		{"upub", upub, 0, "2N3HSFUtaaqm5aaWboxup3uppGp4Z7HrcCP", "BTC"},
		{"vpub", vpub, 0, "tb1qxdz5xktump2xt0832tgqnlhf48jrarulddvaym", "BTC"},
		{"vpub1", vpub1, 0, "tb1qft2am6e3cxg7s8lxwmc0kfyfsunzktq9zlnjj2", "BTC"},

		{"xpub", xpubETH, 0, "0x528c72Db041F83b701D47c31803672fB1B1c8c28", "ETH"},
		{"xpub", xpubETH, 10, "0x8ECda97e617db0F41fa5842269250EB7Fe0E7112", "ETH"},
		{"xpub", xpubETH, 1, "0x35AC04b5a432Fb82785Af2f14D5723D1ECa320fa", "ETH"},

		{"xpub", xpubDash, 0, "XmrveyPEvUorJG3BW8Fc9YqhrqTPiCvCwX", "DASH"},
		{"xpub", xpubDash, 9, "XcwjjjQcHcybCqqpPQf993zRuU89cAFb1f", "DASH"},
		{"tpub", tpubDash, 10, "ydXVxJZgCcHvSFaESkdmbpoK4AC91d8qwy", "DASH"},
		{"tpub", tpubDash, 1, "yWhXAWwpYwh5yfvaTcFiSs2t5j844JMGLT", "DASH"},

		{"xpub", xpubTron, 0, "TMR2H6jLfCaoyeNzXx8aryMgQCB3CY9kQ6", TRON},
		{"xpub", xpubTron, 10, "TR8WHeu5Yx7fxFoVBMrdh6S6cTmGN4PZPw", TRON},
		{"xpub", xpubTron, 1, "TVved1mnMDx7eVFEwcvC2xiR7y1KJPNjjA", TRON},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if address, _ := Generate(tt.network, tt.pubKey, tt.index); !reflect.DeepEqual(address, tt.result) {
				t.Errorf("Generate() = %v, want %v", address, tt.result)
			}
		})
	}
}