From bd54f4eb757d373d2d50ea8451bc85c9106bc5a2 Mon Sep 17 00:00:00 2001
From: Alvaro Denis <dev0003@simelo.tech>
Date: Wed, 26 Feb 2020 18:42:00 -0500
Subject: [PATCH] add unit tests for sky-wallet derivationType ref #255

---
 .../skywallet/sky-wallet_test.go              | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/src/contrib/hardware-wallet/skywallet/sky-wallet_test.go b/src/contrib/hardware-wallet/skywallet/sky-wallet_test.go
index ca0e4812..2d092b8e 100644
--- a/src/contrib/hardware-wallet/skywallet/sky-wallet_test.go
+++ b/src/contrib/hardware-wallet/skywallet/sky-wallet_test.go
@@ -6,6 +6,7 @@ import (
 	"github.com/fibercrypto/fibercryptowallet/src/coin/mocks"
 	"github.com/fibercrypto/fibercryptowallet/src/core"
 	fce "github.com/fibercrypto/fibercryptowallet/src/errors"
+	skyWallet "github.com/fibercrypto/skywallet-go/src/skywallet"
 	messages "github.com/fibercrypto/skywallet-protob/go"
 	"github.com/gogo/protobuf/proto"
 	"github.com/stretchr/testify/require"
@@ -145,4 +146,42 @@ func TestGetFeaturesShouldHandleInvalidMsgResponse(t *testing.T) {
 
 	// Then
 	require.Error(t, err)
+}
+
+func TestVerifyDerivationTypeBip44(t *testing.T) {
+	// Giving
+	addr := &mocks.Address{}
+	addr.On("IsBip32").Return(true)
+	output := &mocks.TransactionOutput{}
+	output.On("GetAddress").Return(addr)
+	input := &mocks.TransactionInput{}
+	input.On("GetSpentOutput").Return(output)
+	inputs := []core.TransactionInput{input}
+	txn := &mocks.Transaction{}
+	txn.On("GetInputs").Return(inputs)
+
+	// When
+	dt := derivationType(txn)
+
+	// Then
+	require.Equal(t, dt, skyWallet.WalletTypeBip44)
+}
+
+func TestVerifyDerivationTypeDeterministic(t *testing.T) {
+	// Giving
+	addr := &mocks.Address{}
+	addr.On("IsBip32").Return(false)
+	output := &mocks.TransactionOutput{}
+	output.On("GetAddress").Return(addr)
+	input := &mocks.TransactionInput{}
+	input.On("GetSpentOutput").Return(output)
+	inputs := []core.TransactionInput{input}
+	txn := &mocks.Transaction{}
+	txn.On("GetInputs").Return(inputs)
+
+	// When
+	dt := derivationType(txn)
+
+	// Then
+	require.Equal(t, dt, skyWallet.WalletTypeDeterministic)
 }
\ No newline at end of file