model_paymail_addresses_test.go

package bux

import (
	"testing"

	"github.com/BuxOrg/bux/utils"
	"github.com/bitcoinschema/go-bitcoin/v2"
	"github.com/libsv/go-bk/bip32"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

// TestNewPaymail will test the method newPaymail()
func TestNewPaymail(t *testing.T) {

	t.Run("paymail basic test", func(t *testing.T) {
		ctx, client, deferMe := CreateTestSQLiteClient(t, true, false, WithAutoMigrate(&PaymailAddress{}))
		defer deferMe()

		paymail := "paymail@tester.com"

		xPub, err := bitcoin.GetHDKeyFromExtendedPublicKey(testXpubAuth)
		require.NoError(t, err)
		require.NotNil(t, xPub)

		// Get the external public key
		var paymailExternalKey *bip32.ExtendedKey
		paymailExternalKey, err = bitcoin.GetHDKeyChild(
			xPub, utils.ChainExternal,
		)
		require.NoError(t, err)
		require.NotNil(t, paymailExternalKey)

		var paymailIdentityKey *bip32.ExtendedKey
		paymailIdentityKey, err = bitcoin.GetHDKeyChild(paymailExternalKey, uint32(utils.MaxInt32))
		require.NoError(t, err)
		require.NotNil(t, paymailIdentityKey)

		paymailExternalXPub := paymailExternalKey.String()
		paymailIdentityXPub := paymailIdentityKey.String()

		p := newPaymail(
			paymail,
			WithClient(client),
			WithXPub(testXpubAuth),
			WithEncryptionKey(testEncryption),
		)
		p.PublicName = "Tester"
		p.Avatar = "img url"
		err = p.Save(ctx)
		require.NoError(t, err)

		p2 := newPaymail(
			paymail,
			WithClient(client),
			WithEncryptionKey(testEncryption),
		)
		p2.ID = "" // Remove ID (to make query work)
		conditions := map[string]interface{}{
			aliasField:  p.Alias,
			domainField: p.Domain,
		}
		err = Get(ctx, p2, conditions, false, 0, false)
		require.NoError(t, err)

		var identityKey *bip32.ExtendedKey
		identityKey, err = p2.GetIdentityXpub()
		require.NoError(t, err)
		require.NotNil(t, identityKey)

		assert.Equal(t, paymail, p2.Alias+"@"+p2.Domain)
		assert.Equal(t, "Tester", p2.PublicName)
		assert.Equal(t, "img url", p2.Avatar)
		assert.Equal(t, "d8c2bed524071d72d859caf90da5f448b5861cd4d4fd47697f94166c13c5a987", p2.XpubID)
		assert.Equal(t, paymailIdentityXPub, identityKey.String())

		// Decrypt the external key
		var decrypted string
		decrypted, err = utils.Decrypt(testEncryption, p2.ExternalXpubKey)
		require.NoError(t, err)
		assert.Equal(t, paymailExternalXPub, decrypted)
	})

	t.Run("test derive child keys", func(t *testing.T) {
		// this is used in paymail to store the derived External xpub only in the DB
		hdKey, err := utils.ValidateXPub(testXPub)
		require.NoError(t, err)

		var internal, external string
		external, internal, err = utils.DeriveAddresses(
			hdKey, 1,
		)
		require.NoError(t, err)

		addressExternal := "16fq7PmmXXbFUG5maT5Xvr2zDBUgN1xdMF"
		addressInternal := "1PQW54xMn5KA6uK7wgfzN4y7ZXMi6o7Qtm"
		assert.Equal(t, addressInternal, internal)
		assert.Equal(t, addressExternal, external)

		childKeyChain0, _ := bitcoin.GetHDKeyChild(hdKey, 0)
		childKeyChain01, _ := bitcoin.GetHDKeyChild(childKeyChain0, 1)
		key0, _ := childKeyChain01.ECPubKey()
		address0, _ := bitcoin.GetAddressFromPubKey(key0, true)
		assert.Equal(t, addressExternal, address0.AddressString)

		childKeyChain1, _ := bitcoin.GetHDKeyChild(hdKey, 1)
		childKeyChain11, _ := bitcoin.GetHDKeyChild(childKeyChain1, 1)
		key1, _ := childKeyChain11.ECPubKey()
		address1, _ := bitcoin.GetAddressFromPubKey(key1, true)
		assert.Equal(t, addressInternal, address1.AddressString)
	})
}