1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/memory/ref_counted_memory.h"
6
7 #include <stdint.h>
8
9 #include <utility>
10
11 #include "base/containers/span.h"
12 #include "base/memory/read_only_shared_memory_region.h"
13 #include "base/stl_util.h"
14 #include "testing/gmock/include/gmock/gmock.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
17 using testing::Each;
18 using testing::ElementsAre;
19
20 namespace base {
21
22 namespace {
23
ConvertToByteSpanAndCheckSize(span<const uint8_t> data,size_t expected_size)24 void ConvertToByteSpanAndCheckSize(span<const uint8_t> data,
25 size_t expected_size) {
26 EXPECT_EQ(expected_size, data.size());
27 }
28
29 } // namespace
30
TEST(RefCountedMemoryUnitTest,RefCountedStaticMemory)31 TEST(RefCountedMemoryUnitTest, RefCountedStaticMemory) {
32 auto mem = MakeRefCounted<RefCountedStaticMemory>("static mem00", 10);
33
34 EXPECT_EQ(10U, mem->size());
35 EXPECT_EQ("static mem", std::string(mem->front_as<char>(), mem->size()));
36
37 ConvertToByteSpanAndCheckSize(*mem, 10);
38 }
39
TEST(RefCountedMemoryUnitTest,RefCountedBytes)40 TEST(RefCountedMemoryUnitTest, RefCountedBytes) {
41 std::vector<uint8_t> data;
42 data.push_back(45);
43 data.push_back(99);
44 scoped_refptr<RefCountedMemory> mem = RefCountedBytes::TakeVector(&data);
45
46 EXPECT_EQ(0U, data.size());
47
48 ASSERT_EQ(2U, mem->size());
49 EXPECT_EQ(45U, mem->front()[0]);
50 EXPECT_EQ(99U, mem->front()[1]);
51
52 scoped_refptr<RefCountedMemory> mem2;
53 {
54 const unsigned char kData[] = {12, 11, 99};
55 mem2 = MakeRefCounted<RefCountedBytes>(kData, size(kData));
56 }
57 ASSERT_EQ(3U, mem2->size());
58 EXPECT_EQ(12U, mem2->front()[0]);
59 EXPECT_EQ(11U, mem2->front()[1]);
60 EXPECT_EQ(99U, mem2->front()[2]);
61
62 ConvertToByteSpanAndCheckSize(*mem2, 3);
63 }
64
TEST(RefCountedMemoryUnitTest,RefCountedBytesMutable)65 TEST(RefCountedMemoryUnitTest, RefCountedBytesMutable) {
66 auto mem = MakeRefCounted<RefCountedBytes>(10);
67
68 ASSERT_EQ(10U, mem->size());
69 EXPECT_THAT(mem->data(), Each(0U));
70
71 // Test non-const versions of data(), front() and front_as<>().
72 mem->data()[0] = 1;
73 mem->front()[1] = 2;
74 mem->front_as<char>()[2] = 3;
75
76 EXPECT_THAT(mem->data(), ElementsAre(1, 2, 3, 0, 0, 0, 0, 0, 0, 0));
77 }
78
TEST(RefCountedMemoryUnitTest,RefCountedString)79 TEST(RefCountedMemoryUnitTest, RefCountedString) {
80 std::string s("destroy me");
81 scoped_refptr<RefCountedMemory> mem = RefCountedString::TakeString(&s);
82
83 EXPECT_EQ(0U, s.size());
84
85 ASSERT_EQ(10U, mem->size());
86 EXPECT_EQ('d', mem->front()[0]);
87 EXPECT_EQ('e', mem->front()[1]);
88 EXPECT_EQ('e', mem->front()[9]);
89
90 ConvertToByteSpanAndCheckSize(*mem, 10);
91 }
92
TEST(RefCountedMemoryUnitTest,Equals)93 TEST(RefCountedMemoryUnitTest, Equals) {
94 std::string s1("same");
95 scoped_refptr<RefCountedMemory> mem1 = RefCountedString::TakeString(&s1);
96
97 std::vector<unsigned char> d2 = {'s', 'a', 'm', 'e'};
98 scoped_refptr<RefCountedMemory> mem2 = RefCountedBytes::TakeVector(&d2);
99
100 EXPECT_TRUE(mem1->Equals(mem2));
101
102 std::string s3("diff");
103 scoped_refptr<RefCountedMemory> mem3 = RefCountedString::TakeString(&s3);
104
105 EXPECT_FALSE(mem1->Equals(mem3));
106 EXPECT_FALSE(mem2->Equals(mem3));
107 }
108
TEST(RefCountedMemoryUnitTest,EqualsNull)109 TEST(RefCountedMemoryUnitTest, EqualsNull) {
110 std::string s("str");
111 scoped_refptr<RefCountedMemory> mem = RefCountedString::TakeString(&s);
112 EXPECT_FALSE(mem->Equals(nullptr));
113 }
114
115 } // namespace base
116