// Copyright 2014 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_SESSIONS_CORE_COMMAND_STORAGE_MANAGER_TEST_HELPER_H_
#define COMPONENTS_SESSIONS_CORE_COMMAND_STORAGE_MANAGER_TEST_HELPER_H_

#include <memory>
#include <vector>

#include "base/functional/callback.h"
#include "base/location.h"
#include "base/memory/raw_ptr.h"
#include "base/task/sequenced_task_runner.h"

namespace sessions {
class SessionCommand;
class CommandStorageManager;
class CommandStorageBackend;

class CommandStorageManagerTestHelper {
 public:
  explicit CommandStorageManagerTestHelper(
      CommandStorageManager* command_storage_manager);

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

  ~CommandStorageManagerTestHelper() = default;

  CommandStorageBackend* GetCleartextBackend();
  CommandStorageBackend* GetEncryptedBackend();

  // This posts the task to the SequencedWorkerPool, or run immediately
  // if the SequencedWorkerPool has been shutdown.
  void RunTaskOnBackendThread(const base::Location& from_here,
                              base::OnceClosure task);

  void RunMessageLoopUntilBackendDone();

  // Returns true if any commands got processed yet - saved or queued.
  bool ProcessedAnyCommands();

  // Read the last session commands directly from file.
  std::vector<std::unique_ptr<SessionCommand>> ReadLastSessionCommands();

  scoped_refptr<base::SequencedTaskRunner> GetBackendTaskRunner();

  // Returns true if the CommandStorageManager should write cleartext files.
  bool ShouldWriteCleartextFiles();

  // Returns true if the CommandStorageManager should write encrypted files.
  bool ShouldWriteEncryptedFiles();

  // See function of same name in CommandStorageBackend for details.
  void ForceAppendCommandsToFailForTesting();

 private:
  raw_ptr<CommandStorageManager> command_storage_manager_;
};

}  // namespace sessions

#endif  // COMPONENTS_SESSIONS_CORE_COMMAND_STORAGE_MANAGER_TEST_HELPER_H_