109 lines
3.1 KiB
Go
109 lines
3.1 KiB
Go
|
package tests
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
"github.com/ethereum/go-ethereum/common/hexutil"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPersonal_ListAccounts(t *testing.T) {
|
||
|
|
rpcRes := call(t, "personal_listAccounts", []string{})
|
||
|
|
|
||
|
|
var res []hexutil.Bytes
|
||
|
|
err := json.Unmarshal(rpcRes.Result, &res)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, 1, len(res))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPersonal_NewAccount(t *testing.T) {
|
||
|
|
rpcRes := call(t, "personal_newAccount", []string{""})
|
||
|
|
var addr common.Address
|
||
|
|
err := json.Unmarshal(rpcRes.Result, &addr)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
rpcRes = call(t, "personal_listAccounts", []string{})
|
||
|
|
var res []hexutil.Bytes
|
||
|
|
err = json.Unmarshal(rpcRes.Result, &res)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, 2, len(res))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPersonal_Sign(t *testing.T) {
|
||
|
|
rpcRes := call(t, "personal_sign", []interface{}{hexutil.Bytes{0x88}, hexutil.Bytes(from), ""})
|
||
|
|
|
||
|
|
var res hexutil.Bytes
|
||
|
|
err := json.Unmarshal(rpcRes.Result, &res)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, 65, len(res))
|
||
|
|
// TODO: check that signature is same as with geth, requires importing a key
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPersonal_EcRecover(t *testing.T) {
|
||
|
|
data := hexutil.Bytes{0x88}
|
||
|
|
rpcRes := call(t, "personal_sign", []interface{}{data, hexutil.Bytes(from), ""})
|
||
|
|
|
||
|
|
var res hexutil.Bytes
|
||
|
|
err := json.Unmarshal(rpcRes.Result, &res)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, 65, len(res))
|
||
|
|
|
||
|
|
rpcRes = call(t, "personal_ecRecover", []interface{}{data, res})
|
||
|
|
var ecrecoverRes common.Address
|
||
|
|
err = json.Unmarshal(rpcRes.Result, &ecrecoverRes)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, from, ecrecoverRes[:])
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPersonal_UnlockAccount(t *testing.T) {
|
||
|
|
pswd := "nootwashere"
|
||
|
|
rpcRes := call(t, "personal_newAccount", []string{pswd})
|
||
|
|
var addr common.Address
|
||
|
|
err := json.Unmarshal(rpcRes.Result, &addr)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// try to sign, should be locked
|
||
|
|
_, err = callWithError("personal_sign", []interface{}{hexutil.Bytes{0x88}, addr, ""})
|
||
|
|
require.NotNil(t, err)
|
||
|
|
|
||
|
|
rpcRes = call(t, "personal_unlockAccount", []interface{}{addr, ""})
|
||
|
|
var unlocked bool
|
||
|
|
err = json.Unmarshal(rpcRes.Result, &unlocked)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, unlocked)
|
||
|
|
|
||
|
|
// try to sign, should work now
|
||
|
|
rpcRes = call(t, "personal_sign", []interface{}{hexutil.Bytes{0x88}, addr, pswd})
|
||
|
|
var res hexutil.Bytes
|
||
|
|
err = json.Unmarshal(rpcRes.Result, &res)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, 65, len(res))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPersonal_LockAccount(t *testing.T) {
|
||
|
|
pswd := "nootwashere"
|
||
|
|
rpcRes := call(t, "personal_newAccount", []string{pswd})
|
||
|
|
var addr common.Address
|
||
|
|
err := json.Unmarshal(rpcRes.Result, &addr)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
rpcRes = call(t, "personal_unlockAccount", []interface{}{addr, ""})
|
||
|
|
var unlocked bool
|
||
|
|
err = json.Unmarshal(rpcRes.Result, &unlocked)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, unlocked)
|
||
|
|
|
||
|
|
rpcRes = call(t, "personal_lockAccount", []interface{}{addr})
|
||
|
|
var locked bool
|
||
|
|
err = json.Unmarshal(rpcRes.Result, &locked)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, locked)
|
||
|
|
|
||
|
|
// try to sign, should be locked
|
||
|
|
_, err = callWithError("personal_sign", []interface{}{hexutil.Bytes{0x88}, addr, ""})
|
||
|
|
require.NotNil(t, err)
|
||
|
|
}
|