package integration

import (
	"testing"
	"time"

	"github.com/dexon-foundation/dexon-consensus-core/core/test"
	"github.com/stretchr/testify/suite"
)

type EventStatsTestSuite struct {
	suite.Suite
}

func (s *EventStatsTestSuite) TestCalculate() {
	// Setup a test with fixed latency in proposing and network,
	// and make sure the calculated statistics is expected.
	var (
		networkLatency   = &FixedLatencyModel{Latency: 100}
		proposingLatency = &FixedLatencyModel{Latency: 300}
		req              = s.Require()
	)

	apps, dbs, validators, err := PrepareValidators(
		7, networkLatency, proposingLatency)
	req.Nil(err)

	sch := test.NewScheduler(test.NewStopByConfirmedBlocks(50, apps, dbs))
	now := time.Now().UTC()
	for vID, v := range validators {
		sch.RegisterEventHandler(vID, v)
		req.Nil(sch.Seed(NewProposeBlockEvent(vID, now)))
	}
	sch.Run(10)
	req.Nil(VerifyApps(apps))
	// Check total statistics result.
	stats, err := NewStats(sch.CloneExecutionHistory(), apps)
	req.Nil(err)
	req.True(stats.All.ProposedBlockCount > 350)
	req.True(stats.All.ReceivedBlockCount > 350)
	req.True(stats.All.StronglyAckedBlockCount > 350)
	req.True(stats.All.TotalOrderedBlockCount >= 350)
	req.True(stats.All.DeliveredBlockCount >= 350)
	req.Equal(stats.All.ProposingLatency, 300*time.Millisecond)
	req.Equal(stats.All.ReceivingLatency, 100*time.Millisecond)
	// Check statistics for each validator.
	for _, vStats := range stats.ByValidator {
		req.True(vStats.ProposedBlockCount > 50)
		req.True(vStats.ReceivedBlockCount > 50)
		req.True(vStats.StronglyAckedBlockCount > 50)
		req.True(vStats.TotalOrderedBlockCount >= 50)
		req.True(vStats.DeliveredBlockCount >= 50)
		req.Equal(vStats.ProposingLatency, 300*time.Millisecond)
		req.Equal(vStats.ReceivingLatency, 100*time.Millisecond)
	}
}

func TestEventStats(t *testing.T) {
	suite.Run(t, new(EventStatsTestSuite))
}