Function TestStorage_AuthorizeCode_ConcurrentConsume

internal/oauth/storage_test.go:137–184

(t *testing.T)

}

func TestStorage_AuthorizeCode_ConcurrentConsume(t *testing.T) {
	st, _, _, userID, clientID := setup(t)
	ctx := context.Background()
	now := time.Now().UTC().Truncate(time.Second)

	req := newRequester("req-conc", clientID, userID, now)
	if err := st.CreateAuthorizeCodeSession(ctx, "code-conc", req); err != nil {
		t.Fatalf("CreateAuthorizeCodeSession: %v", err)
	}

	const N = 16
	var (
		wg          sync.WaitGroup
		successes   atomic.Int32
		invalidated atomic.Int32
		other       atomic.Int32
		start       = make(chan struct{})
	)
	wg.Add(N)
	for i := 0; i < N; i++ {
		go func() {
			defer wg.Done()
			<-start
			err := st.InvalidateAuthorizeCodeSession(ctx, "code-conc")
			switch {
			case err == nil:
				successes.Add(1)
			case errors.Is(err, fosite.ErrInvalidatedAuthorizeCode):
				invalidated.Add(1)
			default:
				other.Add(1)
				t.Errorf("unexpected error: %v", err)
			}
		}()
	}
	close(start)
	wg.Wait()

	if got := successes.Load(); got != 1 {
		t.Errorf("successes = %d, want 1 (only one consume should win)", got)
	}
	if got := invalidated.Load(); got != N-1 {
		t.Errorf("invalidated = %d, want %d", got, N-1)
	}
	if got := other.Load(); got != 0 {
		t.Errorf("other = %d, want 0", got)
	}
}

func TestStorage_RotateRefreshToken_ConcurrentRotate(t *testing.T) {
	st, _, _, userID, clientID := setup(t)