mirror/prysm-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/prysm-pulse.git synced 2025-01-01 15:57:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

sharding: test for adding multiple headers within same period

Browse Source 
Former-commit-id: fc458cbf2960c15c0fa689b22b03614578ab8be3 [formerly 715efa2aeaf765dcab6fc87e1c3da729f2447f8a]
Former-commit-id: 420db139f99b7cb667ea3d5113d36037a88e0904
This commit is contained in:
Terence Tsao 2018-05-02 11:50:31 -07:00
parent 48a6306127
commit ea13800459
3 changed files with 66 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sharding/contracts/sharding_manager.go
View File

File diff suppressed because one or more lines are too long

4
sharding/contracts/sharding_manager.sol
View File

@ -180,6 +180,7 @@ contract SMC {
});
lastSubmittedCollation[_shardId] = block.number / PERIOD_LENGTH;
delete currentVote[_shardId];
emit HeaderAdded(_shardId, _chunkRoot, _period, msg.sender);
}
@ -253,5 +254,6 @@ contract SMC {
bytes32 mask = bytes32(0xff);
currentVote[_shardId] = (currentVote[_shardId] & ~mask) | bytes32(voteCount);
return voteCount;
}
}
}

99
sharding/contracts/sharding_manager_test.go
View File

@ -564,44 +564,27 @@ func TestContractCreation(t *testing.T) {
//}
func TestNormalAddHeader(t *testing.T) {
const notaryCount = 2
var notaryPoolAddr [notaryCount]common.Address
var notaryPoolPrivKeys [notaryCount]*ecdsa.PrivateKey
var txOpts [notaryCount]*bind.TransactOpts
genesis := make(core.GenesisAlloc)
addr := crypto.PubkeyToAddress(mainKey.PublicKey)
backend := backends.NewSimulatedBackend(core.GenesisAlloc{addr: {Balance: accountBalance2000Eth}})
txOpts := bind.NewKeyedTransactor(mainKey)
_, _, smc, _ := deploySMCContract(backend, mainKey)
// initialize back end with 2 accounts and each with 2000 eth balances
for i := 0; i < notaryCount; i++ {
key, _ := crypto.GenerateKey()
notaryPoolPrivKeys[i] = key
notaryPoolAddr[i] = crypto.PubkeyToAddress(key.PublicKey)
txOpts[i] = bind.NewKeyedTransactor(key)
genesis[notaryPoolAddr[i]] = core.GenesisAccount{
Balance: accountBalance2000Eth,
}
}
backend := backends.NewSimulatedBackend(genesis)
_, _, smc, _ := deploySMCContract(backend, notaryPoolPrivKeys[0])
// Fast forward to the next period
// Fast forward to the next period to submit header. Period 1
for i := 0; i < int(sharding.PeriodLength); i++ {
backend.Commit()
}
// register 2 notaries to SMC
for i := 0; i < notaryCount; i++ {
smc.RegisterNotary(txOpts[i])
backend.Commit()
}
// Proposer adds header consists of shard 0, period 1 and chunkroot 0xA
period1 := big.NewInt(1)
shard0 := big.NewInt(0)
chunkRoot := [32]byte{'A'}
_, err := smc.AddHeader(txOpts[0],big.NewInt(0),big.NewInt(1),chunkRoot)
_, err := smc.AddHeader(txOpts, shard0, period1, chunkRoot)
if err != nil {
t.Fatalf("Proposer adds header failed: %v", err)
}
backend.Commit()
p, err := smc.LastSubmittedCollation(&bind.CallOpts{}, big.NewInt(0))
p, err := smc.LastSubmittedCollation(&bind.CallOpts{}, shard0)
if err != nil {
t.Fatalf("Can't get last submitted collation's period number:", err)
}
@ -609,44 +592,86 @@ func TestNormalAddHeader(t *testing.T) {
t.Errorf("Incorrect last period, when header was added. Got:", p)
}
cr, err := smc.CollationRecords(&bind.CallOpts{}, big.NewInt(0), big.NewInt(1))
cr, err := smc.CollationRecords(&bind.CallOpts{}, shard0, period1)
if cr.ChunkRoot != chunkRoot {
t.Errorf("Chunkroot mismatched. Want: %v, Got: %v", chunkRoot, cr)
}
// Fast forward to the next period
// Fast forward to the next period. Period 2
for i := 0; i < int(sharding.PeriodLength); i++ {
backend.Commit()
}
// Proposer adds header consists of shard 0, period 2 and chunkroot 0xB
period2 := big.NewInt(2)
chunkRoot = [32]byte{'B'}
_, err = smc.AddHeader(txOpts[0],big.NewInt(0),big.NewInt(3),chunkRoot)
_, err = smc.AddHeader(txOpts, shard0, period2, chunkRoot)
if err != nil {
t.Fatalf("Proposer adds header failed: %v", err)
}
backend.Commit()
p, err = smc.LastSubmittedCollation(&bind.CallOpts{}, big.NewInt(0))
if p.Cmp(big.NewInt(3)) != 0 {
p, err = smc.LastSubmittedCollation(&bind.CallOpts{}, shard0)
if p.Cmp(big.NewInt(2)) != 0 {
t.Errorf("Incorrect last period, when header was added. Got:", p)
}
cr, err = smc.CollationRecords(&bind.CallOpts{}, big.NewInt(0), big.NewInt(3))
cr, err = smc.CollationRecords(&bind.CallOpts{}, shard0, period2)
if cr.ChunkRoot != chunkRoot {
t.Errorf("Chunkroot mismatched. Want: %v, Got: %v", chunkRoot, cr)
}
// Proposer adds header consists of shard 1, period 2 and chunkroot 0xC
shard1 := big.NewInt(1)
chunkRoot = [32]byte{'C'}
_, err = smc.AddHeader(txOpts[0],big.NewInt(1),big.NewInt(3),chunkRoot)
_, err = smc.AddHeader(txOpts, shard1, period2, chunkRoot)
backend.Commit()
p, err = smc.LastSubmittedCollation(&bind.CallOpts{}, big.NewInt(1))
if p.Cmp(big.NewInt(3)) != 0 {
p, err = smc.LastSubmittedCollation(&bind.CallOpts{}, shard1)
if p.Cmp(big.NewInt(2)) != 0 {
t.Errorf("Incorrect last period, when header was added. Got:", p)
}
cr, err = smc.CollationRecords(&bind.CallOpts{}, big.NewInt(1), big.NewInt(3))
cr, err = smc.CollationRecords(&bind.CallOpts{}, shard1, period2)
if cr.ChunkRoot != chunkRoot {
t.Errorf("Chunkroot mismatched. Want: %v, Got: %v", cr, chunkRoot)
}
}
func TestAddHeadersInSamePeriod(t *testing.T) {
addr := crypto.PubkeyToAddress(mainKey.PublicKey)
backend := backends.NewSimulatedBackend(core.GenesisAlloc{addr: {Balance: accountBalance2000Eth}})
txOpts := bind.NewKeyedTransactor(mainKey)
_, _, smc, _ := deploySMCContract(backend, mainKey)
// Fast forward to the next period to submit header. Period 1
for i := 0; i < int(sharding.PeriodLength); i++ {
backend.Commit()
}
// Proposer adds header consists of shard 0, period 1 and chunkroot 0xA
period1 := big.NewInt(1)
shard0 := big.NewInt(0)
chunkRoot := [32]byte{'A'}
_, err := smc.AddHeader(txOpts, shard0, period1, chunkRoot)
if err != nil {
t.Fatalf("Proposer adds header failed: %v", err)
}
backend.Commit()
p, err := smc.LastSubmittedCollation(&bind.CallOpts{}, shard0)
if p.Cmp(big.NewInt(1)) != 0 {
t.Errorf("Incorrect last period, when header was added. Got:", p)
}
cr, err := smc.CollationRecords(&bind.CallOpts{}, shard0, period1)
if cr.ChunkRoot != chunkRoot {
t.Errorf("Chunkroot mismatched. Want: %v, Got: %v", chunkRoot, cr)
}
// Proposer attemps to add another header chunkroot 0xB on the same period for the same shard
chunkRoot = [32]byte{'B'}
_, err = smc.AddHeader(txOpts, shard0, period1, chunkRoot)
if err == nil {
t.Errorf("Proposer is not allowed to add 2 headers within same period")
}
}