15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/ref_counted_memory.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace base { 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(RefCountedMemoryUnitTest, RefCountedStaticMemory) { 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<RefCountedMemory> mem = new RefCountedStaticMemory( 135d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) "static mem00", 10); 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(10U, mem->size()); 165d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_EQ("static mem", std::string(mem->front_as<char>(), mem->size())); 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(RefCountedMemoryUnitTest, RefCountedBytes) { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<uint8> data; 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data.push_back(45); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data.push_back(99); 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<RefCountedMemory> mem = RefCountedBytes::TakeVector(&data); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0U, data.size()); 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2U, mem->size()); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(45U, mem->front()[0]); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(99U, mem->front()[1]); 305c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu 315c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu scoped_refptr<RefCountedMemory> mem2; 325c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu { 335c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu unsigned char data2[] = { 12, 11, 99 }; 345c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu mem2 = new RefCountedBytes(data2, 3); 355c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu } 365c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu EXPECT_EQ(3U, mem2->size()); 375c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu EXPECT_EQ(12U, mem2->front()[0]); 385c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu EXPECT_EQ(11U, mem2->front()[1]); 395c02ac1a9c1b504631c0a3d2b6e737b5d738bae1Bo Liu EXPECT_EQ(99U, mem2->front()[2]); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(RefCountedMemoryUnitTest, RefCountedString) { 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string s("destroy me"); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<RefCountedMemory> mem = RefCountedString::TakeString(&s); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0U, s.size()); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(10U, mem->size()); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ('d', mem->front()[0]); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ('e', mem->front()[1]); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 53f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)TEST(RefCountedMemoryUnitTest, RefCountedMallocedMemory) { 54f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) void* data = malloc(6); 55f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) memcpy(data, "hello", 6); 56f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 57f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) scoped_refptr<RefCountedMemory> mem = new RefCountedMallocedMemory(data, 6); 58f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 59f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_EQ(6U, mem->size()); 60f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_EQ(0, memcmp("hello", mem->front(), 6)); 61f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)} 62f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST(RefCountedMemoryUnitTest, Equals) { 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string s1("same"); 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<RefCountedMemory> mem1 = RefCountedString::TakeString(&s1); 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<unsigned char> d2; 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) d2.push_back('s'); 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) d2.push_back('a'); 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) d2.push_back('m'); 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) d2.push_back('e'); 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<RefCountedMemory> mem2 = RefCountedBytes::TakeVector(&d2); 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(mem1->Equals(mem2)); 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string s3("diff"); 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<RefCountedMemory> mem3 = RefCountedString::TakeString(&s3); 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(mem1->Equals(mem3)); 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(mem2->Equals(mem3)); 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)TEST(RefCountedMemoryUnitTest, EqualsNull) { 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string s("str"); 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<RefCountedMemory> mem = RefCountedString::TakeString(&s); 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(mem->Equals(NULL)); 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace base 90