Function TestInvalidateChannels

auth/auth_test.go:2861–2958

(t *testing.T)

}

func TestInvalidateChannels(t *testing.T) {
	testCases := []struct {
		name   string
		isUser bool
	}{
		{
			name:   "User",
			isUser: true,
		},
		{
			name:   "Role",
			isUser: false,
		},
	}

	for _, testCase := range testCases {
		t.Run(testCase.name, func(t *testing.T) {
			ctx := base.TestCtx(t)
			bucket := base.GetTestBucket(t)
			defer bucket.Close(ctx)

			leakyBucket := base.NewLeakyBucket(bucket, base.LeakyBucketConfig{})
			leakyDataStore, ok := base.AsLeakyDataStore(leakyBucket.DefaultDataStore())
			require.True(t, ok)

			auth := NewTestAuthenticator(t, leakyDataStore, nil, DefaultAuthenticatorOptions(ctx))

			// Invalidate channels on non-existent user / role and ensure no error
			err := auth.InvalidateDefaultChannels(testCase.name, testCase.isUser, 0)
			assert.NoError(t, err)

			// Create user / role
			var princ Principal
			if testCase.isUser {
				princ, err = auth.NewUser(testCase.name, "pass", nil)
			} else {
				princ, err = auth.NewRole(testCase.name, nil)
			}
			assert.NoError(t, err)

			err = auth.Save(princ)
			assert.NoError(t, err)

			// Invalidate channels at invalSeq but cause cas retry by setting to 5
			enableRetry := false
			// If subdoc operations are supported, perform an initial invalidation at seq 5
			if leakyBucket.IsSupported(sgbucket.BucketStoreFeatureSubdocOperations) {
				err := auth.InvalidateDefaultChannels(testCase.name, testCase.isUser, 5)
				assert.NoError(t, err)

			} else {
				// If subdoc ops aren't supported, use leakyBucket to invalidate
				leakyDataStore.SetUpdateCallback(func(key string) {
					if enableRetry {
						enableRetry = false
						err = auth.InvalidateDefaultChannels(testCase.name, testCase.isUser, 5)
						assert.NoError(t, err)
					}