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

#ifndef COMPONENTS_DOWNLOAD_INTERNAL_BACKGROUND_SERVICE_TEST_TEST_DOWNLOAD_DRIVER_H_
#define COMPONENTS_DOWNLOAD_INTERNAL_BACKGROUND_SERVICE_TEST_TEST_DOWNLOAD_DRIVER_H_

#include <map>
#include <memory>

#include "base/memory/raw_ptr.h"
#include "components/download/internal/background_service/download_driver.h"
#include "net/traffic_annotation/network_traffic_annotation.h"

namespace download {
namespace test {

// Download driver that simulates content layer download logic.
class TestDownloadDriver : public DownloadDriver {
 public:
  TestDownloadDriver();

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

  ~TestDownloadDriver() override;

  // Marks download driver as ready, used to test logic that depends on
  // data initialization.
  void MakeReady();
  void TriggerHardRecoverComplete(bool success);

  // Adds driver entries data that will be returned
  void AddTestData(const std::vector<DriverEntry>& entries);

  // Simulates download events from content layer.
  void NotifyDownloadUpdate(const DriverEntry& entry);
  void NotifyDownloadFailed(const DriverEntry& entry, FailureType failure_type);
  void NotifyDownloadSucceeded(const DriverEntry& entry);

  // DownloadDriver implementation.
  void Initialize(DownloadDriver::Client* client) override;
  void HardRecover() override;
  bool IsReady() const override;
  void Start(
      const RequestParams& params,
      const std::string& guid,
      const base::FilePath& file_path,
      scoped_refptr<network::ResourceRequestBody> post_body,
      const net::NetworkTrafficAnnotationTag& traffic_annotation) override;
  void Remove(const std::string& guid, bool remove_file) override;
  void Pause(const std::string& guid) override;
  void Resume(const std::string& guid) override;
  std::optional<DriverEntry> Find(const std::string& guid) override;
  std::set<std::string> GetActiveDownloads() override;
  size_t EstimateMemoryUsage() const override;

 private:
  bool is_ready_;
  raw_ptr<DownloadDriver::Client> client_;

  // Map of guid --> DriverEntry.
  std::map<std::string, DriverEntry> entries_;
};

}  // namespace test
}  // namespace download

#endif  // COMPONENTS_DOWNLOAD_INTERNAL_BACKGROUND_SERVICE_TEST_TEST_DOWNLOAD_DRIVER_H_