path: root/integration_test/stats_test.go

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))
}