chore: add tests

QuokkaStake · May 5, 2024 · 9b66a67 · 9b66a67
1 parent 0cddab6
commit 9b66a67
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 5 deletions.
diff --git a/pkg/report/generator_test.go b/pkg/report/generator_test.go
@@ -81,6 +81,58 @@ func TestReportGeneratorWithVoteError(t *testing.T) {
 	assert.Equal(t, "proposal", entry.Proposal.ID, "Proposal ID mismatch!")
 }
 
+func TestReportGeneratorWithProposalNotInVotingPeriod(t *testing.T) {
+	t.Parallel()
+
+	stateManager := state.NewStateManager("./state.json", &fs.TestFS{}, logger.GetNopLogger())
+
+	oldState := state.State{
+		ChainInfos: map[string]*state.ChainInfo{
+			"chain": {
+				ProposalVotes: map[string]state.WalletVotes{
+					"proposal": {
+						Proposal: types.Proposal{
+							ID:     "proposal",
+							Status: types.ProposalStatusPassed,
+						},
+						Votes: map[string]state.ProposalVote{
+							"wallet": {},
+						},
+					},
+				},
+			},
+		},
+	}
+	newState := state.State{
+		ChainInfos: map[string]*state.ChainInfo{
+			"chain": {
+				ProposalVotes: map[string]state.WalletVotes{
+					"proposal": {
+						Proposal: types.Proposal{
+							ID:     "proposal",
+							Status: types.ProposalStatusPassed,
+						},
+						Votes: map[string]state.ProposalVote{
+							"wallet": {
+								Vote: &types.Vote{
+									Options: []types.VoteOption{{Option: "YES", Weight: 1}},
+								},
+							},
+						},
+					},
+				},
+			},
+		},
+	}
+
+	generator := NewReportGenerator(stateManager, logger.GetNopLogger(), types.Chains{
+		&types.Chain{Name: "chain"},
+	})
+
+	report := generator.GenerateReport(oldState, newState)
+	assert.Empty(t, report.Entries)
+}
+
 func TestReportGeneratorWithNotVoted(t *testing.T) {
 	t.Parallel()
 

diff --git a/pkg/state/generator_test.go b/pkg/state/generator_test.go
@@ -58,7 +58,7 @@ func TestReportGeneratorProcessProposalWithError(t *testing.T) {
 
 	oldVotes := map[string]WalletVotes{
 		"1": {
-			Proposal: types.Proposal{ID: "1"},
+			Proposal: types.Proposal{ID: "1", Status: types.ProposalStatusVoting},
 		},
 	}
 
@@ -103,7 +103,7 @@ func TestReportGeneratorProcessProposalWithoutError(t *testing.T) {
 
 	oldVotes := map[string]WalletVotes{
 		"1": {
-			Proposal: types.Proposal{ID: "1"},
+			Proposal: types.Proposal{ID: "1", Status: types.ProposalStatusVoting},
 		},
 	}
 
@@ -139,7 +139,7 @@ func TestReportGeneratorProcessVoteWithError(t *testing.T) {
 	chains := types.Chains{chain}
 	fetcher := &fetchers.TestFetcher{WithProposals: true, WithVoteError: true}
 
-	proposal := types.Proposal{ID: "1"}
+	proposal := types.Proposal{ID: "1", Status: types.ProposalStatusVoting}
 	generator := Generator{
 		Logger: *log,
 		Chains: chains,
@@ -196,7 +196,7 @@ func TestReportGeneratorProcessVoteWithDisappearedVote(t *testing.T) {
 	chains := types.Chains{chain}
 	fetcher := &fetchers.TestFetcher{WithProposals: true}
 
-	proposal := types.Proposal{ID: "1"}
+	proposal := types.Proposal{ID: "1", Status: types.ProposalStatusVoting}
 	generator := Generator{
 		Logger: *log,
 		Chains: chains,
@@ -253,7 +253,7 @@ func TestReportGeneratorProcessVoteWithOkVote(t *testing.T) {
 	chains := types.Chains{chain}
 	fetcher := &fetchers.TestFetcher{WithProposals: true, WithVote: true}
 
-	proposal := types.Proposal{ID: "1"}
+	proposal := types.Proposal{ID: "1", Status: types.ProposalStatusVoting}
 	generator := Generator{
 		Logger: *log,
 		Chains: chains,