1 files changed, 100 insertions, 0 deletions
diff --git a/chromium/third_party/blink/renderer/modules/indexeddb/web_idb_transaction_unittest.cc b/chromium/third_party/blink/renderer/modules/indexeddb/web_idb_transaction_unittest.cc
new file mode 100644
index 00000000000..3ef75f65688
--- /dev/null
+++ b/chromium/third_party/blink/renderer/modules/indexeddb/web_idb_transaction_unittest.cc
@@ -0,0 +1,100 @@
+// Copyright 2013 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include <stddef.h>
+#include <stdint.h>
+
+#include <memory>
+#include <utility>
+
+#include "base/macros.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/blink/public/platform/scheduler/test/renderer_scheduler_test_support.h"
+#include "third_party/blink/public/platform/web_blob_info.h"
+#include "third_party/blink/public/web/web_heap.h"
+#include "third_party/blink/renderer/modules/indexeddb/idb_key_range.h"
+#include "third_party/blink/renderer/modules/indexeddb/mock_web_idb_callbacks.h"
+#include "third_party/blink/renderer/modules/indexeddb/web_idb_transaction.h"
+
+using testing::_;
+using testing::Invoke;
+using testing::StrictMock;
+using testing::WithArgs;
+
+namespace blink {
+
+class WebIDBTransactionTest : public testing::Test {};
+
+TEST_F(WebIDBTransactionTest, ValueSizeTest) {
+  // For testing use a much smaller maximum size to prevent allocating >100 MB
+  // of memory, which crashes on memory-constrained systems.
+  const size_t kMaxValueSizeForTesting = 10 * 1024 * 1024;  // 10 MB
+
+  const Vector<char> data(kMaxValueSizeForTesting + 1);
+  const scoped_refptr<SharedBuffer> value_data =
+      SharedBuffer::Create(&data.front(), data.size());
+  const Vector<WebBlobInfo> blob_info;
+  auto value = std::make_unique<IDBValue>(value_data, blob_info);
+  std::unique_ptr<IDBKey> key = IDBKey::CreateNumber(0);
+  const int64_t transaction_id = 1;
+  const int64_t object_store_id = 2;
+  std::unique_ptr<StrictMock<MockWebIDBCallbacks>> callbacks =
+      std::make_unique<StrictMock<MockWebIDBCallbacks>>();
+
+  ASSERT_GT(value_data->size() + key->SizeEstimate(), kMaxValueSizeForTesting);
+  ThreadState::Current()->CollectAllGarbageForTesting();
+  EXPECT_CALL(*callbacks, Error(_, _)).Times(1);
+
+  WebIDBTransaction transaction(
+      blink::scheduler::GetSingleThreadTaskRunnerForTesting(), transaction_id);
+  transaction.max_put_value_size_ = kMaxValueSizeForTesting;
+  transaction.Put(object_store_id, std::move(value), std::move(key),
+                  mojom::IDBPutMode::AddOrUpdate, std::move(callbacks),
+                  Vector<IDBIndexKeys>());
+}
+
+TEST_F(WebIDBTransactionTest, KeyAndValueSizeTest) {
+  // For testing use a much smaller maximum size to prevent allocating >100 MB
+  // of memory, which crashes on memory-constrained systems.
+  const size_t kMaxValueSizeForTesting = 10 * 1024 * 1024;  // 10 MB
+  const size_t kKeySize = 1024 * 1024;
+
+  const Vector<char> data(kMaxValueSizeForTesting - kKeySize);
+  const scoped_refptr<SharedBuffer> value_data =
+      SharedBuffer::Create(&data.front(), data.size());
+  const Vector<WebBlobInfo> blob_info;
+  auto value = std::make_unique<IDBValue>(value_data, blob_info);
+  const int64_t transaction_id = 1;
+  const int64_t object_store_id = 2;
+  std::unique_ptr<StrictMock<MockWebIDBCallbacks>> callbacks =
+      std::make_unique<StrictMock<MockWebIDBCallbacks>>();
+
+  // For this test, we want IDBKey::SizeEstimate() minus kKeySize to be the
+  // smallest value > 0.  An IDBKey with a string has a size_estimate_ equal to
+  // kOverheadSize (~16) + (string.length * sizeof(UChar)).  Create
+  // |kKeySize / sizeof(UChar)| characters in String.
+  const unsigned int number_of_chars = kKeySize / sizeof(UChar);
+  Vector<UChar> key_string_vector;
+  key_string_vector.ReserveInitialCapacity(number_of_chars);
+  key_string_vector.Fill(u'0', number_of_chars);
+  String key_string(key_string_vector);
+  DCHECK_EQ(key_string.length(), number_of_chars);
+
+  std::unique_ptr<IDBKey> key = IDBKey::CreateString(key_string);
+  DCHECK_EQ(value_data->size(), kMaxValueSizeForTesting - kKeySize);
+  DCHECK_GT(key->SizeEstimate() - kKeySize, static_cast<size_t>(0));
+  DCHECK_GT(value_data->size() + key->SizeEstimate(), kMaxValueSizeForTesting);
+
+  ThreadState::Current()->CollectAllGarbageForTesting();
+  EXPECT_CALL(*callbacks, Error(_, _)).Times(1);
+
+  WebIDBTransaction transaction(
+      blink::scheduler::GetSingleThreadTaskRunnerForTesting(), transaction_id);
+  transaction.max_put_value_size_ = kMaxValueSizeForTesting;
+  transaction.Put(object_store_id, std::move(value), std::move(key),
+                  mojom::IDBPutMode::AddOrUpdate, std::move(callbacks),
+                  Vector<IDBIndexKeys>());
+}
+
+}  // namespace blink