90 lines
2.4 KiB
C++
90 lines
2.4 KiB
C++
// Copyright (c) 2011 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 "butil/memory/ref_counted_memory.h"
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
namespace butil {
|
|
|
|
TEST(RefCountedMemoryUnitTest, RefCountedStaticMemory) {
|
|
scoped_refptr<RefCountedMemory> mem = new RefCountedStaticMemory(
|
|
"static mem00", 10);
|
|
|
|
EXPECT_EQ(10U, mem->size());
|
|
EXPECT_EQ("static mem", std::string(mem->front_as<char>(), mem->size()));
|
|
}
|
|
|
|
TEST(RefCountedMemoryUnitTest, RefCountedBytes) {
|
|
std::vector<uint8_t> data;
|
|
data.push_back(45);
|
|
data.push_back(99);
|
|
scoped_refptr<RefCountedMemory> mem = RefCountedBytes::TakeVector(&data);
|
|
|
|
EXPECT_EQ(0U, data.size());
|
|
|
|
EXPECT_EQ(2U, mem->size());
|
|
EXPECT_EQ(45U, mem->front()[0]);
|
|
EXPECT_EQ(99U, mem->front()[1]);
|
|
|
|
scoped_refptr<RefCountedMemory> mem2;
|
|
{
|
|
unsigned char data2[] = { 12, 11, 99 };
|
|
mem2 = new RefCountedBytes(data2, 3);
|
|
}
|
|
EXPECT_EQ(3U, mem2->size());
|
|
EXPECT_EQ(12U, mem2->front()[0]);
|
|
EXPECT_EQ(11U, mem2->front()[1]);
|
|
EXPECT_EQ(99U, mem2->front()[2]);
|
|
}
|
|
|
|
TEST(RefCountedMemoryUnitTest, RefCountedString) {
|
|
std::string s("destroy me");
|
|
scoped_refptr<RefCountedMemory> mem = RefCountedString::TakeString(&s);
|
|
|
|
EXPECT_EQ(0U, s.size());
|
|
|
|
EXPECT_EQ(10U, mem->size());
|
|
EXPECT_EQ('d', mem->front()[0]);
|
|
EXPECT_EQ('e', mem->front()[1]);
|
|
}
|
|
|
|
TEST(RefCountedMemoryUnitTest, RefCountedMallocedMemory) {
|
|
void* data = malloc(6);
|
|
memcpy(data, "hello", 6);
|
|
|
|
scoped_refptr<RefCountedMemory> mem = new RefCountedMallocedMemory(data, 6);
|
|
|
|
EXPECT_EQ(6U, mem->size());
|
|
EXPECT_EQ(0, memcmp("hello", mem->front(), 6));
|
|
}
|
|
|
|
TEST(RefCountedMemoryUnitTest, Equals) {
|
|
std::string s1("same");
|
|
scoped_refptr<RefCountedMemory> mem1 = RefCountedString::TakeString(&s1);
|
|
|
|
std::vector<unsigned char> d2;
|
|
d2.push_back('s');
|
|
d2.push_back('a');
|
|
d2.push_back('m');
|
|
d2.push_back('e');
|
|
scoped_refptr<RefCountedMemory> mem2 = RefCountedBytes::TakeVector(&d2);
|
|
|
|
EXPECT_TRUE(mem1->Equals(mem2));
|
|
|
|
std::string s3("diff");
|
|
scoped_refptr<RefCountedMemory> mem3 = RefCountedString::TakeString(&s3);
|
|
|
|
EXPECT_FALSE(mem1->Equals(mem3));
|
|
EXPECT_FALSE(mem2->Equals(mem3));
|
|
}
|
|
|
|
TEST(RefCountedMemoryUnitTest, EqualsNull) {
|
|
std::string s("str");
|
|
scoped_refptr<RefCountedMemory> mem = RefCountedString::TakeString(&s);
|
|
EXPECT_FALSE(mem->Equals(NULL));
|
|
}
|
|
|
|
} // namespace butil
|