56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package apdu
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/status-im/hardware-wallet-go/hexutils"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestFindTag(t *testing.T) {
|
|
var (
|
|
tagData []byte
|
|
err error
|
|
)
|
|
|
|
data := hexutils.HexToBytes("C1 02 BB CC C2 04 C3 02 11 22 C3 02 88 99")
|
|
|
|
tagData, err = FindTag(data, uint8(0xC1))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "BB CC", hexutils.BytesToHexWithSpaces(tagData))
|
|
|
|
tagData, err = FindTag(data, uint8(0xC2))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "C3 02 11 22", hexutils.BytesToHexWithSpaces(tagData))
|
|
|
|
tagData, err = FindTag(data, uint8(0xC3))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "88 99", hexutils.BytesToHexWithSpaces(tagData))
|
|
|
|
tagData, err = FindTag(data, uint8(0xC2), uint8(0xC3))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "11 22", hexutils.BytesToHexWithSpaces(tagData))
|
|
|
|
// tag not found
|
|
data = hexutils.HexToBytes("C1 00")
|
|
_, err = FindTag(data, uint8(0xC2))
|
|
assert.Equal(t, &ErrTagNotFound{uint8(0xC2)}, err)
|
|
|
|
// sub-tag not found
|
|
data = hexutils.HexToBytes("C1 02 C2 00")
|
|
_, err = FindTag(data, uint8(0xC1), uint8(0xC3))
|
|
assert.Equal(t, &ErrTagNotFound{uint8(0xC3)}, err)
|
|
}
|
|
|
|
func TestFindTagN(t *testing.T) {
|
|
data := hexutils.HexToBytes("0A 01 A1 0A 01 A2")
|
|
|
|
tagData, err := FindTagN(data, 0, uint8(0x0A))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "A1", hexutils.BytesToHexWithSpaces(tagData))
|
|
|
|
tagData, err = FindTagN(data, 1, uint8(0x0A))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "A2", hexutils.BytesToHexWithSpaces(tagData))
|
|
}
|