Function TestConcurrentUserWrites

auth/auth_test.go:701–794

(t *testing.T)

}

func TestConcurrentUserWrites(t *testing.T) {
	ctx := base.TestCtx(t)
	bucket := base.GetTestBucket(t)
	defer bucket.Close(ctx)

	dataStore := bucket.GetSingleDataStore()

	username := "foo"
	password := "password"
	email := "foo@bar.org"

	// Create user
	auth := NewTestAuthenticator(t, dataStore, nil, DefaultAuthenticatorOptions(base.TestCtx(t)))

	// Modify the bcrypt cost to test rehashPassword properly below
	require.Error(t, auth.SetBcryptCost(5))

	user, _ := auth.NewUser(username, password, ch.BaseSetOf(t, "123", "456"))
	user.SetExplicitRoles(ch.TimedSet{"role1": ch.NewVbSimpleSequence(1), "role2": ch.NewVbSimpleSequence(1)}, 1)
	createErr := auth.Save(user)
	if createErr != nil {
		t.Errorf("Error creating user: %v", createErr)
	}

	// Retrieve user to trigger initial calculation of roles, channels
	_, getErr := auth.GetUser(username)
	require.NoError(t, getErr, "Error retrieving user")

	require.NoError(t, auth.SetBcryptCost(DefaultBcryptCost))
	// Reset bcryptCostChanged state after test runs
	defer func() {
		auth.bcryptCostChanged = false
	}()

	var wg sync.WaitGroup
	wg.Add(4)
	// Update user email, password hash, and invalidate user channels, roles concurrently
	go func() {
		user, getErr := auth.GetUser(username)
		require.NoError(t, getErr, "Error retrieving user prior to invalidate channels")
		require.NotNil(t, user, "User is nil prior to invalidate channels")

		invalidateErr := auth.InvalidateDefaultChannels(username, true, 1)
		assert.NoError(t, invalidateErr, "Error invalidating user's channels")
		wg.Done()
	}()

	go func() {
		user, getErr := auth.GetUser(username)
		require.NoError(t, getErr, "Error retrieving user")
		require.NotNil(t, user, "User is nil prior to email update")

		updateErr := auth.UpdateUserEmail(user, email)
		assert.NoError(t, updateErr, "Error updating user email")
		wg.Done()
	}()

	go func() {