chrome/browser/supervised_user/permission_request_creator_mock.h

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

#ifndef CHROME_BROWSER_SUPERVISED_USER_PERMISSION_REQUEST_CREATOR_MOCK_H_
#define CHROME_BROWSER_SUPERVISED_USER_PERMISSION_REQUEST_CREATOR_MOCK_H_

#include <vector>

#include "base/memory/raw_ptr.h"
#include "components/supervised_user/core/browser/permission_request_creator.h"
#include "url/gurl.h"

class Profile;

class PermissionRequestCreatorMock
    : public supervised_user::PermissionRequestCreator {
 public:
  explicit PermissionRequestCreatorMock(Profile* profile);

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

  ~PermissionRequestCreatorMock() override;

  // PermissionRequestCreator:
  bool IsEnabled() const override;
  void CreateURLAccessRequest(const GURL& url_requested,
                              SuccessCallback callback) override;

  // Sets whether the next call to create requests will succeed.
  void SetPermissionResult(bool result);
  void SetEnabled();

  // Delays approvals for incoming requests until
  // |PermissionRequestCreatorMock::HandleDelayedRequests| is called.
  void DelayHandlingForNextRequests();
  void HandleDelayedRequests();

  // Getter methods.
  const std::vector<GURL>& url_requests() const { return url_requests_; }

 private:
  void CreateURLAccessRequestImpl(const GURL& url_requested);

  bool result_ = false;
  bool enabled_ = false;
  bool delay_handling_ = false;
  int last_url_request_handled_index_ = 0;

  raw_ptr<Profile, ExperimentalAsh> profile_ = nullptr;

  std::vector<GURL> url_requests_;
};

#endif  // CHROME_BROWSER_SUPERVISED_USER_PERMISSION_REQUEST_CREATOR_MOCK_H_