Skip to content

Commit

Permalink
no need for send button ack in tests as sequencer solve this internally
Browse files Browse the repository at this point in the history
  • Loading branch information
stdevAlDen committed Feb 10, 2020
1 parent 680e989 commit 8798e00
Showing 1 changed file with 3 additions and 17 deletions.
20 changes: 3 additions & 17 deletions test/integration/util/hardware-wallet.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,36 +2,22 @@ package util

import (
"github.com/fibercrypto/skywallet-go/src/skywallet"
"github.com/fibercrypto/skywallet-go/src/skywallet/wire"
messages "github.com/fibercrypto/skywallet-protob/go"
"testing"
"github.com/stretchr/testify/require"
"testing"
)

func forceWipe(t *testing.T, dev skywallet.Devicer) {
err := dev.SetAutoPressButton(true, skywallet.ButtonRight)
require.NoError(t, err)
msg, err := dev.Wipe()
require.NoError(t, err)
PressAcceptButton(t, dev, msg, messages.MessageType_MessageType_Success)
}

func PressAcceptButton(t *testing.T, dev skywallet.Devicer, prvMsg wire.Message, nextMsg messages.MessageType) wire.Message {
require.Equal(t, uint16(messages.MessageType_MessageType_ButtonRequest), prvMsg.Kind)
msg, err := dev.ButtonAck()
require.NoError(t, err)
require.Equal(t, uint16(nextMsg), msg.Kind)
if msg.Kind == uint16(messages.MessageType_MessageType_Success) {
successMsg, err := skywallet.DecodeSuccessMsg(msg)
require.NoError(t, err)
require.NotNil(t, successMsg)
}
return msg
require.Equal(t, uint16(messages.MessageType_MessageType_Success), msg.Kind)
}

func ForceSetMnemonic(t *testing.T, dev skywallet.Devicer, mnemonic string) {
forceWipe(t, dev)
msg, err := dev.SetMnemonic(mnemonic)
require.NoError(t, err)
PressAcceptButton(t, dev, msg, messages.MessageType_MessageType_Success)
require.Equal(t, uint16(messages.MessageType_MessageType_Success), msg.Kind)
}

0 comments on commit 8798e00

Please sign in to comment.