mirror of
https://github.com/arnaucube/hermez-node.git
synced 2026-02-07 11:26:44 +01:00
48a538faa3
- KVDB/StateDB
- Pass config parameters in a Config type instead of using many
arguments in constructor.
- Add new parameter `NoLast` which disables having an opened DB with a
checkpoint to the last batchNum for thread-safe reads. Last will be
disabled in the StateDB used by the TxSelector and BatchBuilder.
- Add new parameter `NoGapsCheck` which skips checking gaps in the list
of checkpoints and returning errors if there are gaps. Gaps check
will be disabled in the StateDB used by the TxSelector and
BatchBuilder, because we expect to have gaps when there are multiple
coordinators forging (slots not forged by our coordinator will leave
gaps).
106 lines
3.1 KiB
Go
106 lines
3.1 KiB
Go
package debugapi
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math/big"
|
|
"net/http"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/dghubble/sling"
|
|
ethCrypto "github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/hermeznetwork/hermez-node/common"
|
|
"github.com/hermeznetwork/hermez-node/db/statedb"
|
|
"github.com/iden3/go-iden3-crypto/babyjub"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func newAccount(t *testing.T, i int) *common.Account {
|
|
var sk babyjub.PrivateKey
|
|
copy(sk[:], []byte(strconv.Itoa(i))) // only for testing
|
|
pk := sk.Public()
|
|
|
|
var key ecdsa.PrivateKey
|
|
key.D = big.NewInt(int64(i + 1)) // only for testing
|
|
key.PublicKey.X, key.PublicKey.Y = ethCrypto.S256().ScalarBaseMult(key.D.Bytes())
|
|
key.Curve = ethCrypto.S256()
|
|
address := ethCrypto.PubkeyToAddress(key.PublicKey)
|
|
|
|
return &common.Account{
|
|
Idx: common.Idx(256 + i),
|
|
TokenID: common.TokenID(i),
|
|
Nonce: common.Nonce(i),
|
|
Balance: big.NewInt(1000),
|
|
BJJ: pk.Compress(),
|
|
EthAddr: address,
|
|
}
|
|
}
|
|
|
|
func TestDebugAPI(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "tmpdb")
|
|
require.Nil(t, err)
|
|
|
|
sdb, err := statedb.NewStateDB(statedb.Config{Path: dir, Keep: 128, Type: statedb.TypeSynchronizer, NLevels: 32})
|
|
require.Nil(t, err)
|
|
err = sdb.MakeCheckpoint() // Make a checkpoint to increment the batchNum
|
|
require.Nil(t, err)
|
|
|
|
addr := "localhost:12345"
|
|
// We won't test the sync/stats endpoint, so we can se the Syncrhonizer to nil
|
|
debugAPI := NewDebugAPI(addr, sdb, nil)
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
go func() {
|
|
err := debugAPI.Run(ctx)
|
|
require.Nil(t, err)
|
|
}()
|
|
|
|
var accounts []common.Account
|
|
for i := 0; i < 16; i++ {
|
|
account := newAccount(t, i)
|
|
accounts = append(accounts, *account)
|
|
_, err = sdb.CreateAccount(account.Idx, account)
|
|
require.Nil(t, err)
|
|
}
|
|
// Make a checkpoint (batchNum 2) to make the accounts available in Last
|
|
err = sdb.MakeCheckpoint()
|
|
require.Nil(t, err)
|
|
|
|
url := fmt.Sprintf("http://%v/debug/", addr)
|
|
|
|
var batchNum common.BatchNum
|
|
req, err := sling.New().Get(url).Path("sdb/batchnum").ReceiveSuccess(&batchNum)
|
|
require.Equal(t, http.StatusOK, req.StatusCode)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, common.BatchNum(2), batchNum)
|
|
|
|
var mtroot *big.Int
|
|
req, err = sling.New().Get(url).Path("sdb/mtroot").ReceiveSuccess(&mtroot)
|
|
require.Equal(t, http.StatusOK, req.StatusCode)
|
|
require.Nil(t, err)
|
|
// Testing against a hardcoded value obtained by running the test and
|
|
// printing the value previously.
|
|
assert.Equal(t, "21765339739823365993496282904432398015268846626944509989242908567129545640185",
|
|
mtroot.String())
|
|
|
|
var accountAPI common.Account
|
|
req, err = sling.New().Get(url).
|
|
Path(fmt.Sprintf("sdb/accounts/%v", accounts[0].Idx)).
|
|
ReceiveSuccess(&accountAPI)
|
|
require.Equal(t, http.StatusOK, req.StatusCode)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, accounts[0], accountAPI)
|
|
|
|
var accountsAPI []common.Account
|
|
req, err = sling.New().Get(url).Path("sdb/accounts").ReceiveSuccess(&accountsAPI)
|
|
require.Equal(t, http.StatusOK, req.StatusCode)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, accounts, accountsAPI)
|
|
|
|
cancel()
|
|
}
|