chore: add unbonds generator test

QuokkaStake · Jun 22, 2024 · 9c3b1e6 · 9c3b1e6
1 parent 2da7e04
commit 9c3b1e6
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/pkg/generators/delegators_test.go → pkg/generators/delegations_test.go b/pkg/generators/delegators_test.go → pkg/generators/delegations_test.go
diff --git a/pkg/generators/unbonds_test.go b/pkg/generators/unbonds_test.go
@@ -0,0 +1,36 @@
+package generators
+
+import (
+	"main/pkg/constants"
+	"main/pkg/fetchers"
+	statePkg "main/pkg/state"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestUnbondsGeneratorNoState(t *testing.T) {
+	t.Parallel()
+
+	state := statePkg.NewState()
+	generator := NewUnbondsGenerator()
+	results := generator.Generate(state)
+	assert.Empty(t, results)
+}
+
+func TestUnbondsGeneratorNotEmptyState(t *testing.T) {
+	t.Parallel()
+
+	state := statePkg.NewState()
+	state.Set(constants.FetcherNameUnbonds, fetchers.UnbondsData{
+		Unbonds: map[string]map[string]uint64{
+			"chain": {
+				"validator": 100,
+			},
+		},
+	})
+
+	generator := NewUnbondsGenerator()
+	results := generator.Generate(state)
+	assert.NotEmpty(t, results)
+}