From d5baa6601cc19f5a5a48304e9f9c736fce07e3f2 Mon Sep 17 00:00:00 2001
From: Ricardo Domingos <ricardohsd@gmail.com>
Date: Thu, 21 Dec 2017 17:33:47 +0100
Subject: [PATCH] types: Add test for IsVoteTypeValid

---
 types/vote_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/types/vote_test.go b/types/vote_test.go
index d22b398e..51eca12d 100644
--- a/types/vote_test.go
+++ b/types/vote_test.go
@@ -99,3 +99,24 @@ func TestVoteVerifySignature(t *testing.T) {
 	valid = pubKey.VerifyBytes(newSignBytes, signature)
 	require.True(t, valid)
 }
+
+func TestIsVoteTypeValid(t *testing.T) {
+	tc := []struct {
+		name string
+		in   byte
+		out  bool
+	}{
+		{"Prevote", VoteTypePrevote, true},
+		{"Precommit", VoteTypePrecommit, true},
+		{"InvalidType", byte(3), false},
+	}
+
+	for _, tt := range tc {
+		tt := tt
+		t.Run(tt.name, func(st *testing.T) {
+			if rs := IsVoteTypeValid(tt.in); rs != tt.out {
+				t.Errorf("Got unexpected Vote type. Expected:\n%v\nGot:\n%v", rs, tt.out)
+			}
+		})
+	}
+}