1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
// Copyright (c) 2006-2008 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 "base/scoped_bstr_win.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
static const wchar_t kTestString1[] = L"123";
static const wchar_t kTestString2[] = L"456789";
uint32 test1_len = arraysize(kTestString1) - 1;
uint32 test2_len = arraysize(kTestString2) - 1;
void DumbBstrTests() {
ScopedBstr b;
EXPECT_TRUE(b == NULL);
EXPECT_TRUE(b.Length() == 0);
EXPECT_TRUE(b.ByteLength() == 0);
b.Reset(NULL);
EXPECT_TRUE(b == NULL);
EXPECT_TRUE(b.Release() == NULL);
ScopedBstr b2;
b.Swap(b2);
EXPECT_TRUE(b2 == NULL);
}
void GiveMeABstr(BSTR* ret) {
*ret = SysAllocString(kTestString1);
}
void BasicBstrTests() {
ScopedBstr b1(kTestString1);
EXPECT_TRUE(b1.Length() == test1_len);
EXPECT_TRUE(b1.ByteLength() == test1_len * sizeof(kTestString1[0]));
ScopedBstr b2;
b1.Swap(b2);
EXPECT_TRUE(b2.Length() == test1_len);
EXPECT_TRUE(b1.Length() == 0);
EXPECT_TRUE(lstrcmpW(b2, kTestString1) == 0);
BSTR tmp = b2.Release();
EXPECT_TRUE(tmp != NULL);
EXPECT_TRUE(lstrcmpW(tmp, kTestString1) == 0);
EXPECT_TRUE(b2 == NULL);
SysFreeString(tmp);
GiveMeABstr(b2.Receive());
EXPECT_TRUE(b2 != NULL);
b2.Reset();
EXPECT_TRUE(b2.AllocateBytes(100) != NULL);
EXPECT_TRUE(b2.ByteLength() == 100);
EXPECT_TRUE(b2.Length() == 100 / sizeof(kTestString1[0]));
lstrcpyW(static_cast<BSTR>(b2), kTestString1);
EXPECT_TRUE(lstrlen(b2) == test1_len);
EXPECT_TRUE(b2.Length() == 100 / sizeof(kTestString1[0]));
b2.SetByteLen(lstrlen(b2) * sizeof(kTestString2[0]));
EXPECT_TRUE(lstrlen(b2) == b2.Length());
EXPECT_TRUE(b1.Allocate(kTestString2) != NULL);
EXPECT_TRUE(b1.Length() == test2_len);
b1.SetByteLen((test2_len - 1) * sizeof(kTestString2[0]));
EXPECT_TRUE(b1.Length() == test2_len - 1);
}
} // namespace
TEST(ScopedBstrTest, ScopedBstr) {
DumbBstrTests();
BasicBstrTests();
}
#define kSourceStr L"this is a string"
#define kSourceStrEmpty L""
TEST(StackBstrTest, StackBstr) {
ScopedBstr system_bstr(kSourceStr);
StackBstrVar(kSourceStr, stack_bstr);
EXPECT_EQ(VarBstrCmp(system_bstr, stack_bstr, LOCALE_USER_DEFAULT, 0),
VARCMP_EQ);
StackBstrVar(kSourceStrEmpty, empty);
uint32 l1 = SysStringLen(stack_bstr);
uint32 l2 = SysStringLen(StackBstr(kSourceStr));
uint32 l3 = SysStringLen(system_bstr);
EXPECT_TRUE(l1 == l2);
EXPECT_TRUE(l2 == l3);
EXPECT_TRUE(SysStringLen(empty) == 0);
const wchar_t one_more_test[] = L"this is my const string";
EXPECT_EQ(SysStringLen(StackBstr(one_more_test)),
lstrlenW(one_more_test));
}
|
