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
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/351564777): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif
#include <algorithm>
#include <vector>
#include "base/rand_util.h"
#include "mojo/public/cpp/base/big_buffer.h"
#include "mojo/public/cpp/base/big_buffer_mojom_traits.h"
#include "mojo/public/cpp/test_support/test_utils.h"
#include "mojo/public/mojom/base/big_buffer.mojom.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace mojo_base {
namespace big_buffer_unittest {
namespace {
bool BufferEquals(const BigBuffer& a, const BigBuffer& b) {
return a.size() == b.size() && std::equal(a.data(), a.data() + a.size(),
b.data(), b.data() + b.size());
}
} // namespace
TEST(BigBufferTest, EmptyBuffer) {
BigBuffer in;
BigBuffer out;
EXPECT_EQ(BigBuffer::StorageType::kBytes, in.storage_type());
EXPECT_EQ(0u, in.size());
ASSERT_TRUE(mojo::test::SerializeAndDeserialize<mojom::BigBuffer>(in, out));
EXPECT_EQ(BigBuffer::StorageType::kBytes, out.storage_type());
EXPECT_TRUE(BufferEquals(in, out));
}
TEST(BigBufferTest, SmallDataSize) {
BigBuffer in(std::array<uint8_t, 3>{1, 2, 3});
EXPECT_EQ(BigBuffer::StorageType::kBytes, in.storage_type());
BigBuffer out;
ASSERT_TRUE(mojo::test::SerializeAndDeserialize<mojom::BigBuffer>(in, out));
EXPECT_EQ(BigBuffer::StorageType::kBytes, out.storage_type());
EXPECT_TRUE(BufferEquals(in, out));
}
TEST(BigBufferTest, LargeDataSize) {
constexpr size_t kLargeDataSize = BigBuffer::kMaxInlineBytes * 2;
std::array<uint8_t, kLargeDataSize> data;
base::RandBytes(data);
BigBuffer in(data);
EXPECT_EQ(BigBuffer::StorageType::kSharedMemory, in.storage_type());
BigBuffer out;
ASSERT_TRUE(mojo::test::SerializeAndDeserialize<mojom::BigBuffer>(in, out));
EXPECT_EQ(BigBuffer::StorageType::kSharedMemory, out.storage_type());
// NOTE: It's not safe to compare to |in| here since serialization will have
// taken ownership of its internal shared buffer handle.
EXPECT_TRUE(BufferEquals(base::span<const uint8_t>(data), out));
}
TEST(BigBufferTest, InvalidBuffer) {
// Verifies that deserializing invalid BigBuffers and BigBufferViews always
// fails.
BigBufferView out_view;
auto invalid_view = BigBufferView::CreateInvalidForTest();
EXPECT_EQ(invalid_view.storage_type(),
BigBuffer::StorageType::kInvalidBuffer);
EXPECT_FALSE(mojo::test::SerializeAndDeserialize<mojom::BigBuffer>(
invalid_view, out_view));
BigBuffer out_buffer;
auto invalid_buffer =
BigBufferView::ToBigBuffer(BigBufferView::CreateInvalidForTest());
EXPECT_EQ(invalid_buffer.storage_type(),
BigBuffer::StorageType::kInvalidBuffer);
EXPECT_FALSE(mojo::test::SerializeAndDeserialize<mojom::BigBuffer>(
invalid_buffer, out_buffer));
}
} // namespace big_buffer_unittest
} // namespace mojo_base
|
