path: root/integration_test/stats_test.go

package integration

import (
    "testing"
    "time"

    "github.com/dexon-foundation/dexon-consensus/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   = &test.FixedLatencyModel{Latency: 100}
        proposingLatency = &test.FixedLatencyModel{Latency: 300}
        req              = s.Require()
    )
    prvKeys, pubKeys, err := test.NewKeys(7)
    req.NoError(err)
    gov, err := test.NewGovernance(pubKeys, 100*time.Millisecond)
    req.NoError(err)
    nodes, err := PrepareNodes(
        gov, prvKeys, 7, networkLatency, proposingLatency)
    req.NoError(err)
    apps, dbs := CollectAppAndDBFromNodes(nodes)
    sch := test.NewScheduler(test.NewStopByConfirmedBlocks(50, apps, dbs))
    now := time.Now().UTC()
    for _, n := range nodes {
        req.NoError(n.Bootstrap(sch, 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 node.
    for _, vStats := range stats.ByNode {
        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))
}