Function TestServerlessFetchConfigsLimited

rest/serverless_test.go:394–472

Confirms that ServerContext.fetchConfigsWithTTL works correctly

(t *testing.T)

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

	ctx := base.TestCtx(t)
	tb := base.GetTestBucket(t)
	defer tb.Close(ctx)

	rt := NewRestTester(t, &RestTesterConfig{
		CustomTestBucket: tb.NoCloseClone(),
		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)

	// Purposely make configs get caches
	sc.Config.Unsupported.Serverless.MinConfigFetchInterval = base.NewConfigDuration(time.Hour)
	dbConfigsBefore, err := sc.fetchConfigsSince(rt.Context(), sc.Config.Unsupported.Serverless.MinConfigFetchInterval)
	require.NoError(t, err)
	require.NotEmpty(t, dbConfigsBefore["db"])
	timeCached := sc.fetchConfigsLastUpdate
	require.NoError(t, err)
	assert.NotZero(t, timeCached)

	require.NoError(t, err)
	// Update database config in the bucket (increment version)
	newCas, err := sc.BootstrapContext.UpdateConfig(ctx, tb.GetName(), sc.Config.Bootstrap.ConfigGroupID, "db", func(bucketDbConfig *DatabaseConfig) (updatedConfig *DatabaseConfig, err error) {
		bucketDbConfig.Version, err = GenerateDatabaseConfigVersionID(rt.Context(), bucketDbConfig.Version, &bucketDbConfig.DbConfig)
		require.NoError(t, err)
		return bucketDbConfig, nil
	})
	require.NoError(t, err)

	// Fetch configs again and expect same config to be returned
	dbConfigsAfter, err := sc.fetchConfigsSince(rt.Context(), sc.Config.Unsupported.Serverless.MinConfigFetchInterval)
	require.NoError(t, err)
	require.NotEmpty(t, dbConfigsAfter["db"])
	assert.Equal(t, dbConfigsBefore["db"].cfgCas, dbConfigsAfter["db"].cfgCas)
	assert.Equal(t, timeCached, sc.fetchConfigsLastUpdate)

	// Make caching 1ms so it will grab newest config
	sc.Config.Unsupported.Serverless.MinConfigFetchInterval = base.NewConfigDuration(time.Millisecond)
	// Sleep to make sure enough time passes
	time.Sleep(time.Millisecond * 500)
	dbConfigsAfter, err = sc.fetchConfigsSince(rt.Context(), sc.Config.Unsupported.Serverless.MinConfigFetchInterval)
	require.NoError(t, err)
	require.NotEmpty(t, dbConfigsAfter["db"])
	assert.Equal(t, newCas, dbConfigsAfter["db"].cfgCas)