| Index: device/u2f/u2f_message_unittest.cc
|
| diff --git a/device/u2f/u2f_message_unittest.cc b/device/u2f/u2f_message_unittest.cc
|
| index 00fc566df0ef76ab6293018f4c9f20783ec78ca6..947eb78750dbe92db72bec70fbd234fecb9817e4 100644
|
| --- a/device/u2f/u2f_message_unittest.cc
|
| +++ b/device/u2f/u2f_message_unittest.cc
|
| @@ -3,6 +3,7 @@
|
| // found in the LICENSE file.
|
|
|
| #include "device/u2f/u2f_message.h"
|
| +#include "base/memory/ptr_util.h"
|
| #include "net/base/io_buffer.h"
|
| #include "testing/gmock/include/gmock/gmock.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| @@ -16,12 +17,12 @@ TEST_F(U2fMessageTest, TestPacketSize) {
|
| uint32_t channel_id = 0x05060708;
|
| std::vector<uint8_t> data;
|
|
|
| - scoped_refptr<U2fPacket> init_packet(
|
| - new U2fInitPacket(channel_id, 0, data, data.size()));
|
| + auto init_packet =
|
| + base::MakeUnique<U2fInitPacket>(channel_id, 0, data, data.size());
|
| EXPECT_EQ(65, init_packet->GetSerializedBuffer()->size());
|
|
|
| - scoped_refptr<U2fPacket> continuation_packet(
|
| - new U2fContinuationPacket(channel_id, 0, data));
|
| + auto continuation_packet =
|
| + base::MakeUnique<U2fContinuationPacket>(channel_id, 0, data);
|
| EXPECT_EQ(65, continuation_packet->GetSerializedBuffer()->size());
|
| }
|
|
|
| @@ -39,8 +40,8 @@ TEST_F(U2fMessageTest, TestPacketData) {
|
| uint32_t channel_id = 0xF5060708;
|
| std::vector<uint8_t> data{10, 11};
|
| uint8_t cmd = static_cast<uint8_t>(U2fMessage::Type::CMD_WINK);
|
| - scoped_refptr<U2fPacket> init_packet(
|
| - new U2fInitPacket(channel_id, cmd, data, data.size()));
|
| + auto init_packet =
|
| + base::MakeUnique<U2fInitPacket>(channel_id, cmd, data, data.size());
|
| int index = 0;
|
|
|
| EXPECT_EQ(0, init_packet->GetSerializedBuffer()->data()[index++]);
|
| @@ -77,11 +78,11 @@ TEST_F(U2fMessageTest, TestPacketConstructors) {
|
| uint32_t channel_id = 0x05060708;
|
| std::vector<uint8_t> data{10, 11};
|
| uint8_t cmd = static_cast<uint8_t>(U2fMessage::Type::CMD_WINK);
|
| - scoped_refptr<U2fInitPacket> orig_packet(
|
| - new U2fInitPacket(channel_id, cmd, data, data.size()));
|
| + auto orig_packet =
|
| + base::MakeUnique<U2fInitPacket>(channel_id, cmd, data, data.size());
|
|
|
| size_t payload_length = static_cast<size_t>(orig_packet->payload_length());
|
| - scoped_refptr<U2fInitPacket> reconstructed_packet =
|
| + std::unique_ptr<U2fInitPacket> reconstructed_packet =
|
| U2fInitPacket::CreateFromSerializedData(
|
| orig_packet->GetSerializedBuffer(), &payload_length);
|
| EXPECT_EQ(orig_packet->command(), reconstructed_packet->command());
|
| @@ -109,10 +110,10 @@ TEST_F(U2fMessageTest, TestMaxLengthPacketConstructors) {
|
| data.push_back(static_cast<uint8_t>(i % 0xff));
|
|
|
| U2fMessage::Type cmd = U2fMessage::Type::CMD_MSG;
|
| - scoped_refptr<U2fMessage> orig_msg =
|
| + std::unique_ptr<U2fMessage> orig_msg =
|
| U2fMessage::Create(channel_id, cmd, data);
|
| auto it = orig_msg->begin();
|
| - scoped_refptr<U2fMessage> new_msg =
|
| + std::unique_ptr<U2fMessage> new_msg =
|
| U2fMessage::CreateFromSerializedData((*it)->GetSerializedBuffer());
|
| it++;
|
|
|
| @@ -143,18 +144,18 @@ TEST_F(U2fMessageTest, TestMaxLengthPacketConstructors) {
|
| TEST_F(U2fMessageTest, TestMessagePartitoning) {
|
| uint32_t channel_id = 0x01010203;
|
| std::vector<uint8_t> data(U2fMessage::kInitPacketDataSize + 1);
|
| - scoped_refptr<U2fMessage> two_packet_message =
|
| + std::unique_ptr<U2fMessage> two_packet_message =
|
| U2fMessage::Create(channel_id, U2fMessage::Type::CMD_PING, data);
|
| EXPECT_EQ(2U, two_packet_message->NumPackets());
|
|
|
| data.resize(U2fMessage::kInitPacketDataSize);
|
| - scoped_refptr<U2fMessage> one_packet_message =
|
| + std::unique_ptr<U2fMessage> one_packet_message =
|
| U2fMessage::Create(channel_id, U2fMessage::Type::CMD_PING, data);
|
| EXPECT_EQ(1U, one_packet_message->NumPackets());
|
|
|
| data.resize(U2fMessage::kInitPacketDataSize +
|
| U2fMessage::kContinuationPacketDataSize + 1);
|
| - scoped_refptr<U2fMessage> three_packet_message =
|
| + std::unique_ptr<U2fMessage> three_packet_message =
|
| U2fMessage::Create(channel_id, U2fMessage::Type::CMD_PING, data);
|
| EXPECT_EQ(3U, three_packet_message->NumPackets());
|
| }
|
| @@ -162,7 +163,7 @@ TEST_F(U2fMessageTest, TestMessagePartitoning) {
|
| TEST_F(U2fMessageTest, TestMaxSize) {
|
| uint32_t channel_id = 0x00010203;
|
| std::vector<uint8_t> data(U2fMessage::kMaxMessageSize + 1);
|
| - scoped_refptr<U2fMessage> oversize_message =
|
| + std::unique_ptr<U2fMessage> oversize_message =
|
| U2fMessage::Create(channel_id, U2fMessage::Type::CMD_PING, data);
|
| EXPECT_EQ(nullptr, oversize_message);
|
| }
|
| @@ -170,7 +171,7 @@ TEST_F(U2fMessageTest, TestMaxSize) {
|
| TEST_F(U2fMessageTest, TestDeconstruct) {
|
| uint32_t channel_id = 0x0A0B0C0D;
|
| std::vector<uint8_t> data(U2fMessage::kMaxMessageSize, 0x7F);
|
| - scoped_refptr<U2fMessage> filled_message =
|
| + std::unique_ptr<U2fMessage> filled_message =
|
| U2fMessage::Create(channel_id, U2fMessage::Type::CMD_PING, data);
|
|
|
| EXPECT_THAT(data, testing::ContainerEq(filled_message->GetMessagePayload()));
|
| @@ -180,13 +181,13 @@ TEST_F(U2fMessageTest, TestDeserialize) {
|
| uint32_t channel_id = 0x0A0B0C0D;
|
| std::vector<uint8_t> data(U2fMessage::kMaxMessageSize);
|
|
|
| - scoped_refptr<U2fMessage> orig_message =
|
| + std::unique_ptr<U2fMessage> orig_message =
|
| U2fMessage::Create(channel_id, U2fMessage::Type::CMD_PING, data);
|
| std::list<scoped_refptr<net::IOBufferWithSize>> orig_list;
|
| scoped_refptr<net::IOBufferWithSize> buf = orig_message->PopNextPacket();
|
| orig_list.push_back(buf);
|
|
|
| - scoped_refptr<U2fMessage> new_message =
|
| + std::unique_ptr<U2fMessage> new_message =
|
| U2fMessage::CreateFromSerializedData(buf);
|
| while (!new_message->MessageComplete()) {
|
| buf = orig_message->PopNextPacket();
|
|
|