chore: add more tests for proposal

QuokkaStake · May 5, 2024 · 7f7b251 · 7f7b251
1 parent 8bf31df
commit 7f7b251
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/pkg/types/proposal_test.go b/pkg/types/proposal_test.go
@@ -16,3 +16,22 @@ func TestProposalGetVotingTime(t *testing.T) {
 
 	assert.Equal(t, "1 hour 1 minute", proposal.GetTimeLeft(), "Wrong value!")
 }
+
+func TestProposalInVoting(t *testing.T) {
+	t.Parallel()
+
+	assert.True(t, Proposal{Status: ProposalStatusVoting}.IsInVoting())
+	assert.False(t, Proposal{Status: ProposalStatusPassed}.IsInVoting())
+}
+
+func TestProposalStatusSerialize(t *testing.T) {
+	t.Parallel()
+
+	assert.Equal(t, "🗳️Voting", Proposal{Status: ProposalStatusVoting}.SerializeStatus())
+	assert.Equal(t, "💸Deposit", Proposal{Status: ProposalStatusDeposit}.SerializeStatus())
+	assert.Equal(t, "🙌 Passed", Proposal{Status: ProposalStatusPassed}.SerializeStatus())
+	assert.Equal(t, "🙅‍Rejected", Proposal{Status: ProposalStatusRejected}.SerializeStatus())
+	assert.Equal(t, "🤦‍Failed", Proposal{Status: ProposalStatusFailed}.SerializeStatus())
+	assert.Equal(t, "test", Proposal{Status: ProposalStatus("test")}.SerializeStatus())
+
+}