Added and improved tests of plan types

sentinel-official · Jul 11, 2021 · 26a7427 · 26a7427
1 parent 1bc1af7
commit 26a7427
Show file tree

Hide file tree

Showing 3 changed files with 1,082 additions and 0 deletions.
diff --git a/x/plan/types/genesis_test.go b/x/plan/types/genesis_test.go
@@ -0,0 +1,45 @@
+package types
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestDefaultGenesisState(t *testing.T) {
+	var (
+		state GenesisState
+	)
+
+	state = DefaultGenesisState()
+	require.Equal(t, GenesisState(nil), state)
+}
+
+func TestNewGenesisState(t *testing.T) {
+	var (
+		plans GenesisPlans
+		state GenesisState
+	)
+
+	state = NewGenesisState(nil)
+	require.Nil(t, state)
+	require.Len(t, state, 0)
+
+	state = NewGenesisState(plans)
+	require.Equal(t, GenesisState(nil), state)
+	require.Len(t, state, 0)
+
+	plans = append(plans,
+		GenesisPlan{},
+	)
+	state = NewGenesisState(plans)
+	require.Equal(t, GenesisState(plans), state)
+	require.Len(t, state, 1)
+
+	plans = append(plans,
+		GenesisPlan{},
+	)
+	state = NewGenesisState(plans)
+	require.Equal(t, GenesisState(plans), state)
+	require.Len(t, state, 2)
+}