package evidence import ( "testing" "time" "github.com/stretchr/testify/assert" "github.com/tendermint/tendermint/types" dbm "github.com/tendermint/tm-db" ) //------------------------------------------- func TestStoreAddDuplicate(t *testing.T) { assert := assert.New(t) db := dbm.NewMemDB() store := NewStore(db) priority := int64(10) ev := types.NewMockEvidence(2, time.Now().UTC(), 1, []byte("val1")) added := store.AddNewEvidence(ev, priority) assert.True(added) // cant add twice added = store.AddNewEvidence(ev, priority) assert.False(added) } func TestStoreCommitDuplicate(t *testing.T) { assert := assert.New(t) db := dbm.NewMemDB() store := NewStore(db) priority := int64(10) ev := types.NewMockEvidence(2, time.Now().UTC(), 1, []byte("val1")) store.MarkEvidenceAsCommitted(ev) added := store.AddNewEvidence(ev, priority) assert.False(added) } func TestStoreMark(t *testing.T) { assert := assert.New(t) db := dbm.NewMemDB() store := NewStore(db) // before we do anything, priority/pending are empty priorityEv := store.PriorityEvidence() pendingEv := store.PendingEvidence(-1) assert.Equal(0, len(priorityEv)) assert.Equal(0, len(pendingEv)) priority := int64(10) ev := types.NewMockEvidence(2, time.Now().UTC(), 1, []byte("val1")) added := store.AddNewEvidence(ev, priority) assert.True(added) // get the evidence. verify. should be uncommitted ei := store.GetInfo(ev.Height(), ev.Hash()) assert.Equal(ev, ei.Evidence) assert.Equal(priority, ei.Priority) assert.False(ei.Committed) // new evidence should be returns in priority/pending priorityEv = store.PriorityEvidence() pendingEv = store.PendingEvidence(-1) assert.Equal(1, len(priorityEv)) assert.Equal(1, len(pendingEv)) // priority is now empty store.MarkEvidenceAsBroadcasted(ev) priorityEv = store.PriorityEvidence() pendingEv = store.PendingEvidence(-1) assert.Equal(0, len(priorityEv)) assert.Equal(1, len(pendingEv)) // priority and pending are now empty store.MarkEvidenceAsCommitted(ev) priorityEv = store.PriorityEvidence() pendingEv = store.PendingEvidence(-1) assert.Equal(0, len(priorityEv)) assert.Equal(0, len(pendingEv)) // evidence should show committed newPriority := int64(0) ei = store.GetInfo(ev.Height(), ev.Hash()) assert.Equal(ev, ei.Evidence) assert.Equal(newPriority, ei.Priority) assert.True(ei.Committed) } func TestStorePriority(t *testing.T) { assert := assert.New(t) db := dbm.NewMemDB() store := NewStore(db) // sorted by priority and then height cases := []struct { ev types.MockEvidence priority int64 }{ {types.NewMockEvidence(2, time.Now().UTC(), 1, []byte("val1")), 17}, {types.NewMockEvidence(5, time.Now().UTC(), 2, []byte("val2")), 15}, {types.NewMockEvidence(10, time.Now().UTC(), 2, []byte("val2")), 13}, {types.NewMockEvidence(100, time.Now().UTC(), 2, []byte("val2")), 11}, {types.NewMockEvidence(90, time.Now().UTC(), 2, []byte("val2")), 11}, {types.NewMockEvidence(80, time.Now().UTC(), 2, []byte("val2")), 11}, } for _, c := range cases { added := store.AddNewEvidence(c.ev, c.priority) assert.True(added) } evList := store.PriorityEvidence() for i, ev := range evList { assert.Equal(ev, cases[i].ev) } }