storj/storagenode/gracefulexit/endpoint_test.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/pb"
	"storj.io/common/testcontext"
	"storj.io/storj/private/testplanet"
)

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().UTC(), 0)
		require.NoError(t, err)

		nonExitingSatellites, err := storagenode.GracefulExit.Endpoint.GetNonExitingSatellites(ctx, &pb.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 := &pb.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().UTC(), 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, &pb.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())
	})
}