64 lines
2.3 KiB
Go
64 lines
2.3 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/storj/private/testplanet"
|
|
)
|
|
|
|
func TestCheckNodeAvailability(t *testing.T) {
|
|
testplanet.Run(t, testplanet.Config{
|
|
SatelliteCount: 1, StorageNodeCount: 1, UplinkCount: 0,
|
|
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
|
|
node := planet.StorageNodes[0]
|
|
nodeDossier := planet.StorageNodes[0].Local()
|
|
satellite := planet.Satellites[0]
|
|
|
|
node.Contact.Chore.Pause(ctx)
|
|
satellite.Audit.Chore.Loop.Pause()
|
|
|
|
// test that last success and failure checks are before now
|
|
beforeSuccessfulCheck := time.Now()
|
|
dossier, err := satellite.Overlay.Service.Get(ctx, node.ID())
|
|
require.NoError(t, err)
|
|
require.True(t, dossier.Reputation.LastContactSuccess.Before(beforeSuccessfulCheck))
|
|
require.True(t, dossier.Reputation.LastContactFailure.Before(beforeSuccessfulCheck))
|
|
|
|
success, err := satellite.DowntimeTracking.Service.CheckAndUpdateNodeAvailability(ctx, nodeDossier.Id, nodeDossier.Address.GetAddress())
|
|
require.NoError(t, err)
|
|
require.True(t, success)
|
|
|
|
lastFailure := dossier.Reputation.LastContactFailure
|
|
|
|
// now test that CheckAndUpdateNodeAvailability updated with a success, and the last contact failure is the same.
|
|
dossier, err = satellite.Overlay.Service.Get(ctx, node.ID())
|
|
require.NoError(t, err)
|
|
require.True(t, dossier.Reputation.LastContactSuccess.After(beforeSuccessfulCheck))
|
|
require.True(t, dossier.Reputation.LastContactFailure.Equal(lastFailure))
|
|
|
|
lastSuccess := dossier.Reputation.LastContactSuccess
|
|
|
|
// shutdown the node
|
|
err = node.Server.Close()
|
|
require.NoError(t, err)
|
|
|
|
// now test that CheckAndUpdateNodeAvailability updated with a failure, and the last contact success is the same
|
|
beforeFailedCheck := time.Now()
|
|
success, err = satellite.DowntimeTracking.Service.CheckAndUpdateNodeAvailability(ctx, nodeDossier.Id, nodeDossier.Address.GetAddress())
|
|
require.NoError(t, err)
|
|
require.False(t, success)
|
|
|
|
dossier, err = satellite.Overlay.Service.Get(ctx, node.ID())
|
|
require.NoError(t, err)
|
|
require.True(t, dossier.Reputation.LastContactFailure.After(beforeFailedCheck))
|
|
require.True(t, dossier.Reputation.LastContactSuccess.Equal(lastSuccess))
|
|
})
|
|
}
|