chrome/browser/ash/file_manager/trash_unittest_base.h

// Copyright 2022 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_ASH_FILE_MANAGER_TRASH_UNITTEST_BASE_H_
#define CHROME_BROWSER_ASH_FILE_MANAGER_TRASH_UNITTEST_BASE_H_

#include <memory>

#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/ash/crostini/crostini_manager.h"
#include "chrome/browser/ash/drive/drive_integration_service.h"
#include "chrome/browser/ash/drive/drivefs_test_support.h"
#include "chrome/browser/ash/file_manager/fake_disk_mount_manager.h"
#include "chrome/test/base/testing_profile.h"
#include "components/user_manager/scoped_user_manager.h"
#include "content/public/test/browser_task_environment.h"
#include "storage/browser/file_system/file_system_context.h"
#include "storage/browser/file_system/file_system_url.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/storage_key/storage_key.h"

namespace file_manager::io_task {

inline constexpr size_t kTestFileSize = 32;

class TrashBaseTest : public testing::Test {
  TrashBaseTest(const TrashBaseTest&) = delete;
  TrashBaseTest& operator=(const TrashBaseTest&) = delete;

 protected:
  TrashBaseTest();
  ~TrashBaseTest() override;

  void SetUp() override;
  void TearDown() override;

  drive::DriveIntegrationService* CreateDriveIntegrationService(
      Profile* profile);

  storage::FileSystemURL CreateFileSystemURL(
      const base::FilePath& absolute_path);

  const base::FilePath GenerateInfoPath(const std::string& file_name);

  const base::FilePath GenerateFilesPath(const std::string& file_name);

  const std::string CreateTrashInfoContentsFromPath(
      const base::FilePath& file_path,
      const base::FilePath& base_path,
      const base::FilePath& prefix_path);

  const std::string CreateTrashInfoContentsFromPath(
      const base::FilePath& file_path);

  bool EnsureTrashDirectorySetup(const base::FilePath& parent_path);

  content::BrowserTaskEnvironment task_environment_;
  std::unique_ptr<TestingProfile> profile_;
  const blink::StorageKey kTestStorageKey =
      blink::StorageKey::CreateFromStringForTesting("chrome-extension://abc");

  // DriveFS setup methods to ensure the tests have access to a mock
  // DriveIntegrationService tied to the TestingProfile.
  std::unique_ptr<user_manager::ScopedUserManager> scoped_user_manager_;
  std::unique_ptr<drive::FakeDriveFsHelper> fake_drivefs_helper_;
  raw_ptr<drive::DriveIntegrationService, ExperimentalAsh>
      integration_service_ = nullptr;
  drive::DriveIntegrationServiceFactory::FactoryCallback
      create_drive_integration_service_;
  std::unique_ptr<drive::DriveIntegrationServiceFactory::ScopedFactoryForTest>
      service_factory_for_test_;

  raw_ptr<crostini::CrostiniManager, ExperimentalAsh> crostini_manager_;
  file_manager::FakeDiskMountManager disk_mount_manager_;

  base::ScopedTempDir temp_dir_;
  base::FilePath downloads_dir_;
  base::FilePath my_files_dir_;
  base::FilePath drive_dir_;
  base::FilePath crostini_dir_;
  base::FilePath crostini_remote_mount_;
  scoped_refptr<storage::FileSystemContext> file_system_context_;
};

}  // namespace file_manager::io_task

#endif  // CHROME_BROWSER_ASH_FILE_MANAGER_TRASH_IO_TASK_H_