94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
package vaa
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCalculateQuorum(t *testing.T) {
|
|
type Test struct {
|
|
numGuardians int
|
|
quorumResult int
|
|
shouldPanic bool
|
|
}
|
|
|
|
tests := []Test{
|
|
// Positive Test Cases
|
|
{numGuardians: 0, quorumResult: 1},
|
|
{numGuardians: 1, quorumResult: 1},
|
|
{numGuardians: 2, quorumResult: 2},
|
|
{numGuardians: 3, quorumResult: 3},
|
|
{numGuardians: 4, quorumResult: 3},
|
|
{numGuardians: 5, quorumResult: 4},
|
|
{numGuardians: 6, quorumResult: 5},
|
|
{numGuardians: 7, quorumResult: 5},
|
|
{numGuardians: 8, quorumResult: 6},
|
|
{numGuardians: 9, quorumResult: 7},
|
|
{numGuardians: 10, quorumResult: 7},
|
|
{numGuardians: 11, quorumResult: 8},
|
|
{numGuardians: 12, quorumResult: 9},
|
|
{numGuardians: 13, quorumResult: 9},
|
|
{numGuardians: 14, quorumResult: 10},
|
|
{numGuardians: 15, quorumResult: 11},
|
|
{numGuardians: 16, quorumResult: 11},
|
|
{numGuardians: 17, quorumResult: 12},
|
|
{numGuardians: 18, quorumResult: 13},
|
|
{numGuardians: 19, quorumResult: 13},
|
|
{numGuardians: 50, quorumResult: 34},
|
|
{numGuardians: 100, quorumResult: 67},
|
|
{numGuardians: 1000, quorumResult: 667},
|
|
|
|
// Negative Test Cases
|
|
{numGuardians: -1, quorumResult: 1, shouldPanic: true},
|
|
{numGuardians: -1000, quorumResult: 1, shouldPanic: true},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run("", func(t *testing.T) {
|
|
if tc.shouldPanic {
|
|
assert.Panics(t, func() { CalculateQuorum(tc.numGuardians) }, "The code did not panic")
|
|
} else {
|
|
num := CalculateQuorum(tc.numGuardians)
|
|
assert.Equal(t, tc.quorumResult, num)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func FuzzCalculateQuorum(f *testing.F) {
|
|
// Add examples to our fuzz corpus
|
|
f.Add(1)
|
|
f.Add(2)
|
|
f.Add(4)
|
|
f.Add(8)
|
|
f.Add(16)
|
|
f.Add(32)
|
|
f.Add(64)
|
|
f.Add(128)
|
|
f.Fuzz(func(t *testing.T, numGuardians int) {
|
|
// These are known cases, which the implementation will panic on and/or we have explicit
|
|
// unit-test coverage of above, so we can safely ignore it in our fuzz testing
|
|
if numGuardians <= 0 {
|
|
t.Skip()
|
|
}
|
|
|
|
// Let's determine how many guardians are needed for quorum
|
|
num := CalculateQuorum(numGuardians)
|
|
|
|
// Let's always be sure that there are enough guardians to maintain quorum
|
|
assert.LessOrEqual(t, num, numGuardians, "fuzz violation: quorum cannot be achieved because we require more guardians than we have")
|
|
|
|
// Let's always be sure that num is never zero
|
|
assert.NotZero(t, num, "fuzz violation: no guardians are required to achieve quorum")
|
|
|
|
var floorFloat float64 = 0.66666666666666666
|
|
numGuardiansFloat := float64(numGuardians)
|
|
numFloat := float64(num)
|
|
actualFloat := numFloat / numGuardiansFloat
|
|
|
|
// Let's always make sure that the int division does not violate the floor of our float division
|
|
assert.Greater(t, actualFloat, floorFloat, "fuzz violation: quorum has dropped below 2/3rds threshold")
|
|
})
|
|
}
|