Function TestUpdatePrincipalCASRetry

db/database_test.go:1896–1987

(t *testing.T)

}

func TestUpdatePrincipalCASRetry(t *testing.T) {
	base.SetUpTestLogging(t, base.LevelDebug, base.KeyAuth, base.KeyCRUD)

	// ensure we don't batch sequences so that the number of released sequences is deterministic
	defer SuspendSequenceBatching()()

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

	tests := []struct {
		numCASRetries int32
		expectError   bool
	}{
		{numCASRetries: 0},
		{numCASRetries: 1},
		{numCASRetries: 2},
		{numCASRetries: 5},
		{numCASRetries: 10},
		{numCASRetries: auth.PrincipalUpdateMaxCasRetries - 1},
		{numCASRetries: auth.PrincipalUpdateMaxCasRetries, expectError: true},
		{numCASRetries: auth.PrincipalUpdateMaxCasRetries + 1, expectError: true},
	}

	var (
		casRetryCount   atomic.Int32
		totalCASRetries atomic.Int32
		enableCASRetry  base.AtomicBool
	)

	lb := base.NewLeakyBucket(tb, base.LeakyBucketConfig{
		UpdateCallback: func(key string) {
			casRetryCountInt, totalCASRetriesInt := casRetryCount.Load(), totalCASRetries.Load()
			if enableCASRetry.IsTrue() && casRetryCountInt < totalCASRetriesInt {
				casRetryCount.Add(1)
				casRetryCountInt = casRetryCount.Load()
				t.Logf("foreceCASRetry %d/%d: Forcing CAS retry for key: %q", casRetryCountInt, totalCASRetriesInt, key)
				var body []byte
				originalCAS, err := tb.GetMetadataStore().Get(key, &body)
				require.NoError(t, err)
				err = tb.GetMetadataStore().Set(key, 0, nil, body)
				require.NoError(t, err)
				newCAS, err := tb.GetMetadataStore().Get(key, &body)
				require.NoError(t, err)
				t.Logf("foreceCASRetry %d/%d: Doc %q CAS changed from %d to %d", casRetryCountInt, totalCASRetriesInt, key, originalCAS, newCAS)
			}
		},
		IgnoreClose: true,
	})

	db, ctx := setupTestDBForBucket(t, lb)
	defer db.Close(ctx)

	// Create a user with access to channel ABC
	authenticator := db.Authenticator(ctx)
	user, err := authenticator.NewUser("naomi", "letmein", channels.BaseSetOf(t, "ABC"))
	require.NoError(t, err)
	require.NoError(t, authenticator.Save(user))