Function TestAsyncInitConfigUpdates

rest/indextest/index_test.go:628–737

TestAsyncInitConfigUpdates verifies that a database with in-progress async index initialization can accept all config updates from the local node. (prior to CBG-4008, operations would block waiting for async init to complete)

(t *testing.T)

// index initialization can accept all config updates from the local node.
// (prior to CBG-4008, operations would block waiting for async init to complete)
func TestAsyncInitConfigUpdates(t *testing.T) {
	if base.UnitTestUrlIsWalrus() {
		t.Skip("This test only works against Couchbase Server")
	}
	base.TestRequiresCollections(t)
	base.SetUpTestLogging(t, base.LevelDebug, base.KeyHTTP)

	sc, closeFn := rest.StartBootstrapServer(t)
	defer closeFn()

	// Set testing callbacks for async initialization
	collectionCount := int64(0)
	initStarted := make(chan error)
	unblockInit := make(chan error)
	collectionCompleteCallback := func(_ string, _ base.ScopeAndCollectionName, status db.CollectionIndexStatus) {
		if status != db.CollectionIndexStatusReady {
			return
		}
		count := atomic.AddInt64(&collectionCount, 1)
		// On first collection, close initStarted channel
		log.Printf("collection callback count: %v", count)
		if count == 1 {
			log.Printf("closing initStarted")
			close(initStarted)
		}
		rest.WaitForChannel(t, unblockInit, "waiting for test to unblock initialization")
	}
	sc.DatabaseInitManager.SetTestCallbacks(collectionCompleteCallback, nil)

	ctx := base.TestCtx(t)
	// Get a test bucket, and use it to create the database.
	tb := base.GetTestBucket(t)
	defer tb.Close(ctx)

	importFilter := "function(doc) { return true }"
	syncFunc := "function(doc){ channel(doc.channels); }"

	dbConfig := makeDbConfig(t, tb, syncFunc, importFilter)
	dbConfig.StartOffline = base.Ptr(true)
	dbConfigPayload, err := json.Marshal(dbConfig)
	require.NoError(t, err)
	dbName := "db"

	keyspace := dbName
	if len(dbConfig.Scopes) > 0 {
		keyspaces := getRESTKeyspaces(dbName, dbConfig.Scopes)
		keyspace = keyspaces[0]
	}

	// Create database with offline=true
	resp := rest.BootstrapAdminRequest(t, sc, http.MethodPut, "/"+dbName+"/", string(dbConfigPayload))
	resp.RequireStatus(http.StatusCreated)

	// Wait for init to start before interacting with the db, validate db state is offline
	rest.WaitForChannel(t, initStarted, "waiting for initialization to start")
	log.Printf("initialization started")
	waitAndRequireDBState(t, sc, dbName, db.DBOffline)