summaryrefslogtreecommitdiff
path: root/agreement/asyncVoteVerifier_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'agreement/asyncVoteVerifier_test.go')
-rw-r--r--agreement/asyncVoteVerifier_test.go50
1 files changed, 50 insertions, 0 deletions
diff --git a/agreement/asyncVoteVerifier_test.go b/agreement/asyncVoteVerifier_test.go
new file mode 100644
index 000000000..6cfadedd8
--- /dev/null
+++ b/agreement/asyncVoteVerifier_test.go
@@ -0,0 +1,50 @@
+// Copyright (C) 2019-2021 Algorand, Inc.
+// This file is part of go-algorand
+//
+// go-algorand is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as
+// published by the Free Software Foundation, either version 3 of the
+// License, or (at your option) any later version.
+//
+// go-algorand is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with go-algorand. If not, see <https://www.gnu.org/licenses/>.
+
+package agreement
+
+import (
+ "context"
+ "testing"
+
+ "github.com/stretchr/testify/require"
+
+ "github.com/algorand/go-algorand/test/partitiontest"
+ "github.com/algorand/go-algorand/util/execpool"
+)
+
+type expiredExecPool struct {
+ execpool.ExecutionPool
+}
+
+func (fp *expiredExecPool) EnqueueBacklog(enqueueCtx context.Context, t execpool.ExecFunc, arg interface{}, out chan interface{}) error {
+ // generate an error, to see if we correctly report that on the verifyVote() call.
+ return context.Canceled
+}
+
+// Test async vote verifier against a full execution pool.
+func TestVerificationAgainstFullExecutionPool(t *testing.T) {
+ partitiontest.PartitionTest(t)
+ mainPool := execpool.MakePool(t)
+ defer mainPool.Shutdown()
+
+ voteVerifier := MakeAsyncVoteVerifier(&expiredExecPool{mainPool})
+ defer voteVerifier.Quit()
+ verifyErr := voteVerifier.verifyVote(context.Background(), nil, unauthenticatedVote{}, 0, message{}, make(chan<- asyncVerifyVoteResponse, 1))
+ require.Error(t, context.Canceled, verifyErr)
+ verifyEqVoteErr := voteVerifier.verifyEqVote(context.Background(), nil, unauthenticatedEquivocationVote{}, 0, message{}, make(chan<- asyncVerifyVoteResponse, 1))
+ require.Error(t, context.Canceled, verifyEqVoteErr)
+}