third_party/blink/renderer/platform/loader/testing/test_resource_fetcher_properties.h

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

#ifndef THIRD_PARTY_BLINK_RENDERER_PLATFORM_LOADER_TESTING_TEST_RESOURCE_FETCHER_PROPERTIES_H_
#define THIRD_PARTY_BLINK_RENDERER_PLATFORM_LOADER_TESTING_TEST_RESOURCE_FETCHER_PROPERTIES_H_

#include "base/memory/scoped_refptr.h"
#include "third_party/blink/renderer/platform/heap/handle.h"
#include "third_party/blink/renderer/platform/loader/fetch/resource_fetcher_properties.h"

namespace blink {

class FetchClientSettingsObject;
class SecurityOrigin;

// TestResourceFetcherProperties is a ResourceFetcherProperties implementation
// for tests.
class TestResourceFetcherProperties final : public ResourceFetcherProperties {
 public:
  TestResourceFetcherProperties();
  explicit TestResourceFetcherProperties(scoped_refptr<const SecurityOrigin>);
  explicit TestResourceFetcherProperties(const FetchClientSettingsObject&);
  ~TestResourceFetcherProperties() override = default;

  void Trace(Visitor* visitor) override;

  // ResourceFetcherProperties implementation
  const FetchClientSettingsObject& GetFetchClientSettingsObject()
      const override {
    return *fetch_client_settings_object_;
  }
  bool IsMainFrame() const override { return is_main_frame_; }
  ControllerServiceWorkerMode GetControllerServiceWorkerMode() const override {
    return service_worker_mode_;
  }
  int64_t ServiceWorkerId() const override {
    DCHECK_NE(GetControllerServiceWorkerMode(),
              ControllerServiceWorkerMode::kNoController);
    return service_worker_id_;
  }
  bool IsPaused() const override { return paused_; }
  bool IsDetached() const override { return false; }
  bool IsLoadComplete() const override { return load_complete_; }
  bool ShouldBlockLoadingSubResource() const override {
    return should_block_loading_sub_resource_;
  }

  void SetIsMainFrame(bool value) { is_main_frame_ = value; }
  void SetControllerServiceWorkerMode(ControllerServiceWorkerMode mode) {
    service_worker_mode_ = mode;
  }
  void SetServiceWorkerId(int64_t id) { service_worker_id_ = id; }
  void SetIsPaused(bool value) { paused_ = value; }
  void SetIsLoadComplete(bool value) { load_complete_ = value; }
  void SetShouldBlockLoadingSubResource(bool value) {
    should_block_loading_sub_resource_ = value;
  }

 private:
  const Member<const FetchClientSettingsObject> fetch_client_settings_object_;
  bool is_main_frame_ = false;
  ControllerServiceWorkerMode service_worker_mode_ =
      ControllerServiceWorkerMode::kNoController;
  int64_t service_worker_id_ = 0;
  bool paused_ = false;
  bool load_complete_ = false;
  bool should_block_loading_sub_resource_ = false;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_PLATFORM_LOADER_TESTING_TEST_RESOURCE_FETCHER_PROPERTIES_H_