8
8
9
9
"github.com/stretchr/testify/require"
10
10
11
+ "github.com/ava-labs/avalanchego/ids"
11
12
"github.com/ava-labs/avalanchego/utils/crypto/secp256k1"
12
13
)
13
14
@@ -32,21 +33,21 @@ func TestAllocateFundedKeys(t *testing.T) {
32
33
testCases := []struct {
33
34
name string
34
35
count int
35
- expectedAddresses []string
36
+ expectedAddresses []ids. ShortID
36
37
}{{
37
38
name : "single key" ,
38
39
count : 1 ,
39
- expectedAddresses : []string {
40
- keys [4 ].Address (). String () ,
40
+ expectedAddresses : []ids. ShortID {
41
+ keys [4 ].Address (),
41
42
},
42
43
}, {
43
44
name : "multiple keys" ,
44
45
count : 4 ,
45
- expectedAddresses : []string {
46
- keys [0 ].Address (). String () ,
47
- keys [1 ].Address (). String () ,
48
- keys [2 ].Address (). String () ,
49
- keys [3 ].Address (). String () ,
46
+ expectedAddresses : []ids. ShortID {
47
+ keys [0 ].Address (),
48
+ keys [1 ].Address (),
49
+ keys [2 ].Address (),
50
+ keys [3 ].Address (),
50
51
},
51
52
}, {
52
53
name : "insufficient keys available" ,
@@ -59,9 +60,9 @@ func TestAllocateFundedKeys(t *testing.T) {
59
60
require .ErrorIs (err , errRequestedKeyCountExceedsAvailable )
60
61
} else {
61
62
require .NoError (err )
62
- addresses := []string {}
63
+ addresses := []ids. ShortID {}
63
64
for _ , key := range keys {
64
- addresses = append (addresses , key .Address (). String () )
65
+ addresses = append (addresses , key .Address ())
65
66
}
66
67
require .Equal (tc .expectedAddresses , addresses )
67
68
}
0 commit comments