improve test coverage

ChainSafe · Aug 23, 2023 · 0e9bdbb · 0e9bdbb
1 parent 5bb1f74
commit 0e9bdbb
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/dot/parachain/dispute/backend_test.go b/dot/parachain/dispute/backend_test.go
@@ -3,6 +3,7 @@ package dispute
 import (
 	"sync"
 	"testing"
+	"time"
 
 	"github.com/ChainSafe/gossamer/dot/parachain/dispute/types"
 
@@ -78,6 +79,33 @@ func TestOverlayBackend_CandidateVotes(t *testing.T) {
 	require.Equal(t, candidateVotes1, candidateVotes)
 }
 
+func TestOverlayBackend_GetActiveDisputes(t *testing.T) {
+	t.Parallel()
+
+	// with
+	db, err := badger.Open(badger.DefaultOptions(t.TempDir()))
+	require.NoError(t, err)
+	disputes := btree.New(types.DisputeComparator)
+
+	dispute1, err := types.DummyDispute(1, common.Hash{1}, types.DisputeStatusActive)
+	require.NoError(t, err)
+	disputes.Set(dispute1)
+
+	dispute2, err := types.DummyDispute(2, common.Hash{2}, types.DisputeStatusActive)
+	require.NoError(t, err)
+	disputes.Set(dispute2)
+
+	// when
+	backend := newOverlayBackend(db)
+	err = backend.SetRecentDisputes(disputes)
+	require.NoError(t, err)
+
+	// then
+	activeDisputes, err := backend.GetActiveDisputes(time.Now().Unix())
+	require.NoError(t, err)
+	require.True(t, compareBTrees(disputes, activeDisputes))
+}
+
 func TestOverlayBackend_Concurrency(t *testing.T) {
 	t.Parallel()