// (c) 2019-2020, Ava Labs, Inc. All rights reserved.
// See the file LICENSE for licensing terms.

package snowball

import (
	"math/rand"
	"testing"

	"github.com/prometheus/client_golang/prometheus"
)

func TestSnowballOptimized(t *testing.T) {
	numColors := 10
	numNodes := 100
	params := Parameters{
		Metrics: prometheus.NewRegistry(),
		K:       20, Alpha: 15, BetaVirtuous: 20, BetaRogue: 30,
	}
	seed := int64(0)

	nBitwise := Network{}
	nBitwise.Initialize(params, numColors)

	nNaive := nBitwise

	rand.Seed(seed)
	for i := 0; i < numNodes; i++ {
		nBitwise.AddNode(&Tree{})
	}

	rand.Seed(seed)
	for i := 0; i < numNodes; i++ {
		nNaive.AddNode(&Flat{})
	}

	numRounds := 0
	for !nBitwise.Finalized() && !nBitwise.Disagreement() && !nNaive.Finalized() && !nNaive.Disagreement() {
		rand.Seed(int64(numRounds) + seed)
		nBitwise.Round()

		rand.Seed(int64(numRounds) + seed)
		nNaive.Round()
		numRounds++
	}

	if nBitwise.Disagreement() || nNaive.Disagreement() {
		t.Fatalf("Network agreed on inconsistent values")
	}

	// Although this can theoretically fail with a correct implementation, it
	// shouldn't in practice
	if !nBitwise.Finalized() {
		t.Fatalf("Network agreed on values faster with naive implementation")
	}
	if !nBitwise.Agreement() {
		t.Fatalf("Network agreed on inconsistent values")
	}
}