ballot_test.go

package election

import (
	"bsc-shamir/crypto/common"
	"math/big"
	"testing"
)

func TestBallotZero(t *testing.T) {
	params := common.DefaultParams()
	ballots := CreateBallots(5, []*big.Int{
		big.NewInt(1),
		big.NewInt(2),
		big.NewInt(3),
		big.NewInt(4),
		big.NewInt(5),
	}, big.NewInt(0))
	for i, ballot := range ballots {
		if !ballot.Verify(params) {
			t.Errorf("Failed to verify ballot %s\n", i)
		}
	}
}

func TestBallotOne(t *testing.T) {
	params := common.DefaultParams()
	ballots := CreateBallots(5, []*big.Int{
		big.NewInt(1),
		big.NewInt(2),
		big.NewInt(3),
		big.NewInt(4),
		big.NewInt(5),
	}, big.NewInt(1))
	for i, ballot := range ballots {
		if !ballot.Verify(params) {
			t.Errorf("Failed to verify ballot %s\n", i)
		}
	}
}

func TestBallotHash(t *testing.T) {
	ballots := CreateBallots(5, []*big.Int{
		big.NewInt(1),
		big.NewInt(2),
		big.NewInt(3),
		big.NewInt(4),
		big.NewInt(5),
	}, big.NewInt(1))

	h := ballots["1"].Hash()
	t.Log("Hash", h)

	for i, ballot := range ballots {
		if ballot.Hash() != h {
			t.Errorf("Ballot %s have hash %s", i, h)
		}
	}
}