mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-25 13:07:17 +00:00
60f4b3a46d
* fix/add_4th_parameter_to_parity_list
* Change blocknum type from interface{} to rpc.BlockNumberOrHash
* Change latest block check
106 lines
3.7 KiB
Go
106 lines
3.7 KiB
Go
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/ledgerwatch/erigon/cmd/rpcdaemon/rpcdaemontest"
|
|
"github.com/ledgerwatch/erigon/common"
|
|
"github.com/ledgerwatch/erigon/common/hexutil"
|
|
"github.com/ledgerwatch/erigon/rpc"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var latestBlock = rpc.BlockNumberOrHashWithNumber(rpc.LatestBlockNumber)
|
|
|
|
func TestParityAPIImpl_ListStorageKeys_NoOffset(t *testing.T) {
|
|
assert := assert.New(t)
|
|
db := rpcdaemontest.CreateTestKV(t)
|
|
api := NewParityAPIImpl(db)
|
|
answers := []string{
|
|
"0000000000000000000000000000000000000000000000000000000000000000",
|
|
"0000000000000000000000000000000000000000000000000000000000000002",
|
|
"0a2127994676ca91e4eb3d2a1e46ec9dcee074dc2643bb5ebd4e9ac6541a3148",
|
|
"0fe673b4bc06161f39bc26f4e8831c810a72ffe69e5c8cb26f7f54752618e696",
|
|
"120e23dcb7e4437386073613853db77b10011a2404eefc716b97c7767e37f8eb",
|
|
}
|
|
addr := common.HexToAddress("0x920fd5070602feaea2e251e9e7238b6c376bcae5")
|
|
result, err := api.ListStorageKeys(context.Background(), addr, 5, nil, latestBlock)
|
|
if err != nil {
|
|
t.Errorf("calling ListStorageKeys: %v", err)
|
|
}
|
|
assert.Equal(len(answers), len(result))
|
|
for k, v := range result {
|
|
assert.Equal(answers[k], common.Bytes2Hex(v))
|
|
}
|
|
}
|
|
|
|
func TestParityAPIImpl_ListStorageKeys_WithOffset_ExistingPrefix(t *testing.T) {
|
|
assert := assert.New(t)
|
|
db := rpcdaemontest.CreateTestKV(t)
|
|
api := NewParityAPIImpl(db)
|
|
answers := []string{
|
|
"29d05770ca9ee7088a64e18c8e5160fc62c3c2179dc8ef9b4dbc970c9e51b4d8",
|
|
"29edc84535d98b29835079d685b97b41ee8e831e343cc80793057e462353a26d",
|
|
"2c05ac60f9aa2df5e64ef977f271e4b9a2d13951f123a2cb5f5d4ad5eb344f1a",
|
|
"4644be453c81744b6842ddf615d7fca0e14a23b09734be63d44c23452de95631",
|
|
"4974416255391052161ba8184fe652f3bf8c915592c65f7de127af8e637dce5d",
|
|
}
|
|
addr := common.HexToAddress("0x920fd5070602feaea2e251e9e7238b6c376bcae5")
|
|
offset := common.Hex2Bytes("29")
|
|
b := hexutil.Bytes(offset)
|
|
result, err := api.ListStorageKeys(context.Background(), addr, 5, &b, latestBlock)
|
|
if err != nil {
|
|
t.Errorf("calling ListStorageKeys: %v", err)
|
|
}
|
|
assert.Equal(len(answers), len(result))
|
|
for k, v := range result {
|
|
assert.Equal(answers[k], common.Bytes2Hex(v))
|
|
}
|
|
}
|
|
|
|
func TestParityAPIImpl_ListStorageKeys_WithOffset_NonExistingPrefix(t *testing.T) {
|
|
assert := assert.New(t)
|
|
db := rpcdaemontest.CreateTestKV(t)
|
|
api := NewParityAPIImpl(db)
|
|
answers := []string{
|
|
"4644be453c81744b6842ddf615d7fca0e14a23b09734be63d44c23452de95631",
|
|
"4974416255391052161ba8184fe652f3bf8c915592c65f7de127af8e637dce5d",
|
|
}
|
|
addr := common.HexToAddress("0x920fd5070602feaea2e251e9e7238b6c376bcae5")
|
|
offset := common.Hex2Bytes("30")
|
|
b := hexutil.Bytes(offset)
|
|
result, err := api.ListStorageKeys(context.Background(), addr, 2, &b, latestBlock)
|
|
if err != nil {
|
|
t.Errorf("calling ListStorageKeys: %v", err)
|
|
}
|
|
assert.Equal(len(answers), len(result))
|
|
for k, v := range result {
|
|
assert.Equal(answers[k], common.Bytes2Hex(v))
|
|
}
|
|
}
|
|
|
|
func TestParityAPIImpl_ListStorageKeys_WithOffset_EmptyResponse(t *testing.T) {
|
|
assert := assert.New(t)
|
|
db := rpcdaemontest.CreateTestKV(t)
|
|
api := NewParityAPIImpl(db)
|
|
addr := common.HexToAddress("0x920fd5070602feaea2e251e9e7238b6c376bcae5")
|
|
offset := common.Hex2Bytes("ff")
|
|
b := hexutil.Bytes(offset)
|
|
result, err := api.ListStorageKeys(context.Background(), addr, 2, &b, latestBlock)
|
|
if err != nil {
|
|
t.Errorf("calling ListStorageKeys: %v", err)
|
|
}
|
|
assert.Equal(0, len(result))
|
|
}
|
|
|
|
func TestParityAPIImpl_ListStorageKeys_AccNotFound(t *testing.T) {
|
|
assert := assert.New(t)
|
|
db := rpcdaemontest.CreateTestKV(t)
|
|
api := NewParityAPIImpl(db)
|
|
addr := common.HexToAddress("0x920fd5070602feaea2e251e9e7238b6c376bcaef")
|
|
_, err := api.ListStorageKeys(context.Background(), addr, 2, nil, latestBlock)
|
|
assert.Error(err, fmt.Errorf("acc not found"))
|
|
}
|