5d70b6abef
montly bandwidth summaries for single/all nodes, single/all satellites added. Change-Id: Ic384886c10622df74a4bd0645e2d7f2a85477644
104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
// Copyright (C) 2021 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package bandwidth_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/storj/multinode/bandwidth"
|
|
)
|
|
|
|
func TestUsageRolloutDailyCache(t *testing.T) {
|
|
newTimestamp := func(month time.Month, day int) time.Time {
|
|
return time.Date(2021, month, day, 0, 0, 0, 0, time.UTC)
|
|
}
|
|
|
|
testData := []struct {
|
|
Date time.Time
|
|
Egress []bandwidth.Egress
|
|
Ingress []bandwidth.Ingress
|
|
Delete []int64
|
|
}{
|
|
{
|
|
Date: newTimestamp(time.May, 2),
|
|
Ingress: []bandwidth.Ingress{{Repair: 1, Usage: 0}, {Repair: 10, Usage: 20}},
|
|
Egress: []bandwidth.Egress{{Repair: 10, Audit: 20, Usage: 30}, {Repair: 10, Audit: 20, Usage: 30}},
|
|
Delete: []int64{10, 20, 30},
|
|
},
|
|
{
|
|
Date: newTimestamp(time.May, 3),
|
|
Ingress: []bandwidth.Ingress{{Repair: 1, Usage: 0}, {Repair: 10, Usage: 20}},
|
|
Egress: []bandwidth.Egress{{Repair: 101, Audit: 201, Usage: 301}, {Repair: 101, Audit: 201, Usage: 301}},
|
|
Delete: []int64{101, 201, 301},
|
|
},
|
|
{
|
|
Date: newTimestamp(time.May, 4),
|
|
Ingress: []bandwidth.Ingress{{Repair: 12, Usage: 20}, {Repair: 120, Usage: 220}},
|
|
Egress: []bandwidth.Egress{{Repair: 310, Audit: 320, Usage: 330}, {Repair: 100, Audit: 200, Usage: 300}},
|
|
Delete: []int64{100, 200, 300},
|
|
},
|
|
{
|
|
Date: newTimestamp(time.May, 1),
|
|
Ingress: []bandwidth.Ingress{{Repair: 123, Usage: 123}, {Repair: 123, Usage: 123}},
|
|
Egress: []bandwidth.Egress{{Repair: 20, Audit: 20, Usage: 20}, {Repair: 30, Audit: 30, Usage: 30}},
|
|
Delete: []int64{2, 3, 4},
|
|
},
|
|
}
|
|
expected := []bandwidth.UsageRollup{
|
|
{
|
|
IntervalStart: newTimestamp(time.May, 1),
|
|
Ingress: bandwidth.Ingress{Repair: 246, Usage: 246},
|
|
Egress: bandwidth.Egress{Repair: 50, Audit: 50, Usage: 50},
|
|
Delete: 9,
|
|
},
|
|
{
|
|
IntervalStart: newTimestamp(time.May, 2),
|
|
Ingress: bandwidth.Ingress{Repair: 11, Usage: 20},
|
|
Egress: bandwidth.Egress{Repair: 20, Audit: 40, Usage: 60},
|
|
Delete: 60,
|
|
},
|
|
{
|
|
IntervalStart: newTimestamp(time.May, 3),
|
|
Ingress: bandwidth.Ingress{Repair: 11, Usage: 20},
|
|
Egress: bandwidth.Egress{Repair: 202, Audit: 402, Usage: 602},
|
|
Delete: 603,
|
|
},
|
|
{
|
|
IntervalStart: newTimestamp(time.May, 4),
|
|
Ingress: bandwidth.Ingress{Repair: 132, Usage: 240},
|
|
Egress: bandwidth.Egress{Repair: 410, Audit: 520, Usage: 630},
|
|
Delete: 600,
|
|
},
|
|
}
|
|
|
|
cache := make(bandwidth.UsageRollupDailyCache)
|
|
for _, entry := range testData {
|
|
_, month, day := entry.Date.Date()
|
|
for _, egr := range entry.Egress {
|
|
cache.Add(bandwidth.UsageRollup{
|
|
Egress: egr,
|
|
IntervalStart: newTimestamp(month, day),
|
|
})
|
|
}
|
|
for _, ing := range entry.Ingress {
|
|
cache.Add(bandwidth.UsageRollup{
|
|
Ingress: ing,
|
|
IntervalStart: newTimestamp(month, day),
|
|
})
|
|
}
|
|
for _, del := range entry.Delete {
|
|
cache.Add(bandwidth.UsageRollup{
|
|
Delete: del,
|
|
IntervalStart: newTimestamp(month, day),
|
|
})
|
|
}
|
|
}
|
|
|
|
stamps := cache.Sorted()
|
|
require.Equal(t, expected, stamps)
|
|
}
|