89 lines
3.1 KiB
Go
89 lines
3.1 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package gracefulexit_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/common/testcontext"
|
|
"storj.io/storj/private/testplanet"
|
|
"storj.io/storj/storagenode/internalpb"
|
|
)
|
|
|
|
func TestGetNonExitingSatellites(t *testing.T) {
|
|
testplanet.Run(t, testplanet.Config{
|
|
SatelliteCount: 3, StorageNodeCount: 1, UplinkCount: 1,
|
|
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
|
|
totalSatelliteCount := len(planet.Satellites)
|
|
exitingSatelliteCount := 1
|
|
exitingSatellite := planet.Satellites[0]
|
|
storagenode := planet.StorageNodes[0]
|
|
|
|
// set a satellite to already be exiting
|
|
err := storagenode.DB.Satellites().InitiateGracefulExit(ctx, exitingSatellite.ID(), time.Now(), 0)
|
|
require.NoError(t, err)
|
|
|
|
nonExitingSatellites, err := storagenode.GracefulExit.Endpoint.GetNonExitingSatellites(ctx, &internalpb.GetNonExitingSatellitesRequest{})
|
|
require.NoError(t, err)
|
|
require.Len(t, nonExitingSatellites.GetSatellites(), totalSatelliteCount-exitingSatelliteCount)
|
|
|
|
for _, satellite := range nonExitingSatellites.GetSatellites() {
|
|
require.NotEqual(t, exitingSatellite.ID(), satellite.NodeId)
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestInitiateGracefulExit(t *testing.T) {
|
|
testplanet.Run(t, testplanet.Config{
|
|
SatelliteCount: 3, StorageNodeCount: 1, UplinkCount: 1,
|
|
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
|
|
storagenode := planet.StorageNodes[0]
|
|
exitingSatelliteID := planet.Satellites[0].ID()
|
|
|
|
req := &internalpb.InitiateGracefulExitRequest{
|
|
NodeId: exitingSatelliteID,
|
|
}
|
|
|
|
resp, err := storagenode.GracefulExit.Endpoint.InitiateGracefulExit(ctx, req)
|
|
require.NoError(t, err)
|
|
// check progress is 0
|
|
require.EqualValues(t, 0, resp.GetPercentComplete())
|
|
require.False(t, resp.GetSuccessful())
|
|
|
|
exitStatuses, err := storagenode.DB.Satellites().ListGracefulExits(ctx)
|
|
require.NoError(t, err)
|
|
require.Len(t, exitStatuses, 1)
|
|
require.Equal(t, exitingSatelliteID, exitStatuses[0].SatelliteID)
|
|
})
|
|
}
|
|
|
|
func TestGetExitProgress(t *testing.T) {
|
|
testplanet.Run(t, testplanet.Config{
|
|
SatelliteCount: 3, StorageNodeCount: 1, UplinkCount: 1,
|
|
}, func(t *testing.T, ctx *testcontext.Context, planet *testplanet.Planet) {
|
|
exitingSatellite := planet.Satellites[0]
|
|
storagenode := planet.StorageNodes[0]
|
|
|
|
// start graceful exit
|
|
err := storagenode.DB.Satellites().InitiateGracefulExit(ctx, exitingSatellite.ID(), time.Now(), 100)
|
|
require.NoError(t, err)
|
|
err = storagenode.DB.Satellites().UpdateGracefulExit(ctx, exitingSatellite.ID(), 20)
|
|
require.NoError(t, err)
|
|
|
|
// check graceful exit progress
|
|
resp, err := storagenode.GracefulExit.Endpoint.GetExitProgress(ctx, &internalpb.GetExitProgressRequest{})
|
|
require.NoError(t, err)
|
|
require.Len(t, resp.GetProgress(), 1)
|
|
progress := resp.GetProgress()[0]
|
|
require.Equal(t, progress.GetDomainName(), exitingSatellite.Addr())
|
|
require.Equal(t, progress.NodeId, exitingSatellite.ID())
|
|
require.EqualValues(t, 20, progress.GetPercentComplete())
|
|
require.False(t, progress.GetSuccessful())
|
|
require.Empty(t, progress.GetCompletionReceipt())
|
|
})
|
|
}
|