70 lines
2.1 KiB
C++
70 lines
2.1 KiB
C++
// Copyright 2018 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 <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::vector<uint8_t>{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::vector<uint8_t> data(kLargeDataSize);
|
|
base::RandBytes(data.data(), kLargeDataSize);
|
|
|
|
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(data, out));
|
|
}
|
|
|
|
} // namespace big_buffer_unittest
|
|
} // namespace mojo_base
|