Function TestCASUpdatePrincipal

auth/auth_test.go:635–699

(t *testing.T)

}

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

	dataStore := bucket.GetSingleDataStore()

	// Create user
	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, err := auth.NewUser(username, password, ch.BaseSetOf(t, "123", "456"))
	require.NoError(t, err)
	user.SetExplicitRoles(ch.TimedSet{"role1": ch.NewVbSimpleSequence(1), "role2": ch.NewVbSimpleSequence(1)}, 1)
	require.NoError(t, auth.Save(user))
	user, err = auth.GetUser(username)
	require.NoError(t, err)
	require.NotNilf(t, user, "User is nil prior to invalidate channels: %v", err)

	// updateEmailWithConflictCallback causes CAS failure three times
	updateCount := uint64(0)
	updateEmailWithConflictCallback := func(currentPrincipal Principal) (updatedPrincipal Principal, err error) {
		currentUser, ok := currentPrincipal.(User)
		if !ok {
			return nil, base.ErrUpdateCancel
		}

		log.Printf("attempting update with CAS:%v", currentUser.Cas())
		if updateCount < 3 {
			// Update principal externally to trigger CAS error three times
			concurrentUser, err := auth.GetUser(username)
			assert.NoError(t, err)
			log.Printf("setting explicit channels to %v", updateCount)
			concurrentUser.SetExplicitChannels(ch.TimedSet{"ch1": ch.NewVbSimpleSequence(updateCount)}, updateCount)
			updateErr := auth.Save(concurrentUser)
			assert.NoError(t, updateErr)

			updatedConcurrentUser, err := auth.GetUser(username)
			assert.NoError(t, err)
			log.Printf("Forcing cas failure, updated CAS: %v", updatedConcurrentUser.Cas())

			updateCount++
		}

		err = currentUser.SetEmail(email)
		if err != nil {
			return nil, err
		}
		return currentUser, nil
	}