1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83


content / browser / webid / test / federated_auth_request_request_token_callback_helper.h [blame]

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_WEBID_TEST_FEDERATED_AUTH_REQUEST_REQUEST_TOKEN_CALLBACK_HELPER_H_
#define CONTENT_BROWSER_WEBID_TEST_FEDERATED_AUTH_REQUEST_REQUEST_TOKEN_CALLBACK_HELPER_H_

#include <optional>
#include <string>

#include "base/functional/bind.h"
#include "base/functional/callback_forward.h"
#include "base/run_loop.h"
#include "content/browser/webid/federated_auth_request_impl.h"
#include "third_party/blink/public/mojom/webid/federated_auth_request.mojom.h"
#include "url/gurl.h"

namespace content {

// Helper class for waiting for the FederatedAuthRequestImpl::RequestToken()
// callback.
class FederatedAuthRequestRequestTokenCallbackHelper {
 public:
  FederatedAuthRequestRequestTokenCallbackHelper();
  ~FederatedAuthRequestRequestTokenCallbackHelper();

  FederatedAuthRequestRequestTokenCallbackHelper(
      const FederatedAuthRequestRequestTokenCallbackHelper&) = delete;
  FederatedAuthRequestRequestTokenCallbackHelper& operator=(
      const FederatedAuthRequestRequestTokenCallbackHelper&) = delete;

  std::optional<blink::mojom::RequestTokenStatus> status() const {
    return status_;
  }
  std::optional<GURL> selected_idp_config_url() const {
    return selected_idp_config_url_;
  }
  std::optional<std::string> token() const { return token_; }
  bool is_auto_selected() const { return is_auto_selected_; }

  // Returns base::OnceClosure which quits base::RunLoop started by
  // WaitForCallback().
  base::OnceClosure quit_closure() {
    return base::BindOnce(&FederatedAuthRequestRequestTokenCallbackHelper::Quit,
                          base::Unretained(this));
  }

  // This can only be called once per lifetime of this object.
  FederatedAuthRequestImpl::RequestTokenCallback callback() {
    return base::BindOnce(
        &FederatedAuthRequestRequestTokenCallbackHelper::ReceiverMethod,
        base::Unretained(this));
  }

  bool was_callback_called() const { return was_called_; }

  // Returns when callback() is called, which can be immediately if it has
  // already been called.
  void WaitForCallback();

  void Reset();

 private:
  void ReceiverMethod(blink::mojom::RequestTokenStatus status,
                      const std::optional<GURL>& selected_idp_config_url,
                      const std::optional<std::string>& token,
                      blink::mojom::TokenErrorPtr error,
                      bool is_auto_selected);

  void Quit();

  bool was_called_ = false;
  base::RunLoop wait_for_callback_loop_;
  std::optional<blink::mojom::RequestTokenStatus> status_;
  std::optional<GURL> selected_idp_config_url_;
  std::optional<std::string> token_;
  blink::mojom::TokenErrorPtr error_;
  bool is_auto_selected_{false};
};

}  // namespace content

#endif  // CONTENT_BROWSER_WEBID_TEST_FEDERATED_AUTH_REQUEST_REQUEST_TOKEN_CALLBACK_HELPER_H_