// Copyright 2013 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_BROWSER_INDEXED_DB_INSTANCE_FAKE_TRANSACTION_H_
#define CONTENT_BROWSER_INDEXED_DB_INSTANCE_FAKE_TRANSACTION_H_

#include <stdint.h>

#include <memory>

#include "content/browser/indexed_db/instance/backing_store.h"

namespace content::indexed_db {

class FakeTransaction : public BackingStore::Transaction {
 public:
  FakeTransaction(Status phase_two_result,
                  std::unique_ptr<BackingStore::Transaction> wrapped);
  ~FakeTransaction() override;

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

  Status Begin(std::vector<PartitionedLock> locks) override;
  StatusOr<bool> CommitPhaseOne(BlobWriteCallback,
                                SerializeFsaCallback) override;
  Status CommitPhaseTwo() override;
  void Rollback() override;
  Status SetDatabaseVersion(int64_t version) override;
  Status CreateObjectStore(int64_t object_store_id,
                           const std::u16string& name,
                           blink::IndexedDBKeyPath key_path,
                           bool auto_increment) override;
  Status DeleteObjectStore(int64_t object_store_id) override;
  Status RenameObjectStore(int64_t object_store_id,
                           const std::u16string& new_name) override;
  Status ClearObjectStore(int64_t object_store_id) override;
  Status CreateIndex(int64_t object_store_id,
                     blink::IndexedDBIndexMetadata index) override;
  Status DeleteIndex(int64_t object_store_id, int64_t index_id) override;
  Status RenameIndex(int64_t object_store_id,
                     int64_t index_id,
                     const std::u16string& new_name) override;
  StatusOr<IndexedDBValue> GetRecord(int64_t object_store_id,
                                     const blink::IndexedDBKey& key) override;
  StatusOr<BackingStore::RecordIdentifier> PutRecord(
      int64_t object_store_id,
      const blink::IndexedDBKey& key,
      IndexedDBValue value) override;
  Status DeleteRange(int64_t object_store_id,
                     const blink::IndexedDBKeyRange&) override;
  StatusOr<int64_t> GetKeyGeneratorCurrentNumber(
      int64_t object_store_id) override;
  Status MaybeUpdateKeyGeneratorCurrentNumber(int64_t object_store_id,
                                              int64_t new_number,
                                              bool was_generated) override;
  StatusOr<std::optional<BackingStore::RecordIdentifier>>
  KeyExistsInObjectStore(int64_t object_store_id,
                         const blink::IndexedDBKey& key) override;
  Status PutIndexDataForRecord(
      int64_t object_store_id,
      int64_t index_id,
      const blink::IndexedDBKey& key,
      const BackingStore::RecordIdentifier& record) override;
  StatusOr<blink::IndexedDBKey> GetFirstPrimaryKeyForIndexKey(
      int64_t object_store_id,
      int64_t index_id,
      const blink::IndexedDBKey& key) override;
  StatusOr<uint32_t> GetObjectStoreKeyCount(
      int64_t object_store_id,
      blink::IndexedDBKeyRange key_range) override;
  StatusOr<uint32_t> GetIndexKeyCount(
      int64_t object_store_id,
      int64_t index_id,
      blink::IndexedDBKeyRange key_range) override;
  StatusOr<std::unique_ptr<indexed_db::BackingStore::Cursor>>
  OpenObjectStoreKeyCursor(int64_t object_store_id,
                           const blink::IndexedDBKeyRange& key_range,
                           blink::mojom::IDBCursorDirection) override;
  StatusOr<std::unique_ptr<indexed_db::BackingStore::Cursor>>
  OpenObjectStoreCursor(int64_t object_store_id,
                        const blink::IndexedDBKeyRange& key_range,
                        blink::mojom::IDBCursorDirection) override;
  StatusOr<std::unique_ptr<indexed_db::BackingStore::Cursor>>
  OpenIndexKeyCursor(int64_t object_store_id,
                     int64_t index_id,
                     const blink::IndexedDBKeyRange& key_range,
                     blink::mojom::IDBCursorDirection) override;
  StatusOr<std::unique_ptr<indexed_db::BackingStore::Cursor>> OpenIndexCursor(
      int64_t object_store_id,
      int64_t index_id,
      const blink::IndexedDBKeyRange& key_range,
      blink::mojom::IDBCursorDirection) override;
  blink::mojom::IDBValuePtr BuildMojoValue(
      IndexedDBValue value,
      DeserializeFsaCallback deserialize_fsa_handle) override;

 private:
  Status result_;
  std::unique_ptr<BackingStore::Transaction> wrapped_transaction_;
};

}  // namespace content::indexed_db

#endif  // CONTENT_BROWSER_INDEXED_DB_INSTANCE_FAKE_TRANSACTION_H_