Function TestDatabaseStartupFailure

rest/server_context_test.go:952–998

(t *testing.T)

}

func TestDatabaseStartupFailure(t *testing.T) {
	if !base.IsEnterpriseEdition() {
		t.Skip("EE only test, requires heartbeater error")
	}
	if !base.UnitTestUrlIsWalrus() {
		t.Skip("LeakyBucketConfig not supported on CBS")
	}

	touchErr := errors.New("touch error")
	rt := NewRestTester(t, &RestTesterConfig{
		LeakyBucketConfig: &base.LeakyBucketConfig{
			TouchCallback: func(key string) error {
				if strings.Contains(key, "heartbeat_timeout") {
					return touchErr
				}
				return nil
			},
		},
		PersistentConfig: true,
	})
	defer rt.Close()

	dbConfig := DatabaseConfig{
		DbConfig: rt.NewDbConfig(),
	}

	// this call does use the leaky bucket
	dbContext, err := rt.ServerContext().AddDatabaseFromConfigWithBucket(rt.Context(), t, dbConfig, rt.Bucket())
	require.ErrorIs(t, err, touchErr)
	require.Nil(t, dbContext)
	require.Equal(t, "Offline", rt.GetDBState())

	require.Equal(t, int64(1), rt.GetDatabase().DbStats.Database().TotalOnlineFatalErrors.Value())
	require.Equal(t, int64(0), rt.GetDatabase().DbStats.Database().TotalInitFatalErrors.Value())
	// assert that the db is reported with appropriate error state in all dbs
	allDbs := rt.ServerContext().allDatabaseSummaries()
	require.Len(t, allDbs, 1)
	invalDb := allDbs[0]
	require.NotNil(t, invalDb.DatabaseError)
	assert.Equal(t, db.RunStateString[db.DBOffline], invalDb.State)
	assert.Equal(t, invalDb.DatabaseError.ErrMsg, db.DatabaseErrorMap[db.DatabaseOnlineProcessError])

	// assert that you can attempt again to bring db back online again after failure
	resp := rt.SendAdminRequest(http.MethodPost, "/"+invalDb.Bucket+"/_online", "")
	RequireStatus(t, resp, http.StatusOK)
	rt.WaitForDBOnline()
}

func TestDatabaseCollectionDeletedErrorState(t *testing.T) {
	if base.UnitTestUrlIsWalrus() {