15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2010 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/win/scoped_bstr.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace base { 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace win { 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const wchar_t kTestString1[] = L"123"; 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const wchar_t kTestString2[] = L"456789"; 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)size_t test1_len = arraysize(kTestString1) - 1; 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)size_t test2_len = arraysize(kTestString2) - 1; 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void DumbBstrTests() { 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedBstr b; 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b == NULL); 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, b.Length()); 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, b.ByteLength()); 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) b.Reset(NULL); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b == NULL); 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b.Release() == NULL); 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedBstr b2; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) b.Swap(b2); 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b2 == NULL); 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void GiveMeABstr(BSTR* ret) { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *ret = SysAllocString(kTestString1); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void BasicBstrTests() { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedBstr b1(kTestString1); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(test1_len, b1.Length()); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(test1_len * sizeof(kTestString1[0]), b1.ByteLength()); 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedBstr b2; 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) b1.Swap(b2); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(test1_len, b2.Length()); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, b1.Length()); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, lstrcmp(b2, kTestString1)); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BSTR tmp = b2.Release(); 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tmp != NULL); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, lstrcmp(tmp, kTestString1)); 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b2 == NULL); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SysFreeString(tmp); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GiveMeABstr(b2.Receive()); 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b2 != NULL); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) b2.Reset(); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b2.AllocateBytes(100) != NULL); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(100, b2.ByteLength()); 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(100 / sizeof(kTestString1[0]), b2.Length()); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) lstrcpy(static_cast<BSTR>(b2), kTestString1); 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(test1_len, lstrlen(b2)); 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(100 / sizeof(kTestString1[0]), b2.Length()); 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) b2.SetByteLen(lstrlen(b2) * sizeof(kTestString2[0])); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(b2.Length(), lstrlen(b2)); 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(b1.Allocate(kTestString2) != NULL); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(test2_len, b1.Length()); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) b1.SetByteLen((test2_len - 1) * sizeof(kTestString2[0])); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(test2_len - 1, b1.Length()); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(ScopedBstrTest, ScopedBstr) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DumbBstrTests(); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BasicBstrTests(); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace win 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace base 78