[tests] Update peer bootstrapper single node integration test (#3294)

src/dbnode/integration/integration.go

@@ -32,9 +32,11 @@ import (
 	"github.com/m3db/m3/src/dbnode/namespace"
 	"github.com/m3db/m3/src/dbnode/persist/fs"
 	persistfs "github.com/m3db/m3/src/dbnode/persist/fs"
+	"github.com/m3db/m3/src/dbnode/runtime"
 	"github.com/m3db/m3/src/dbnode/storage"
 	"github.com/m3db/m3/src/dbnode/storage/bootstrap"
 	"github.com/m3db/m3/src/dbnode/storage/bootstrap/bootstrapper"
+	"github.com/m3db/m3/src/dbnode/storage/bootstrap/bootstrapper/commitlog"
 	bfs "github.com/m3db/m3/src/dbnode/storage/bootstrap/bootstrapper/fs"
 	"github.com/m3db/m3/src/dbnode/storage/bootstrap/bootstrapper/peers"
 	"github.com/m3db/m3/src/dbnode/storage/bootstrap/bootstrapper/uninitialized"
@@ -118,16 +120,17 @@ func newMultiAddrAdminClient(
 
 // BootstrappableTestSetupOptions defines options for test setups.
 type BootstrappableTestSetupOptions struct {
-	FinalBootstrapper           string
-	BootstrapBlocksBatchSize    int
-	BootstrapBlocksConcurrency  int
-	BootstrapConsistencyLevel   topology.ReadConsistencyLevel
-	TopologyInitializer         topology.Initializer
-	TestStatsReporter           xmetrics.TestStatsReporter
-	DisablePeersBootstrapper    bool
-	UseTChannelClientForWriting bool
-	EnableRepairs               bool
-	AdminClientCustomOpts       []client.CustomAdminOption
+	FinalBootstrapper            string
+	BootstrapBlocksBatchSize     int
+	BootstrapBlocksConcurrency   int
+	BootstrapConsistencyLevel    topology.ReadConsistencyLevel
+	TopologyInitializer          topology.Initializer
+	TestStatsReporter            xmetrics.TestStatsReporter
+	DisableCommitLogBootstrapper bool
+	DisablePeersBootstrapper     bool
+	UseTChannelClientForWriting  bool
+	EnableRepairs                bool
+	AdminClientCustomOpts        []client.CustomAdminOption
 }
 
 type closeFn func()
@@ -166,6 +169,7 @@ func NewDefaultBootstrappableTestSetups( // nolint:gocyclo
 	for i := 0; i < replicas; i++ {
 		var (
 			instance                    = i
+			usingCommitLogBootstrapper  = !setupOpts[i].DisableCommitLogBootstrapper
 			usingPeersBootstrapper      = !setupOpts[i].DisablePeersBootstrapper
 			finalBootstrapperToUse      = setupOpts[i].FinalBootstrapper
 			useTChannelClientForWriting = setupOpts[i].UseTChannelClientForWriting
@@ -303,6 +307,17 @@ func NewDefaultBootstrappableTestSetups( // nolint:gocyclo
 			require.NoError(t, err)
 		}
 
+		if usingCommitLogBootstrapper {
+			bootstrapCommitlogOpts := commitlog.NewOptions().
+				SetResultOptions(bsOpts).
+				SetCommitLogOptions(setup.StorageOpts().CommitLogOptions()).
+				SetRuntimeOptionsManager(runtime.NewOptionsManager())
+
+			finalBootstrapper, err = commitlog.NewCommitLogBootstrapperProvider(bootstrapCommitlogOpts,
+				mustInspectFilesystem(fsOpts), finalBootstrapper)
+			require.NoError(t, err)
+		}
+
 		persistMgr, err := persistfs.NewPersistManager(fsOpts)
 		require.NoError(t, err)
 

src/dbnode/integration/peers_bootstrap_high_concurrency_test.go

@@ -107,9 +107,10 @@ func testPeersBootstrapHighConcurrency(
 			DisablePeersBootstrapper: true,
 		},
 		{
-			DisablePeersBootstrapper:   false,
-			BootstrapBlocksBatchSize:   batchSize,
-			BootstrapBlocksConcurrency: concurrency,
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+			BootstrapBlocksBatchSize:     batchSize,
+			BootstrapBlocksConcurrency:   concurrency,
 		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)

src/dbnode/integration/peers_bootstrap_index_aggregate_test.go

@@ -67,7 +67,10 @@ func TestPeersBootstrapIndexAggregateQuery(t *testing.T) {
 
 	setupOpts := []BootstrappableTestSetupOptions{
 		{DisablePeersBootstrapper: true},
-		{DisablePeersBootstrapper: false},
+		{
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()

src/dbnode/integration/peers_bootstrap_index_test.go

@@ -72,7 +72,10 @@ func TestPeersBootstrapIndexWithIndexingEnabled(t *testing.T) {
 
 	setupOpts := []BootstrappableTestSetupOptions{
 		{DisablePeersBootstrapper: true},
-		{DisablePeersBootstrapper: false},
+		{
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()

src/dbnode/integration/peers_bootstrap_merge_local_test.go

@@ -79,9 +79,10 @@ func testPeersBootstrapMergeLocal(t *testing.T, setTestOpts setTestOptions, upda
 				UseTChannelClientForWriting: true,
 			},
 			{
-				DisablePeersBootstrapper:    false,
-				UseTChannelClientForWriting: true,
-				TestStatsReporter:           reporter,
+				DisableCommitLogBootstrapper: true,
+				DisablePeersBootstrapper:     false,
+				UseTChannelClientForWriting:  true,
+				TestStatsReporter:            reporter,
 			},
 		}
 	)

src/dbnode/integration/peers_bootstrap_merge_peer_blocks_test.go

@@ -72,7 +72,10 @@ func testPeersBootstrapMergePeerBlocks(t *testing.T, setTestOpts setTestOptions,
 	setupOpts := []BootstrappableTestSetupOptions{
 		{DisablePeersBootstrapper: true},
 		{DisablePeersBootstrapper: true},
-		{DisablePeersBootstrapper: false},
+		{
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()

src/dbnode/integration/peers_bootstrap_node_down_test.go

@@ -60,7 +60,10 @@ func TestPeersBootstrapNodeDown(t *testing.T) {
 	setupOpts := []BootstrappableTestSetupOptions{
 		{DisablePeersBootstrapper: true},
 		{DisablePeersBootstrapper: true},
-		{DisablePeersBootstrapper: false},
+		{
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()

src/dbnode/integration/peers_bootstrap_partial_data_test.go

@@ -67,7 +67,10 @@ func TestPeersBootstrapPartialData(t *testing.T) {
 
 	setupOpts := []BootstrappableTestSetupOptions{
 		{DisablePeersBootstrapper: true},
-		{DisablePeersBootstrapper: false},
+		{
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()

src/dbnode/integration/peers_bootstrap_select_best_test.go

@@ -59,7 +59,10 @@ func TestPeersBootstrapSelectBest(t *testing.T) {
 	setupOpts := []BootstrappableTestSetupOptions{
 		{DisablePeersBootstrapper: true},
 		{DisablePeersBootstrapper: true},
-		{DisablePeersBootstrapper: false},
+		{
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()

src/dbnode/integration/peers_bootstrap_simple_test.go

@@ -69,7 +69,10 @@ func testPeersBootstrapSimple(t *testing.T, setTestOpts setTestOptions, updateIn
 
 	setupOpts := []BootstrappableTestSetupOptions{
 		{DisablePeersBootstrapper: true},
-		{DisablePeersBootstrapper: false},
+		{
+			DisableCommitLogBootstrapper: true,
+			DisablePeersBootstrapper:     false,
+		},
 	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()

src/dbnode/integration/peers_bootstrap_single_node_test.go

@@ -26,18 +26,73 @@ import (
 	"testing"
 	"time"
 
+	"github.com/m3db/m3/src/cluster/services"
+	"github.com/m3db/m3/src/cluster/shard"
 	"github.com/m3db/m3/src/dbnode/integration/generate"
 	"github.com/m3db/m3/src/dbnode/namespace"
 	"github.com/m3db/m3/src/dbnode/retention"
-	"github.com/m3db/m3/src/dbnode/storage/bootstrap/bootstrapper"
+	"github.com/m3db/m3/src/dbnode/sharding"
+	"github.com/m3db/m3/src/dbnode/storage/bootstrap/bootstrapper/uninitialized"
+	"github.com/m3db/m3/src/dbnode/topology"
+	"github.com/m3db/m3/src/dbnode/topology/testutil"
 	xtest "github.com/m3db/m3/src/x/test"
 
 	"github.com/stretchr/testify/require"
 )
 
-// TestPeersBootstrapSingleNode makes sure that we can include the peer bootstrapper
-// in a single-node topology without causing a bootstrap failure or infinite hang.
-func TestPeersBootstrapSingleNode(t *testing.T) {
+// TestPeersBootstrapSingleNodeUninitialized makes sure that we can include the peer bootstrapper
+// in a single-node topology of a non-initialized cluster without causing a bootstrap failure or infinite hang.
+func TestPeersBootstrapSingleNodeUninitialized(t *testing.T) {
+	opts := NewTestOptions(t)
+
+	// Define a topology with initializing shards
+	minShard := uint32(0)
+	maxShard := uint32(opts.NumShards()) - uint32(1)
+	instances := []services.ServiceInstance{
+		node(t, 0, newClusterShardsRange(minShard, maxShard, shard.Initializing)),
+	}
+
+	hostShardSets := []topology.HostShardSet{}
+	for _, instance := range instances {
+		h, err := topology.NewHostShardSetFromServiceInstance(instance, sharding.DefaultHashFn(opts.NumShards()))
+		require.NoError(t, err)
+		hostShardSets = append(hostShardSets, h)
+	}
+
+	shards := testutil.ShardsRange(minShard, maxShard, shard.Initializing)
+	shardSet, err := sharding.NewShardSet(
+		shards,
+		sharding.DefaultHashFn(int(maxShard)),
+	)
+	require.NoError(t, err)
+
+	topoOpts := topology.NewStaticOptions().
+		SetReplicas(len(instances)).
+		SetHostShardSets(hostShardSets).
+		SetShardSet(shardSet)
+	topoInit := topology.NewStaticInitializer(topoOpts)
+
+	setupOpts := []BootstrappableTestSetupOptions{
+		{
+			DisablePeersBootstrapper: false,
+			TopologyInitializer:      topoInit,
+			// This will bootstrap w/ unfulfilled ranges.
+			FinalBootstrapper: uninitialized.UninitializedTopologyBootstrapperName,
+		},
+	}
+	testPeersBootstrapSingleNode(t, setupOpts)
+}
+
+// TestPeersBootstrapSingleNodeInitialized makes sure that we can include the peer bootstrapper
+// in a single-node topology of already initialized cluster without causing a bootstrap failure or infinite hang.
+func TestPeersBootstrapSingleNodeInitialized(t *testing.T) {
+	setupOpts := []BootstrappableTestSetupOptions{
+		{DisablePeersBootstrapper: false},
+	}
+	testPeersBootstrapSingleNode(t, setupOpts)
+}
+
+func testPeersBootstrapSingleNode(t *testing.T, setupOpts []BootstrappableTestSetupOptions) {
 	if testing.Short() {
 		t.SkipNow()
 	}
@@ -58,13 +113,6 @@ func TestPeersBootstrapSingleNode(t *testing.T) {
 		SetUseTChannelClientForWriting(true).
 		SetUseTChannelClientForReading(true)
 
-	setupOpts := []BootstrappableTestSetupOptions{
-		{
-			DisablePeersBootstrapper: false,
-			// This will bootstrap w/ unfulfilled ranges.
-			FinalBootstrapper: bootstrapper.NoOpAllBootstrapperName,
-		},
-	}
 	setups, closeFn := NewDefaultBootstrappableTestSetups(t, opts, setupOpts)
 	defer closeFn()