Skip to content

Commit 857dbfc

Browse files
committed
Add test
1 parent fa7274a commit 857dbfc

1 file changed

Lines changed: 28 additions & 5 deletions

File tree

internal/accounts/accounts_test.go

Lines changed: 28 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -250,18 +250,41 @@ func Test_Create(t *testing.T) {
250250
}
251251

252252
func Test_Get(t *testing.T) {
253-
srv, _, _ := util.TestMocks(t)
253+
srv, state, _ := util.TestMocks(t)
254254

255-
t.Run("Success", func(t *testing.T) {
256-
inArgs := []string{"0x01"}
255+
t.Run("Success with address", func(t *testing.T) {
256+
// Use the emulator service account address which is always valid
257+
inArgs := []string{"f8d6e0586b0a20c7"}
257258

258259
srv.GetAccount.Run(func(args mock.Arguments) {
259260
addr := args.Get(1).(flow.Address)
260-
assert.Equal(t, "0000000000000001", addr.String())
261+
assert.Equal(t, "f8d6e0586b0a20c7", addr.String())
261262
srv.GetAccount.Return(tests.NewAccountWithAddress(inArgs[0]), nil)
262263
})
263264

264-
result, err := get(inArgs, command.GlobalFlags{}, util.NoLogger, nil, srv.Mock)
265+
result, err := get(inArgs, command.GlobalFlags{}, util.NoLogger, srv.Mock, state)
266+
assert.NoError(t, err)
267+
assert.NotNil(t, result)
268+
})
269+
270+
t.Run("Success with account name", func(t *testing.T) {
271+
testAddr := flow.HexToAddress("f8d6e0586b0a20c7")
272+
testAccount := &accounts.Account{
273+
Name: "test-account",
274+
Address: testAddr,
275+
Key: accounts.NewHexKeyFromPrivateKey(0, crypto.SHA3_256, util.GenerateTestPrivateKey()),
276+
}
277+
state.Accounts().AddOrUpdate(testAccount)
278+
279+
inArgs := []string{"test-account"}
280+
281+
srv.GetAccount.Run(func(args mock.Arguments) {
282+
addr := args.Get(1).(flow.Address)
283+
assert.Equal(t, "f8d6e0586b0a20c7", addr.String())
284+
srv.GetAccount.Return(tests.NewAccountWithAddress(addr.Hex()), nil)
285+
})
286+
287+
result, err := get(inArgs, command.GlobalFlags{}, util.NoLogger, srv.Mock, state)
265288
assert.NoError(t, err)
266289
assert.NotNil(t, result)
267290
})

0 commit comments

Comments
 (0)