From 0e9bdbbfd6b9b6fbba4cdc5aed16d25aee35875b Mon Sep 17 00:00:00 2001
From: Kanishka <kanishkatn@gmail.com>
Date: Thu, 24 Aug 2023 01:34:37 +0530
Subject: [PATCH] improve test coverage

---
 dot/parachain/dispute/backend_test.go | 28 +++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/dot/parachain/dispute/backend_test.go b/dot/parachain/dispute/backend_test.go
index b8299e051d0..d1005241f27 100644
--- 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()