f3b4bf2b7c
ctx is created in most tests, instead pass in as argument to reduce code duplication. Change-Id: I466c51c008392001129c8b007c9d6b3619935ac4
69 lines
2.5 KiB
Go
69 lines
2.5 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package downtime_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/common/testcontext"
|
|
"storj.io/common/testrand"
|
|
"storj.io/storj/satellite"
|
|
"storj.io/storj/satellite/satellitedb/satellitedbtest"
|
|
)
|
|
|
|
func TestDowntime(t *testing.T) {
|
|
// test basic downtime functionality
|
|
satellitedbtest.Run(t, func(ctx *testcontext.Context, t *testing.T, db satellite.DB) {
|
|
downtimeDB := db.DowntimeTracking()
|
|
|
|
now := time.Now()
|
|
oneYearAgo := time.Now().Add(-time.Hour * 24 * 365)
|
|
|
|
// node1 was offline for one hour recently, and offline for two hours a year ago
|
|
nodeID1 := testrand.NodeID()
|
|
err := downtimeDB.Add(ctx, nodeID1, now, time.Hour)
|
|
require.NoError(t, err)
|
|
err = downtimeDB.Add(ctx, nodeID1, oneYearAgo, 2*time.Hour)
|
|
require.NoError(t, err)
|
|
|
|
// node2 was offline for two hours a year ago
|
|
nodeID2 := testrand.NodeID()
|
|
err = downtimeDB.Add(ctx, nodeID2, oneYearAgo, 2*time.Hour)
|
|
require.NoError(t, err)
|
|
|
|
// if we only check recent history, node1 offline time should be 1 hour
|
|
duration, err := downtimeDB.GetOfflineTime(ctx, nodeID1, now.Add(-time.Hour), now.Add(time.Hour))
|
|
require.NoError(t, err)
|
|
require.Equal(t, duration, time.Hour)
|
|
|
|
// if we only check recent history, node2 should not be offline at all
|
|
duration, err = downtimeDB.GetOfflineTime(ctx, nodeID2, now.Add(-time.Hour), now.Add(time.Hour))
|
|
require.NoError(t, err)
|
|
require.Equal(t, duration, time.Duration(0))
|
|
|
|
// if we only check old history, node1 offline time should be 2 hours
|
|
duration, err = downtimeDB.GetOfflineTime(ctx, nodeID1, oneYearAgo.Add(-time.Hour), oneYearAgo.Add(time.Hour))
|
|
require.NoError(t, err)
|
|
require.Equal(t, duration, 2*time.Hour)
|
|
|
|
// if we only check old history, node2 offline time should be 2 hours
|
|
duration, err = downtimeDB.GetOfflineTime(ctx, nodeID2, oneYearAgo.Add(-time.Hour), oneYearAgo.Add(time.Hour))
|
|
require.NoError(t, err)
|
|
require.Equal(t, duration, 2*time.Hour)
|
|
|
|
// if we check all history (from before oneYearAgo to after now), node1 should be offline for 3 hours
|
|
duration, err = downtimeDB.GetOfflineTime(ctx, nodeID1, oneYearAgo.Add(-time.Hour), now.Add(time.Hour))
|
|
require.NoError(t, err)
|
|
require.Equal(t, duration, 3*time.Hour)
|
|
|
|
// if we check all history (from before oneYearAgo to after now), node2 should be offline for 2 hours
|
|
duration, err = downtimeDB.GetOfflineTime(ctx, nodeID2, oneYearAgo.Add(-time.Hour), now.Add(time.Hour))
|
|
require.NoError(t, err)
|
|
require.Equal(t, duration, 2*time.Hour)
|
|
})
|
|
}
|