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

sharding/contracts: clean up

Browse Source 
Former-commit-id: 3f42edb35e132531dc784846add3eb760f7fef3d [formerly 52d0ffcfac31d95aecc80ac94ff5d9fb2ccdfb68]
Former-commit-id: 4e868f4bf6399ea2b6185eb90adedd1e757d5c5f
This commit is contained in:
Terence Tsao 2018-05-17 10:30:46 -07:00
parent ab04043fcb
commit b36de30334
1 changed files with 4 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
sharding/contracts/sharding_manager_test.go
View File

@ -89,8 +89,8 @@ func (a testAcconts) registerNotaries(t *testing.T, backend *backends.SimulatedB
return nil return nil
} }
// addHeader is a helper function to add header and verify // addHeader is a helper function to add header to smc
// header is correctly added to the SMC // it also verifies header is correctly added to the SMC
func (a testAccount) addHeader(t *testing.T, backend *backends.SimulatedBackend, smc *SMC, shard *big.Int, period *big.Int, chunkRoot uint8) error { func (a testAccount) addHeader(t *testing.T, backend *backends.SimulatedBackend, smc *SMC, shard *big.Int, period *big.Int, chunkRoot uint8) error {
_, err := smc.AddHeader(a.txOpts, shard, period, [32]byte{chunkRoot}) _, err := smc.AddHeader(a.txOpts, shard, period, [32]byte{chunkRoot})
if err != nil { if err != nil {
@ -114,7 +114,7 @@ func (a testAccount) addHeader(t *testing.T, backend *backends.SimulatedBackend,
} }
// addHeader is a helper function for notary to submit vote on a given header // addHeader is a helper function for notary to submit vote on a given header
// it also verifies vote is correctly submitted // it also verifies vote is properly submitted and casted
func (a testAccount) submitVote(t *testing.T, backend *backends.SimulatedBackend, smc *SMC, shard *big.Int, period *big.Int, index *big.Int, chunkRoot uint8) error { func (a testAccount) submitVote(t *testing.T, backend *backends.SimulatedBackend, smc *SMC, shard *big.Int, period *big.Int, index *big.Int, chunkRoot uint8) error {
_, err := smc.SubmitVote(a.txOpts, shard, period, index, [32]byte{chunkRoot}) _, err := smc.SubmitVote(a.txOpts, shard, period, index, [32]byte{chunkRoot})
if err != nil { if err != nil {
@ -122,20 +122,6 @@ func (a testAccount) submitVote(t *testing.T, backend *backends.SimulatedBackend
} }
backend.Commit() backend.Commit()
p, err := smc.LastSubmittedCollation(&bind.CallOpts{}, shard)
if err != nil {
t.Fatalf("Can't get last submitted collation's period number: %v", err)
}
if p.Cmp(period) != 0 {
t.Errorf("Incorrect last period, when header was added. Got: %v", p)
}
cr, err := smc.CollationRecords(&bind.CallOpts{}, shard, period)
if cr.ChunkRoot != [32]byte{chunkRoot} {
t.Errorf("Chunkroot mismatched. Want: %v, Got: %v", chunkRoot, cr)
}
return nil
v, err := smc.HasVoted(&bind.CallOpts{}, shard, index) v, err := smc.HasVoted(&bind.CallOpts{}, shard, index)
if err != nil { if err != nil {
t.Fatalf("Check notary's vote failed: %v", err) t.Fatalf("Check notary's vote failed: %v", err)
@ -299,8 +285,7 @@ func TestNotaryDeregisterThenRegister(t *testing.T) {
} }
// Notary 0 re-registers again. // Notary 0 re-registers again.
smc.RegisterNotary(notaryPool[0].txOpts) notaryPool.registerNotaries(t, backend, smc, notaryDeposit, 0, 1)
backend.Commit()
numNotaries, _ = smc.NotaryPoolLength(&bind.CallOpts{}) numNotaries, _ = smc.NotaryPoolLength(&bind.CallOpts{})
if numNotaries.Cmp(big.NewInt(0)) != 0 { if numNotaries.Cmp(big.NewInt(0)) != 0 {