115 lines
3.0 KiB
Go
115 lines
3.0 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package reports_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap/zaptest"
|
|
|
|
"storj.io/storj/cmd/satellite/reports"
|
|
"storj.io/storj/satellite/attribution"
|
|
)
|
|
|
|
func TestSumAttributionByUserAgent(t *testing.T) {
|
|
log := zaptest.NewLogger(t)
|
|
// test empty user agents
|
|
attributions := []*attribution.BucketUsage{
|
|
{
|
|
UserAgent: []byte{},
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
{
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
}
|
|
totals := reports.SumAttributionByUserAgent(attributions, log)
|
|
require.Equal(t, 0, len(totals))
|
|
|
|
// test user agent with additional entries and uppercase letters is summed with
|
|
// the first one
|
|
attributions = []*attribution.BucketUsage{
|
|
{
|
|
UserAgent: []byte("teststorj"),
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
{
|
|
UserAgent: []byte("TESTSTORJ/other"),
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
}
|
|
totals = reports.SumAttributionByUserAgent(attributions, log)
|
|
require.Equal(t, 1, len(totals))
|
|
require.Equal(t, float64(2), totals["teststorj"].ByteHours)
|
|
require.Equal(t, float64(2), totals["teststorj"].SegmentHours)
|
|
require.Equal(t, float64(2), totals["teststorj"].ObjectHours)
|
|
require.Equal(t, float64(2), totals["teststorj"].BucketHours)
|
|
require.Equal(t, int64(2), totals["teststorj"].BytesEgress)
|
|
|
|
// test two user agents are summed separately
|
|
attributions = []*attribution.BucketUsage{
|
|
{
|
|
UserAgent: []byte("teststorj1"),
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
{
|
|
UserAgent: []byte("teststorj1"),
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
{
|
|
UserAgent: []byte("teststorj2"),
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
{
|
|
UserAgent: []byte("teststorj2"),
|
|
ByteHours: 1,
|
|
SegmentHours: 1,
|
|
ObjectHours: 1,
|
|
Hours: 1,
|
|
EgressData: 1,
|
|
},
|
|
}
|
|
totals = reports.SumAttributionByUserAgent(attributions, log)
|
|
require.Equal(t, 2, len(totals))
|
|
require.Equal(t, float64(2), totals["teststorj1"].ByteHours)
|
|
require.Equal(t, float64(2), totals["teststorj1"].SegmentHours)
|
|
require.Equal(t, float64(2), totals["teststorj1"].ObjectHours)
|
|
require.Equal(t, float64(2), totals["teststorj1"].BucketHours)
|
|
require.Equal(t, int64(2), totals["teststorj1"].BytesEgress)
|
|
require.Equal(t, float64(2), totals["teststorj2"].ByteHours)
|
|
require.Equal(t, float64(2), totals["teststorj2"].SegmentHours)
|
|
require.Equal(t, float64(2), totals["teststorj2"].ObjectHours)
|
|
require.Equal(t, float64(2), totals["teststorj2"].BucketHours)
|
|
require.Equal(t, int64(2), totals["teststorj2"].BytesEgress)
|
|
}
|