Function TestServerlessUnsuspendFetchFallback

rest/serverless_test.go:350–391

Confirms that when the database config is not in sc.dbConfigs, the fetch callback is check if the config is in a bucket

(t *testing.T)

// Confirms that when the database config is not in sc.dbConfigs, the fetch callback is check if the config is in a bucket
func TestServerlessUnsuspendFetchFallback(t *testing.T) {
	RequireBucketSpecificCredentials(t)
	ctx := base.TestCtx(t)
	tb := base.GetTestBucket(t)
	defer tb.Close(ctx)

	rt := NewRestTester(t, &RestTesterConfig{
		CustomTestBucket: tb.NoCloseClone(),
		serverless:       true,
		PersistentConfig: true,
		MutateStartupConfig: func(config *StartupConfig) {
			config.Bootstrap.ConfigUpdateFrequency = base.NewConfigDuration(0)
		},
	})
	defer rt.Close()
	sc := rt.ServerContext()

	resp := rt.SendAdminRequest(http.MethodPut, "/db/",
		fmt.Sprintf(
			`{"bucket": "%s", "num_index_replicas": 0, "enable_shared_bucket_access": %t, "use_views": %t}`,
			tb.GetName(), base.TestUseXattrs(), base.TestsDisableGSI(),
		),
	)
	RequireStatus(t, resp, http.StatusCreated)

	// Suspend the database and remove it from dbConfigs, forcing unsuspendDatabase to fetch config from the bucket
	err := sc.suspendDatabase(t, rt.Context(), "db")
	assert.NoError(t, err)
	delete(sc._dbConfigs, "db")
	delete(sc._dbRegistry, "db")
	assert.Nil(t, sc._databases["db"])

	// Unsuspend db and confirm unsuspending worked
	dbCtx, err := sc.GetDatabase(rt.Context(), "db")
	assert.NoError(t, err)
	assert.NotNil(t, dbCtx)
	assert.NotNil(t, sc._databases["db"])

	// Attempt to get invalid database
	_, err = sc.GetDatabase(rt.Context(), "invalid")
	assert.Contains(t, err.Error(), "no such database")
}

// Confirms that ServerContext.fetchConfigsWithTTL works correctly
func TestServerlessFetchConfigsLimited(t *testing.T) {