storj/storagenode/bandwidth/service_test.go

45 lines
1.4 KiB
Go
Raw Normal View History

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package bandwidth_test
import (
"testing"
"time"
"github.com/stretchr/testify/require"
"storj.io/storj/pkg/pb"
"storj.io/storj/private/testcontext"
"storj.io/storj/private/testplanet"
"storj.io/storj/private/teststorj"
)
// Simple test for ensuring the service runs Rollups in the Loop
func TestBandwidthService(t *testing.T) {
testplanet.Run(t, testplanet.Config{
SatelliteCount: 1, StorageNodeCount: 3, UplinkCount: 1,
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
testID1 := teststorj.NodeIDFromString("testId1")
for _, storageNode := range planet.StorageNodes {
// stop bandwidth service, so we can run it manually
storageNode.Bandwidth.Loop.Pause()
// Create data for an hour ago so we can rollup
err := storageNode.DB.Bandwidth().Add(ctx, testID1, pb.PieceAction_PUT, 2, time.Now().Add(time.Hour*-2))
require.NoError(t, err)
err = storageNode.DB.Bandwidth().Add(ctx, testID1, pb.PieceAction_GET, 3, time.Now().Add(time.Hour*-2))
require.NoError(t, err)
err = storageNode.DB.Bandwidth().Add(ctx, testID1, pb.PieceAction_GET_AUDIT, 4, time.Now().Add(time.Hour*-2))
require.NoError(t, err)
storageNode.Bandwidth.Loop.TriggerWait()
usage, err := storageNode.DB.Bandwidth().Summary(ctx, time.Now().Add(time.Hour*-48), time.Now())
require.NoError(t, err)
require.Equal(t, int64(9), usage.Total())
}
})
}