You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
259 lines
7.9 KiB
259 lines
7.9 KiB
// Copyright (c) 2014 The btcsuite developers |
|
// Use of this source code is governed by an ISC |
|
// license that can be found in the LICENSE file. |
|
|
|
package btcjson_test |
|
|
|
import ( |
|
"bytes" |
|
"encoding/json" |
|
"fmt" |
|
"reflect" |
|
"testing" |
|
|
|
"github.com/btcsuite/btcd/btcjson" |
|
) |
|
|
|
// TestWalletSvrWsCmds tests all of the wallet server websocket-specific |
|
// commands marshal and unmarshal into valid results include handling of |
|
// optional fields being omitted in the marshalled command, while optional |
|
// fields with defaults have the default assigned on unmarshalled commands. |
|
func TestWalletSvrWsCmds(t *testing.T) { |
|
t.Parallel() |
|
|
|
testID := int(1) |
|
tests := []struct { |
|
name string |
|
newCmd func() (interface{}, error) |
|
staticCmd func() interface{} |
|
marshalled string |
|
unmarshalled interface{} |
|
}{ |
|
{ |
|
name: "createencryptedwallet", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("createencryptedwallet", "pass") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewCreateEncryptedWalletCmd("pass") |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"createencryptedwallet","params":["pass"],"id":1}`, |
|
unmarshalled: &btcjson.CreateEncryptedWalletCmd{Passphrase: "pass"}, |
|
}, |
|
{ |
|
name: "exportwatchingwallet", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("exportwatchingwallet") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewExportWatchingWalletCmd(nil, nil) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"exportwatchingwallet","params":[],"id":1}`, |
|
unmarshalled: &btcjson.ExportWatchingWalletCmd{ |
|
Account: nil, |
|
Download: btcjson.Bool(false), |
|
}, |
|
}, |
|
{ |
|
name: "exportwatchingwallet optional1", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("exportwatchingwallet", "acct") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewExportWatchingWalletCmd(btcjson.String("acct"), nil) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"exportwatchingwallet","params":["acct"],"id":1}`, |
|
unmarshalled: &btcjson.ExportWatchingWalletCmd{ |
|
Account: btcjson.String("acct"), |
|
Download: btcjson.Bool(false), |
|
}, |
|
}, |
|
{ |
|
name: "exportwatchingwallet optional2", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("exportwatchingwallet", "acct", true) |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewExportWatchingWalletCmd(btcjson.String("acct"), |
|
btcjson.Bool(true)) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"exportwatchingwallet","params":["acct",true],"id":1}`, |
|
unmarshalled: &btcjson.ExportWatchingWalletCmd{ |
|
Account: btcjson.String("acct"), |
|
Download: btcjson.Bool(true), |
|
}, |
|
}, |
|
{ |
|
name: "getunconfirmedbalance", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("getunconfirmedbalance") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewGetUnconfirmedBalanceCmd(nil) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"getunconfirmedbalance","params":[],"id":1}`, |
|
unmarshalled: &btcjson.GetUnconfirmedBalanceCmd{ |
|
Account: nil, |
|
}, |
|
}, |
|
{ |
|
name: "getunconfirmedbalance optional1", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("getunconfirmedbalance", "acct") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewGetUnconfirmedBalanceCmd(btcjson.String("acct")) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"getunconfirmedbalance","params":["acct"],"id":1}`, |
|
unmarshalled: &btcjson.GetUnconfirmedBalanceCmd{ |
|
Account: btcjson.String("acct"), |
|
}, |
|
}, |
|
{ |
|
name: "listaddresstransactions", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("listaddresstransactions", `["1Address"]`) |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewListAddressTransactionsCmd([]string{"1Address"}, nil) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"listaddresstransactions","params":[["1Address"]],"id":1}`, |
|
unmarshalled: &btcjson.ListAddressTransactionsCmd{ |
|
Addresses: []string{"1Address"}, |
|
Account: nil, |
|
}, |
|
}, |
|
{ |
|
name: "listaddresstransactions optional1", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("listaddresstransactions", `["1Address"]`, "acct") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewListAddressTransactionsCmd([]string{"1Address"}, |
|
btcjson.String("acct")) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"listaddresstransactions","params":[["1Address"],"acct"],"id":1}`, |
|
unmarshalled: &btcjson.ListAddressTransactionsCmd{ |
|
Addresses: []string{"1Address"}, |
|
Account: btcjson.String("acct"), |
|
}, |
|
}, |
|
{ |
|
name: "listalltransactions", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("listalltransactions") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewListAllTransactionsCmd(nil) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"listalltransactions","params":[],"id":1}`, |
|
unmarshalled: &btcjson.ListAllTransactionsCmd{ |
|
Account: nil, |
|
}, |
|
}, |
|
{ |
|
name: "listalltransactions optional", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("listalltransactions", "acct") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewListAllTransactionsCmd(btcjson.String("acct")) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"listalltransactions","params":["acct"],"id":1}`, |
|
unmarshalled: &btcjson.ListAllTransactionsCmd{ |
|
Account: btcjson.String("acct"), |
|
}, |
|
}, |
|
{ |
|
name: "recoveraddresses", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("recoveraddresses", "acct", 10) |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewRecoverAddressesCmd("acct", 10) |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"recoveraddresses","params":["acct",10],"id":1}`, |
|
unmarshalled: &btcjson.RecoverAddressesCmd{ |
|
Account: "acct", |
|
N: 10, |
|
}, |
|
}, |
|
{ |
|
name: "walletislocked", |
|
newCmd: func() (interface{}, error) { |
|
return btcjson.NewCmd("walletislocked") |
|
}, |
|
staticCmd: func() interface{} { |
|
return btcjson.NewWalletIsLockedCmd() |
|
}, |
|
marshalled: `{"jsonrpc":"1.0","method":"walletislocked","params":[],"id":1}`, |
|
unmarshalled: &btcjson.WalletIsLockedCmd{}, |
|
}, |
|
} |
|
|
|
t.Logf("Running %d tests", len(tests)) |
|
for i, test := range tests { |
|
// Marshal the command as created by the new static command |
|
// creation function. |
|
marshalled, err := btcjson.MarshalCmd(testID, test.staticCmd()) |
|
if err != nil { |
|
t.Errorf("MarshalCmd #%d (%s) unexpected error: %v", i, |
|
test.name, err) |
|
continue |
|
} |
|
|
|
if !bytes.Equal(marshalled, []byte(test.marshalled)) { |
|
t.Errorf("Test #%d (%s) unexpected marshalled data - "+ |
|
"got %s, want %s", i, test.name, marshalled, |
|
test.marshalled) |
|
continue |
|
} |
|
|
|
// Ensure the command is created without error via the generic |
|
// new command creation function. |
|
cmd, err := test.newCmd() |
|
if err != nil { |
|
t.Errorf("Test #%d (%s) unexpected NewCmd error: %v ", |
|
i, test.name, err) |
|
} |
|
|
|
// Marshal the command as created by the generic new command |
|
// creation function. |
|
marshalled, err = btcjson.MarshalCmd(testID, cmd) |
|
if err != nil { |
|
t.Errorf("MarshalCmd #%d (%s) unexpected error: %v", i, |
|
test.name, err) |
|
continue |
|
} |
|
|
|
if !bytes.Equal(marshalled, []byte(test.marshalled)) { |
|
t.Errorf("Test #%d (%s) unexpected marshalled data - "+ |
|
"got %s, want %s", i, test.name, marshalled, |
|
test.marshalled) |
|
continue |
|
} |
|
|
|
var request btcjson.Request |
|
if err := json.Unmarshal(marshalled, &request); err != nil { |
|
t.Errorf("Test #%d (%s) unexpected error while "+ |
|
"unmarshalling JSON-RPC request: %v", i, |
|
test.name, err) |
|
continue |
|
} |
|
|
|
cmd, err = btcjson.UnmarshalCmd(&request) |
|
if err != nil { |
|
t.Errorf("UnmarshalCmd #%d (%s) unexpected error: %v", i, |
|
test.name, err) |
|
continue |
|
} |
|
|
|
if !reflect.DeepEqual(cmd, test.unmarshalled) { |
|
t.Errorf("Test #%d (%s) unexpected unmarshalled command "+ |
|
"- got %s, want %s", i, test.name, |
|
fmt.Sprintf("(%T) %+[1]v", cmd), |
|
fmt.Sprintf("(%T) %+[1]v\n", test.unmarshalled)) |
|
continue |
|
} |
|
} |
|
}
|
|
|