update the TestCheckBalance and add more test cases
Some checks failed
release / docker (push) Has been cancelled
Some checks failed
release / docker (push) Has been cancelled
This commit is contained in:
parent
57426b3565
commit
a3dae12c7a
@ -1840,20 +1840,42 @@ func TestCheckBalance(t *testing.T) {
|
|||||||
name string
|
name string
|
||||||
sessionId string
|
sessionId string
|
||||||
publicKey string
|
publicKey string
|
||||||
|
alias string
|
||||||
activeSym string
|
activeSym string
|
||||||
activeBal string
|
activeBal string
|
||||||
expectedResult resource.Result
|
expectedResult resource.Result
|
||||||
expectError bool
|
expectError bool
|
||||||
}{
|
}{
|
||||||
|
{
|
||||||
|
name: "User with no active sym",
|
||||||
|
sessionId: "session123",
|
||||||
|
publicKey: "0X98765432109",
|
||||||
|
alias: "",
|
||||||
|
activeSym: "",
|
||||||
|
activeBal: "",
|
||||||
|
expectedResult: resource.Result{Content: "Balance: 0.00 \n"},
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
{
|
{
|
||||||
name: "User with active sym",
|
name: "User with active sym",
|
||||||
sessionId: "session123",
|
sessionId: "session123",
|
||||||
publicKey: "0X98765432109",
|
publicKey: "0X98765432109",
|
||||||
|
alias: "",
|
||||||
activeSym: "ETH",
|
activeSym: "ETH",
|
||||||
activeBal: "1.5",
|
activeBal: "1.5",
|
||||||
expectedResult: resource.Result{Content: "Balance: 1.50 ETH\n"},
|
expectedResult: resource.Result{Content: "Balance: 1.50 ETH\n"},
|
||||||
expectError: false,
|
expectError: false,
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
name: "User with active sym and alias",
|
||||||
|
sessionId: "session123",
|
||||||
|
publicKey: "0X98765432109",
|
||||||
|
alias: "user72",
|
||||||
|
activeSym: "SRF",
|
||||||
|
activeBal: "10.967",
|
||||||
|
expectedResult: resource.Result{Content: "user72 balance: 10.96 SRF\n"},
|
||||||
|
expectError: false,
|
||||||
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
@ -1866,13 +1888,25 @@ func TestCheckBalance(t *testing.T) {
|
|||||||
accountService: mockAccountService,
|
accountService: mockAccountService,
|
||||||
}
|
}
|
||||||
|
|
||||||
err := store.WriteEntry(ctx, tt.sessionId, storedb.DATA_ACTIVE_SYM, []byte(tt.activeSym))
|
if tt.alias != "" {
|
||||||
if err != nil {
|
err := store.WriteEntry(ctx, tt.sessionId, storedb.DATA_ACCOUNT_ALIAS, []byte(tt.alias))
|
||||||
t.Fatal(err)
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
err = store.WriteEntry(ctx, tt.sessionId, storedb.DATA_ACTIVE_BAL, []byte(tt.activeBal))
|
|
||||||
if err != nil {
|
if tt.activeSym != "" {
|
||||||
t.Fatal(err)
|
err := store.WriteEntry(ctx, tt.sessionId, storedb.DATA_ACTIVE_SYM, []byte(tt.activeSym))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if tt.activeBal != "" {
|
||||||
|
err := store.WriteEntry(ctx, tt.sessionId, storedb.DATA_ACTIVE_BAL, []byte(tt.activeBal))
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
res, err := h.CheckBalance(ctx, "check_balance", []byte(""))
|
res, err := h.CheckBalance(ctx, "check_balance", []byte(""))
|
||||||
|
Loading…
Reference in New Issue
Block a user