Add TestGetSignerUIDShouldFailForFailResponse ref fibercrypto#139

simelo · Oct 14, 2019 · 8b04905 · 8b04905
1 parent c4d7f5f
commit 8b04905
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/hardware/sky-wallet_test.go b/src/hardware/sky-wallet_test.go
@@ -53,6 +53,26 @@ func TestGetSignerUIDShouldFailOnDeviceError(t *testing.T) {
 	// When
 	devId := sw.GetSignerUID()
 
+	// Then
+	require.Equal(t, core.UID("undefined"), devId)
+}
+
+func TestGetSignerUIDShouldFailForFailResponse(t *testing.T) {
+	// Giving
+	dev := mocks.Devicer{}
+	f := messages.Failure{}
+	fb, err := proto.Marshal(&f)
+	require.Nil(t, err)
+	msg := wire.Message{
+		Kind: uint16(messages.MessageType_MessageType_Failure),
+		Data: fb,
+	}
+	dev.On("GetFeatures").Return(msg, nil)
+	sw := NewSkyWallet(&dev)
+
+	// When
+	devId := sw.GetSignerUID()
+
 	// Then
 	require.Equal(t, core.UID("undefined"), devId)
 }