1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
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_impl.h"
using testing::_;
using testing::Invoke;
using testing::StrictMock;
using testing::WithArgs;
namespace blink {
class WebIDBTransactionImplTest : public testing::Test {};
TEST_F(WebIDBTransactionImplTest, 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);
WebIDBTransactionImpl transaction_impl(
blink::scheduler::GetSingleThreadTaskRunnerForTesting(), transaction_id);
transaction_impl.max_put_value_size_ = kMaxValueSizeForTesting;
transaction_impl.Put(object_store_id, std::move(value), std::move(key),
mojom::IDBPutMode::AddOrUpdate, std::move(callbacks),
Vector<IDBIndexKeys>());
}
TEST_F(WebIDBTransactionImplTest, 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);
WebIDBTransactionImpl transaction_impl(
blink::scheduler::GetSingleThreadTaskRunnerForTesting(), transaction_id);
transaction_impl.max_put_value_size_ = kMaxValueSizeForTesting;
transaction_impl.Put(object_store_id, std::move(value), std::move(key),
mojom::IDBPutMode::AddOrUpdate, std::move(callbacks),
Vector<IDBIndexKeys>());
}
} // namespace blink
|
