301 lines
9.7 KiB
Go
301 lines
9.7 KiB
Go
// Copyright (C) 2018 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package statdb_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"storj.io/storj/internal/testcontext"
|
|
"storj.io/storj/pkg/statdb"
|
|
"storj.io/storj/pkg/storj"
|
|
"storj.io/storj/satellite"
|
|
"storj.io/storj/satellite/satellitedb/satellitedbtest"
|
|
)
|
|
|
|
func getRatio(success, total int64) (ratio float64) {
|
|
ratio = float64(success) / float64(total)
|
|
return ratio
|
|
}
|
|
|
|
func TestStatdb(t *testing.T) {
|
|
satellitedbtest.Run(t, func(t *testing.T, db satellite.DB) {
|
|
ctx := testcontext.New(t)
|
|
defer ctx.Cleanup()
|
|
|
|
testDatabase(ctx, t, db.StatDB())
|
|
})
|
|
}
|
|
|
|
func testDatabase(ctx context.Context, t *testing.T, sdb statdb.DB) {
|
|
nodeID := storj.NodeID{1, 2, 3, 4, 5}
|
|
currAuditSuccess := int64(4)
|
|
currAuditCount := int64(10)
|
|
currUptimeSuccess := int64(8)
|
|
currUptimeCount := int64(25)
|
|
|
|
{ // TestCreateNewAndWithStats
|
|
auditSuccessRatio := getRatio(currAuditSuccess, currAuditCount)
|
|
uptimeRatio := getRatio(currUptimeSuccess, currUptimeCount)
|
|
|
|
nodeStats := &statdb.NodeStats{
|
|
AuditSuccessRatio: auditSuccessRatio,
|
|
UptimeRatio: uptimeRatio,
|
|
AuditCount: currAuditCount,
|
|
AuditSuccessCount: currAuditSuccess,
|
|
UptimeCount: currUptimeCount,
|
|
UptimeSuccessCount: currUptimeSuccess,
|
|
}
|
|
|
|
stats, err := sdb.Create(ctx, nodeID, nodeStats)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, auditSuccessRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, uptimeRatio, stats.UptimeRatio)
|
|
|
|
stats, err = sdb.Get(ctx, nodeID)
|
|
assert.NoError(t, err)
|
|
|
|
assert.EqualValues(t, nodeID, stats.NodeID)
|
|
assert.EqualValues(t, currAuditCount, stats.AuditCount)
|
|
assert.EqualValues(t, currAuditSuccess, stats.AuditSuccessCount)
|
|
assert.EqualValues(t, auditSuccessRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, currUptimeCount, stats.UptimeCount)
|
|
assert.EqualValues(t, currUptimeSuccess, stats.UptimeSuccessCount)
|
|
assert.EqualValues(t, uptimeRatio, stats.UptimeRatio)
|
|
}
|
|
|
|
{ // TestCreateExists
|
|
auditSuccessRatio := getRatio(currAuditSuccess, currAuditCount)
|
|
uptimeRatio := getRatio(currUptimeSuccess, currUptimeCount)
|
|
|
|
nodeStats := &statdb.NodeStats{
|
|
AuditSuccessRatio: auditSuccessRatio,
|
|
UptimeRatio: uptimeRatio,
|
|
AuditCount: currAuditCount,
|
|
AuditSuccessCount: currAuditSuccess,
|
|
UptimeCount: currUptimeCount,
|
|
UptimeSuccessCount: currUptimeSuccess,
|
|
}
|
|
_, err := sdb.Create(ctx, nodeID, nodeStats)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
{ // TestGetDoesNotExist
|
|
noNodeID := storj.NodeID{255, 255, 255, 255}
|
|
|
|
_, err := sdb.Get(ctx, noNodeID)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
{ // TestFindInvalidNodes
|
|
for _, tt := range []struct {
|
|
nodeID storj.NodeID
|
|
auditSuccessCount int64
|
|
auditCount int64
|
|
auditSuccessRatio float64
|
|
uptimeSuccessCount int64
|
|
uptimeCount int64
|
|
uptimeRatio float64
|
|
}{
|
|
{storj.NodeID{1}, 20, 20, 1, 20, 20, 1}, // good audit success
|
|
{storj.NodeID{2}, 5, 20, 0.25, 20, 20, 1}, // bad audit success, good uptime
|
|
{storj.NodeID{3}, 20, 20, 1, 5, 20, 0.25}, // good audit success, bad uptime
|
|
{storj.NodeID{4}, 0, 0, 0, 20, 20, 1}, // "bad" audit success, no audits
|
|
{storj.NodeID{5}, 20, 20, 1, 0, 0, 0.25}, // "bad" uptime success, no checks
|
|
{storj.NodeID{6}, 0, 1, 0, 5, 5, 1}, // bad audit success exactly one audit
|
|
{storj.NodeID{7}, 0, 20, 0, 20, 20, 1}, // bad ratios, excluded from query
|
|
} {
|
|
nodeStats := &statdb.NodeStats{
|
|
AuditSuccessRatio: tt.auditSuccessRatio,
|
|
UptimeRatio: tt.uptimeRatio,
|
|
AuditCount: tt.auditCount,
|
|
AuditSuccessCount: tt.auditSuccessCount,
|
|
UptimeCount: tt.uptimeCount,
|
|
UptimeSuccessCount: tt.uptimeSuccessCount,
|
|
}
|
|
|
|
_, err := sdb.Create(ctx, tt.nodeID, nodeStats)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
nodeIds := storj.NodeIDList{
|
|
storj.NodeID{1}, storj.NodeID{2},
|
|
storj.NodeID{3}, storj.NodeID{4},
|
|
storj.NodeID{5}, storj.NodeID{6},
|
|
}
|
|
maxStats := &statdb.NodeStats{
|
|
AuditSuccessRatio: 0.5,
|
|
UptimeRatio: 0.5,
|
|
}
|
|
|
|
invalid, err := sdb.FindInvalidNodes(ctx, nodeIds, maxStats)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Contains(t, invalid, storj.NodeID{2})
|
|
assert.Contains(t, invalid, storj.NodeID{3})
|
|
assert.Contains(t, invalid, storj.NodeID{6})
|
|
assert.Len(t, invalid, 3)
|
|
}
|
|
|
|
{ // TestUpdateExists
|
|
auditSuccessRatio := getRatio(currAuditSuccess, currAuditCount)
|
|
uptimeRatio := getRatio(currUptimeSuccess, currUptimeCount)
|
|
|
|
stats, err := sdb.Get(ctx, nodeID)
|
|
assert.NoError(t, err)
|
|
|
|
assert.EqualValues(t, nodeID, stats.NodeID)
|
|
assert.EqualValues(t, currAuditCount, stats.AuditCount)
|
|
assert.EqualValues(t, currAuditSuccess, stats.AuditSuccessCount)
|
|
assert.EqualValues(t, auditSuccessRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, currUptimeCount, stats.UptimeCount)
|
|
assert.EqualValues(t, currUptimeSuccess, stats.UptimeSuccessCount)
|
|
assert.EqualValues(t, uptimeRatio, stats.UptimeRatio)
|
|
|
|
updateReq := &statdb.UpdateRequest{
|
|
NodeID: nodeID,
|
|
AuditSuccess: true,
|
|
IsUp: false,
|
|
}
|
|
stats, err = sdb.Update(ctx, updateReq)
|
|
assert.NoError(t, err)
|
|
|
|
currAuditSuccess++
|
|
currAuditCount++
|
|
currUptimeCount++
|
|
newAuditRatio := getRatio(currAuditSuccess, currAuditCount)
|
|
newUptimeRatio := getRatio(currUptimeSuccess, currUptimeCount)
|
|
|
|
assert.EqualValues(t, newAuditRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, newUptimeRatio, stats.UptimeRatio)
|
|
}
|
|
|
|
{ // TestUpdateUptimeExists
|
|
auditSuccessRatio := getRatio(currAuditSuccess, currAuditCount)
|
|
uptimeRatio := getRatio(currUptimeSuccess, currUptimeCount)
|
|
|
|
stats, err := sdb.Get(ctx, nodeID)
|
|
assert.NoError(t, err)
|
|
|
|
assert.EqualValues(t, nodeID, stats.NodeID)
|
|
assert.EqualValues(t, currAuditCount, stats.AuditCount)
|
|
assert.EqualValues(t, currAuditSuccess, stats.AuditSuccessCount)
|
|
assert.EqualValues(t, auditSuccessRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, currUptimeCount, stats.UptimeCount)
|
|
assert.EqualValues(t, currUptimeSuccess, stats.UptimeSuccessCount)
|
|
assert.EqualValues(t, uptimeRatio, stats.UptimeRatio)
|
|
|
|
stats, err = sdb.UpdateUptime(ctx, nodeID, false)
|
|
assert.NoError(t, err)
|
|
|
|
currUptimeCount++
|
|
newUptimeRatio := getRatio(currUptimeSuccess, currUptimeCount)
|
|
assert.EqualValues(t, auditSuccessRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, currAuditCount, stats.AuditCount)
|
|
assert.EqualValues(t, newUptimeRatio, stats.UptimeRatio)
|
|
}
|
|
|
|
{ // TestUpdateAuditSuccessExists
|
|
auditSuccessRatio := getRatio(currAuditSuccess, currAuditCount)
|
|
uptimeRatio := getRatio(currUptimeSuccess, currUptimeCount)
|
|
|
|
stats, err := sdb.Get(ctx, nodeID)
|
|
assert.NoError(t, err)
|
|
|
|
assert.EqualValues(t, nodeID, stats.NodeID)
|
|
assert.EqualValues(t, currAuditCount, stats.AuditCount)
|
|
assert.EqualValues(t, currAuditSuccess, stats.AuditSuccessCount)
|
|
assert.EqualValues(t, auditSuccessRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, currUptimeCount, stats.UptimeCount)
|
|
assert.EqualValues(t, currUptimeSuccess, stats.UptimeSuccessCount)
|
|
assert.EqualValues(t, uptimeRatio, stats.UptimeRatio)
|
|
|
|
stats, err = sdb.UpdateAuditSuccess(ctx, nodeID, false)
|
|
assert.NoError(t, err)
|
|
|
|
currAuditCount++
|
|
newAuditRatio := getRatio(currAuditSuccess, currAuditCount)
|
|
assert.EqualValues(t, newAuditRatio, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, currAuditCount, stats.AuditCount)
|
|
assert.EqualValues(t, uptimeRatio, stats.UptimeRatio)
|
|
}
|
|
|
|
{ // TestUpdateBatchExists
|
|
nodeID1 := storj.NodeID{255, 1}
|
|
nodeID2 := storj.NodeID{255, 2}
|
|
|
|
auditSuccessCount1 := int64(4)
|
|
auditCount1 := int64(10)
|
|
auditRatio1 := getRatio(auditSuccessCount1, auditCount1)
|
|
|
|
uptimeSuccessCount1 := int64(8)
|
|
uptimeCount1 := int64(25)
|
|
uptimeRatio1 := getRatio(uptimeSuccessCount1, uptimeCount1)
|
|
|
|
nodeStats := &statdb.NodeStats{
|
|
AuditSuccessCount: auditSuccessCount1,
|
|
AuditCount: auditCount1,
|
|
AuditSuccessRatio: auditRatio1,
|
|
UptimeSuccessCount: uptimeSuccessCount1,
|
|
UptimeCount: uptimeCount1,
|
|
UptimeRatio: uptimeRatio1,
|
|
}
|
|
|
|
stats, err := sdb.Create(ctx, nodeID1, nodeStats)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, auditRatio1, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, uptimeRatio1, stats.UptimeRatio)
|
|
|
|
auditSuccessCount2 := int64(7)
|
|
auditCount2 := int64(10)
|
|
auditRatio2 := getRatio(auditSuccessCount2, auditCount2)
|
|
|
|
uptimeSuccessCount2 := int64(8)
|
|
uptimeCount2 := int64(20)
|
|
uptimeRatio2 := getRatio(uptimeSuccessCount2, uptimeCount2)
|
|
|
|
nodeStats = &statdb.NodeStats{
|
|
AuditSuccessCount: auditSuccessCount2,
|
|
AuditCount: auditCount2,
|
|
AuditSuccessRatio: auditRatio2,
|
|
UptimeSuccessCount: uptimeSuccessCount2,
|
|
UptimeCount: uptimeCount2,
|
|
UptimeRatio: uptimeRatio2,
|
|
}
|
|
|
|
stats, err = sdb.Create(ctx, nodeID2, nodeStats)
|
|
assert.NoError(t, err)
|
|
assert.EqualValues(t, auditRatio2, stats.AuditSuccessRatio)
|
|
assert.EqualValues(t, uptimeRatio2, stats.UptimeRatio)
|
|
|
|
updateReqList := []*statdb.UpdateRequest{
|
|
&statdb.UpdateRequest{
|
|
NodeID: nodeID1,
|
|
AuditSuccess: true,
|
|
IsUp: false,
|
|
},
|
|
&statdb.UpdateRequest{
|
|
NodeID: nodeID2,
|
|
AuditSuccess: true,
|
|
IsUp: true,
|
|
},
|
|
}
|
|
statsList, _, err := sdb.UpdateBatch(ctx, updateReqList)
|
|
assert.NoError(t, err)
|
|
|
|
newAuditRatio1 := getRatio(auditSuccessCount1+1, auditCount1+1)
|
|
newUptimeRatio1 := getRatio(uptimeSuccessCount1, uptimeCount1+1)
|
|
newAuditRatio2 := getRatio(auditSuccessCount2+1, auditCount2+1)
|
|
newUptimeRatio2 := getRatio(uptimeSuccessCount2+1, uptimeCount2+1)
|
|
stats1 := statsList[0]
|
|
stats2 := statsList[1]
|
|
assert.EqualValues(t, newAuditRatio1, stats1.AuditSuccessRatio)
|
|
assert.EqualValues(t, newUptimeRatio1, stats1.UptimeRatio)
|
|
assert.EqualValues(t, newAuditRatio2, stats2.AuditSuccessRatio)
|
|
assert.EqualValues(t, newUptimeRatio2, stats2.UptimeRatio)
|
|
}
|
|
}
|