cerc-io/ipld-eth-statedb
16
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

add simple unit test cases to test behavior when slots or contracts are destructed

Browse Source
This commit is contained in:
i-norden 2023-03-07 20:41:19 -06:00
parent bd7a73afe8
commit affaeeb481
1 changed files with 145 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

169
statedb_test.go
View File

@ -28,12 +28,14 @@ var (
BlockNumber = big.NewInt(1337)
Header = types.Header{Number: BlockNumber}
BlockHash = Header.Hash()
RandomHash = crypto.Keccak256Hash([]byte("I am a random hash"))
RandomHash2 = crypto.Keccak256Hash([]byte("I am another random hash"))
RandomHash3 = crypto.Keccak256Hash([]byte("I"))
BlockParentHash = common.HexToHash("0123456701234567012345670123456701234567012345670123456701234567")
AccountPK, _ = crypto.HexToECDSA("8a1f9a8f95be41cd7ccb6168179afb4504aefe388d1e14474d32c45c72ce7b7a")
AccountAddress = crypto.PubkeyToAddress(AccountPK.PublicKey) //0x703c4b2bD70c169f5717101CaeE543299Fc946C7
AccountLeafKey = crypto.Keccak256Hash(AccountAddress.Bytes())
AccountPath = []byte{AccountLeafKey[0] & 0xf0} // first nibble of path
AccountCode = []byte{0, 1, 2, 3, 4, 5, 6, 7}
AccountCodeHash = crypto.Keccak256Hash(AccountCode)
@ -59,8 +61,12 @@ var (
AccountCodeCID, _ = util.Keccak256ToCid(ipld.RawBinary, AccountCodeHash.Bytes())
StoredValueRLP, _ = rlp.EncodeToBytes(StoredValue)
StoredValueRLP2, _ = rlp.EncodeToBytes("something")
StorageRLP, _ = rlp.EncodeToBytes(&[]interface{}{StoragePartialPath, StoredValueRLP})
StorageCID, _ = ipld.RawdataToCid(ipld.MEthStorageTrie, StorageRLP, multihash.KECCAK_256)
RemovedNodeStateCID = "baglacgzayxjemamg64rtzet6pwznzrydydsqbnstzkbcoo337lmaixmfurya"
RemovedNodeStorageCID = "bagmacgzayxjemamg64rtzet6pwznzrydydsqbnstzkbcoo337lmaixmfurya"
)
func TestSuite(t *testing.T) {
@ -87,9 +93,70 @@ func TestSuite(t *testing.T) {
require.NoError(t, tx.Commit(testCtx))
})
require.NoError(t, insertHeaderCID(pool))
require.NoError(t, insertStateCID(pool))
require.NoError(t, insertStorageCID(pool))
require.NoError(t, insertHeaderCID(pool, BlockHash.String(), BlockNumber.Uint64()))
require.NoError(t, insertHeaderCID(pool, RandomHash.String(), BlockNumber.Uint64()+1))
require.NoError(t, insertHeaderCID(pool, RandomHash2.String(), BlockNumber.Uint64()+2))
require.NoError(t, insertHeaderCID(pool, RandomHash3.String(), BlockNumber.Uint64()+3))
require.NoError(t, insertStateCID(pool, stateModel{
BlockNumber: BlockNumber.Uint64(),
BlockHash: BlockHash.String(),
LeafKey: AccountLeafKey.String(),
CID: AccountCID.String(),
Diff: true,
Balance: Account.Balance.Uint64(),
Nonce: Account.Nonce,
CodeHash: AccountCodeHash.String(),
StorageRoot: Account.Root.String(),
Removed: false,
}))
require.NoError(t, insertStateCID(pool, stateModel{
BlockNumber: BlockNumber.Uint64() + 3,
BlockHash: RandomHash3.String(),
LeafKey: AccountLeafKey.String(),
CID: RemovedNodeStorageCID,
Diff: true,
Removed: true,
}))
require.NoError(t, insertStorageCID(pool, storageModel{
BlockNumber: BlockNumber.Uint64(),
BlockHash: BlockHash.String(),
LeafKey: AccountLeafKey.String(),
StorageLeafKey: StorageLeafKey.String(),
StorageCID: StorageCID.String(),
Diff: true,
Value: StoredValueRLP,
Removed: false,
}))
require.NoError(t, insertStorageCID(pool, storageModel{
BlockNumber: BlockNumber.Uint64() + 1,
BlockHash: RandomHash.String(),
LeafKey: AccountLeafKey.String(),
StorageLeafKey: StorageLeafKey.String(),
StorageCID: RemovedNodeStateCID,
Diff: true,
Value: []byte{},
Removed: true,
}))
require.NoError(t, insertStorageCID(pool, storageModel{
BlockNumber: BlockNumber.Uint64() + 2,
BlockHash: RandomHash2.String(),
LeafKey: AccountLeafKey.String(),
StorageLeafKey: StorageLeafKey.String(),
StorageCID: StorageCID.String(),
Diff: true,
Value: StoredValueRLP2,
Removed: false,
}))
require.NoError(t, insertStorageCID(pool, storageModel{
BlockNumber: BlockNumber.Uint64() + 3,
BlockHash: RandomHash3.String(),
LeafKey: AccountLeafKey.String(),
StorageLeafKey: StorageLeafKey.String(),
StorageCID: RemovedNodeStateCID,
Diff: true,
Value: []byte{},
Removed: true,
}))
require.NoError(t, insertContractCode(pool))
db, err := statedb.NewStateDatabaseWithPool(pool)
@ -108,9 +175,33 @@ func TestSuite(t *testing.T) {
require.NoError(t, err)
require.Equal(t, &Account, acct)
acct2, err := db.StateAccount(AccountLeafKey, RandomHash)
require.NoError(t, err)
require.Equal(t, &Account, acct2)
acct3, err := db.StateAccount(AccountLeafKey, RandomHash2)
require.NoError(t, err)
require.Equal(t, &Account, acct3)
acct4, err := db.StateAccount(AccountLeafKey, RandomHash3)
require.NoError(t, err)
require.Equal(t, (*types.StateAccount)(nil), acct4)
val, err := db.StorageValue(AccountLeafKey, StorageLeafKey, BlockHash)
require.NoError(t, err)
require.Equal(t, StoredValueRLP, val)
val2, err := db.StorageValue(AccountLeafKey, StorageLeafKey, RandomHash)
require.NoError(t, err)
require.Equal(t, ([]byte)(nil), val2)
val3, err := db.StorageValue(AccountLeafKey, StorageLeafKey, RandomHash2)
require.NoError(t, err)
require.Equal(t, StoredValueRLP2, val3)
val4, err := db.StorageValue(AccountLeafKey, StorageLeafKey, RandomHash3)
require.NoError(t, err)
require.Equal(t, ([]byte)(nil), val4)
})
t.Run("StateDB", func(t *testing.T) {
@ -162,7 +253,7 @@ func TestSuite(t *testing.T) {
})
}
func insertHeaderCID(db *pgxpool.Pool) error {
func insertHeaderCID(db *pgxpool.Pool, blockHash string, blockNumber uint64) error {
sql := `INSERT INTO eth.header_cids (
block_number,
block_hash,
@ -180,8 +271,8 @@ func insertHeaderCID(db *pgxpool.Pool) error {
coinbase
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14)`
_, err := db.Exec(testCtx, sql,
BlockNumber.Uint64(),
BlockHash.String(),
blockNumber,
blockHash,
BlockParentHash.String(),
HeaderCID.String(),
0, []string{}, 0,
@ -196,7 +287,20 @@ func insertHeaderCID(db *pgxpool.Pool) error {
return err
}
func insertStateCID(db *pgxpool.Pool) error {
type stateModel struct {
BlockNumber uint64
BlockHash string
LeafKey string
CID string
Diff bool
Balance uint64
Nonce uint64
CodeHash string
StorageRoot string
Removed bool
}
func insertStateCID(db *pgxpool.Pool, cidModel stateModel) error {
sql := `INSERT INTO eth.state_cids (
block_number,
header_id,
@ -210,21 +314,32 @@ func insertStateCID(db *pgxpool.Pool) error {
removed
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)`
_, err := db.Exec(testCtx, sql,
BlockNumber.Uint64(),
BlockHash.String(),
AccountLeafKey.String(),
AccountCID.String(),
false,
Account.Balance.Uint64(),
Account.Nonce,
AccountCodeHash.String(),
Account.Root.String(),
false,
cidModel.BlockNumber,
cidModel.BlockHash,
cidModel.LeafKey,
cidModel.CID,
cidModel.Diff,
cidModel.Balance,
cidModel.Nonce,
cidModel.CodeHash,
cidModel.StorageRoot,
cidModel.Removed,
)
return err
}
func insertStorageCID(db *pgxpool.Pool) error {
type storageModel struct {
BlockNumber uint64
BlockHash string
LeafKey string
StorageLeafKey string
StorageCID string
Diff bool
Value []byte
Removed bool
}
func insertStorageCID(db *pgxpool.Pool, cidModel storageModel) error {
sql := `INSERT INTO eth.storage_cids (
block_number,
header_id,
@ -236,14 +351,14 @@ func insertStorageCID(db *pgxpool.Pool) error {
removed
) VALUES ($1, $2, $3, $4, $5, $6, $7, $8)`
_, err := db.Exec(testCtx, sql,
BlockNumber.Uint64(),
BlockHash.String(),
AccountLeafKey.String(),
StorageLeafKey.String(),
StorageCID.String(),
false,
StoredValueRLP,
false,
cidModel.BlockNumber,
cidModel.BlockHash,
cidModel.LeafKey,
cidModel.StorageLeafKey,
cidModel.StorageCID,
cidModel.Diff,
cidModel.Value,
cidModel.Removed,
)
return err
}