// Copyright 2016 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_TEST_MOCK_BACKGROUND_SYNC_CONTROLLER_H_
#define CONTENT_TEST_MOCK_BACKGROUND_SYNC_CONTROLLER_H_

#include <stdint.h>

#include "base/time/time.h"
#include "content/public/browser/background_sync_controller.h"
#include "content/public/browser/background_sync_parameters.h"
#include "url/origin.h"

namespace content {

// Mocks a BackgroundSyncController, tracking state for use in tests.
class MockBackgroundSyncController : public BackgroundSyncController {
 public:
  MockBackgroundSyncController();

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

  ~MockBackgroundSyncController() override;

  // BackgroundSyncController:
  void NotifyOneShotBackgroundSyncRegistered(const url::Origin& origin,
                                             bool can_fire,
                                             bool is_reregistered) override;
  void ScheduleBrowserWakeUpWithDelay(
      blink::mojom::BackgroundSyncType sync_type,
      base::TimeDelta delay) override;
  void CancelBrowserWakeup(blink::mojom::BackgroundSyncType sync_type) override;
  void GetParameterOverrides(BackgroundSyncParameters* parameters) override;
  base::TimeDelta GetNextEventDelay(
      const BackgroundSyncRegistration& registration,
      BackgroundSyncParameters* parameters,
      base::TimeDelta time_till_soonest_scheduled_event_for_origin) override;
  std::unique_ptr<BackgroundSyncController::BackgroundSyncEventKeepAlive>
  CreateBackgroundSyncEventKeepAlive() override;
  void NoteSuspendedPeriodicSyncOrigins(
      std::set<url::Origin> suspended_registrations) override;
  void NoteRegisteredPeriodicSyncOrigins(
      std::set<url::Origin> registered_origins) override;
  void AddToTrackedOrigins(const url::Origin& origin) override;
  void RemoveFromTrackedOrigins(const url::Origin& origin) override;

  int registration_count() const { return registration_count_; }
  const url::Origin& registration_origin() const {
    return registration_origin_;
  }
  int run_in_background_count() const {
    return run_in_background_for_one_shot_sync_count_;
  }
  int run_in_background_periodic_sync_count() const {
    return run_in_background_for_periodic_sync_count_;
  }

  base::TimeDelta GetBrowserWakeupDelay(
      blink::mojom::BackgroundSyncType sync_type) const {
    if (sync_type == blink::mojom::BackgroundSyncType::ONE_SHOT)
      return one_shot_sync_browser_wakeup_delay_;
    else
      return periodic_sync_browser_wakeup_delay_;
  }

  BackgroundSyncParameters* background_sync_parameters() {
    return &background_sync_parameters_;
  }
  void ReviveSuspendedPeriodicSyncOrigin(url::Origin origin_to_revive) {
    suspended_periodic_sync_origins_.erase(origin_to_revive);
  }

 private:
  void ApplyFieldTrialParamsOverrides();

  int registration_count_ = 0;
  url::Origin registration_origin_;

  int run_in_background_for_one_shot_sync_count_ = 0;
  int run_in_background_for_periodic_sync_count_ = 0;
  base::TimeDelta periodic_sync_browser_wakeup_delay_;
  base::TimeDelta one_shot_sync_browser_wakeup_delay_;
  BackgroundSyncParameters background_sync_parameters_;
  std::set<url::Origin> suspended_periodic_sync_origins_;
  std::set<url::Origin> periodic_sync_origins_;
};

}  // namespace content

#endif  // CONTENT_TEST_MOCK_BACKGROUND_SYNC_CONTROLLER_H_