OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/win/scoped_variant.h" | 5 #include "base/win/scoped_variant.h" |
6 #include "testing/gtest/include/gtest/gtest.h" | 6 #include "testing/gtest/include/gtest/gtest.h" |
7 | 7 |
8 namespace base { | 8 namespace base { |
9 namespace win { | 9 namespace win { |
10 | 10 |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
65 | 65 |
66 protected: | 66 protected: |
67 int ref_; | 67 int ref_; |
68 }; | 68 }; |
69 | 69 |
70 } // namespace | 70 } // namespace |
71 | 71 |
72 TEST(ScopedVariantTest, ScopedVariant) { | 72 TEST(ScopedVariantTest, ScopedVariant) { |
73 ScopedVariant var; | 73 ScopedVariant var; |
74 EXPECT_TRUE(var.type() == VT_EMPTY); | 74 EXPECT_TRUE(var.type() == VT_EMPTY); |
75 // V_BSTR(&var) = NULL; <- NOTE: Assignment like that is not supported | 75 // V_BSTR(var.ptr()) = NULL; <- NOTE: Assignment like that is not supported. |
76 | 76 |
77 ScopedVariant var_bstr(L"VT_BSTR"); | 77 ScopedVariant var_bstr(L"VT_BSTR"); |
78 EXPECT_EQ(VT_BSTR, V_VT(&var_bstr)); | 78 EXPECT_EQ(VT_BSTR, V_VT(var_bstr.ptr())); |
79 EXPECT_TRUE(V_BSTR(&var_bstr) != NULL); // can't use EXPECT_NE for BSTR | 79 EXPECT_TRUE(V_BSTR(var_bstr.ptr()) != NULL); // can't use EXPECT_NE for BSTR |
80 var_bstr.Reset(); | 80 var_bstr.Reset(); |
81 EXPECT_NE(VT_BSTR, V_VT(&var_bstr)); | 81 EXPECT_NE(VT_BSTR, V_VT(var_bstr.ptr())); |
82 var_bstr.Set(kTestString2); | 82 var_bstr.Set(kTestString2); |
83 EXPECT_EQ(VT_BSTR, V_VT(&var_bstr)); | 83 EXPECT_EQ(VT_BSTR, V_VT(var_bstr.ptr())); |
84 | 84 |
85 VARIANT tmp = var_bstr.Release(); | 85 VARIANT tmp = var_bstr.Release(); |
86 EXPECT_EQ(VT_EMPTY, V_VT(&var_bstr)); | 86 EXPECT_EQ(VT_EMPTY, V_VT(var_bstr.ptr())); |
87 EXPECT_EQ(VT_BSTR, V_VT(&tmp)); | 87 EXPECT_EQ(VT_BSTR, V_VT(&tmp)); |
88 EXPECT_EQ(0, lstrcmp(V_BSTR(&tmp), kTestString2)); | 88 EXPECT_EQ(0, lstrcmp(V_BSTR(&tmp), kTestString2)); |
89 | 89 |
90 var.Reset(tmp); | 90 var.Reset(tmp); |
91 EXPECT_EQ(VT_BSTR, V_VT(&var)); | 91 EXPECT_EQ(VT_BSTR, V_VT(var.ptr())); |
92 EXPECT_EQ(0, lstrcmpW(V_BSTR(&var), kTestString2)); | 92 EXPECT_EQ(0, lstrcmpW(V_BSTR(var.ptr()), kTestString2)); |
93 | 93 |
94 var_bstr.Swap(var); | 94 var_bstr.Swap(var); |
95 EXPECT_EQ(VT_EMPTY, V_VT(&var)); | 95 EXPECT_EQ(VT_EMPTY, V_VT(var.ptr())); |
96 EXPECT_EQ(VT_BSTR, V_VT(&var_bstr)); | 96 EXPECT_EQ(VT_BSTR, V_VT(var_bstr.ptr())); |
97 EXPECT_EQ(0, lstrcmpW(V_BSTR(&var_bstr), kTestString2)); | 97 EXPECT_EQ(0, lstrcmpW(V_BSTR(var_bstr.ptr()), kTestString2)); |
98 var_bstr.Reset(); | 98 var_bstr.Reset(); |
99 | 99 |
100 // Test the Compare and Copy routines. | 100 // Test the Compare and Copy routines. |
101 GiveMeAVariant(var_bstr.Receive()); | 101 GiveMeAVariant(var_bstr.Receive()); |
102 ScopedVariant var_bstr2(V_BSTR(&var_bstr)); | 102 ScopedVariant var_bstr2(V_BSTR(var_bstr.ptr())); |
103 EXPECT_EQ(0, var_bstr.Compare(var_bstr2)); | 103 EXPECT_EQ(0, var_bstr.Compare(var_bstr2)); |
104 var_bstr2.Reset(); | 104 var_bstr2.Reset(); |
105 EXPECT_NE(0, var_bstr.Compare(var_bstr2)); | 105 EXPECT_NE(0, var_bstr.Compare(var_bstr2)); |
106 var_bstr2.Reset(var_bstr.Copy()); | 106 var_bstr2.Reset(var_bstr.Copy()); |
107 EXPECT_EQ(0, var_bstr.Compare(var_bstr2)); | 107 EXPECT_EQ(0, var_bstr.Compare(var_bstr2)); |
108 var_bstr2.Reset(); | 108 var_bstr2.Reset(); |
109 var_bstr2.Set(V_BSTR(&var_bstr)); | 109 var_bstr2.Set(V_BSTR(var_bstr.ptr())); |
110 EXPECT_EQ(0, var_bstr.Compare(var_bstr2)); | 110 EXPECT_EQ(0, var_bstr.Compare(var_bstr2)); |
111 var_bstr2.Reset(); | 111 var_bstr2.Reset(); |
112 var_bstr.Reset(); | 112 var_bstr.Reset(); |
113 | 113 |
114 // Test for the SetDate setter. | 114 // Test for the SetDate setter. |
115 SYSTEMTIME sys_time; | 115 SYSTEMTIME sys_time; |
116 ::GetSystemTime(&sys_time); | 116 ::GetSystemTime(&sys_time); |
117 DATE date; | 117 DATE date; |
118 ::SystemTimeToVariantTime(&sys_time, &date); | 118 ::SystemTimeToVariantTime(&sys_time, &date); |
119 var.Reset(); | 119 var.Reset(); |
120 var.SetDate(date); | 120 var.SetDate(date); |
121 EXPECT_EQ(VT_DATE, var.type()); | 121 EXPECT_EQ(VT_DATE, var.type()); |
122 EXPECT_EQ(date, V_DATE(&var)); | 122 EXPECT_EQ(date, V_DATE(var.ptr())); |
123 | 123 |
124 // Simple setter tests. These do not require resetting the variant | 124 // Simple setter tests. These do not require resetting the variant |
125 // after each test since the variant type is not "leakable" (i.e. doesn't | 125 // after each test since the variant type is not "leakable" (i.e. doesn't |
126 // need to be freed explicitly). | 126 // need to be freed explicitly). |
127 | 127 |
128 // We need static cast here since char defaults to int (!?). | 128 // We need static cast here since char defaults to int (!?). |
129 var.Set(static_cast<int8>('v')); | 129 var.Set(static_cast<int8>('v')); |
130 EXPECT_EQ(VT_I1, var.type()); | 130 EXPECT_EQ(VT_I1, var.type()); |
131 EXPECT_EQ('v', V_I1(&var)); | 131 EXPECT_EQ('v', V_I1(var.ptr())); |
132 | 132 |
133 var.Set(static_cast<short>(123)); | 133 var.Set(static_cast<short>(123)); |
134 EXPECT_EQ(VT_I2, var.type()); | 134 EXPECT_EQ(VT_I2, var.type()); |
135 EXPECT_EQ(123, V_I2(&var)); | 135 EXPECT_EQ(123, V_I2(var.ptr())); |
136 | 136 |
137 var.Set(static_cast<int32>(123)); | 137 var.Set(static_cast<int32>(123)); |
138 EXPECT_EQ(VT_I4, var.type()); | 138 EXPECT_EQ(VT_I4, var.type()); |
139 EXPECT_EQ(123, V_I4(&var)); | 139 EXPECT_EQ(123, V_I4(var.ptr())); |
140 | 140 |
141 var.Set(static_cast<int64>(123)); | 141 var.Set(static_cast<int64>(123)); |
142 EXPECT_EQ(VT_I8, var.type()); | 142 EXPECT_EQ(VT_I8, var.type()); |
143 EXPECT_EQ(123, V_I8(&var)); | 143 EXPECT_EQ(123, V_I8(var.ptr())); |
144 | 144 |
145 var.Set(static_cast<uint8>(123)); | 145 var.Set(static_cast<uint8>(123)); |
146 EXPECT_EQ(VT_UI1, var.type()); | 146 EXPECT_EQ(VT_UI1, var.type()); |
147 EXPECT_EQ(123, V_UI1(&var)); | 147 EXPECT_EQ(123, V_UI1(var.ptr())); |
148 | 148 |
149 var.Set(static_cast<unsigned short>(123)); | 149 var.Set(static_cast<unsigned short>(123)); |
150 EXPECT_EQ(VT_UI2, var.type()); | 150 EXPECT_EQ(VT_UI2, var.type()); |
151 EXPECT_EQ(123, V_UI2(&var)); | 151 EXPECT_EQ(123, V_UI2(var.ptr())); |
152 | 152 |
153 var.Set(static_cast<uint32>(123)); | 153 var.Set(static_cast<uint32>(123)); |
154 EXPECT_EQ(VT_UI4, var.type()); | 154 EXPECT_EQ(VT_UI4, var.type()); |
155 EXPECT_EQ(123, V_UI4(&var)); | 155 EXPECT_EQ(123, V_UI4(var.ptr())); |
156 | 156 |
157 var.Set(static_cast<uint64>(123)); | 157 var.Set(static_cast<uint64>(123)); |
158 EXPECT_EQ(VT_UI8, var.type()); | 158 EXPECT_EQ(VT_UI8, var.type()); |
159 EXPECT_EQ(123, V_UI8(&var)); | 159 EXPECT_EQ(123, V_UI8(var.ptr())); |
160 | 160 |
161 var.Set(123.123f); | 161 var.Set(123.123f); |
162 EXPECT_EQ(VT_R4, var.type()); | 162 EXPECT_EQ(VT_R4, var.type()); |
163 EXPECT_EQ(123.123f, V_R4(&var)); | 163 EXPECT_EQ(123.123f, V_R4(var.ptr())); |
164 | 164 |
165 var.Set(static_cast<double>(123.123)); | 165 var.Set(static_cast<double>(123.123)); |
166 EXPECT_EQ(VT_R8, var.type()); | 166 EXPECT_EQ(VT_R8, var.type()); |
167 EXPECT_EQ(123.123, V_R8(&var)); | 167 EXPECT_EQ(123.123, V_R8(var.ptr())); |
168 | 168 |
169 var.Set(true); | 169 var.Set(true); |
170 EXPECT_EQ(VT_BOOL, var.type()); | 170 EXPECT_EQ(VT_BOOL, var.type()); |
171 EXPECT_EQ(VARIANT_TRUE, V_BOOL(&var)); | 171 EXPECT_EQ(VARIANT_TRUE, V_BOOL(var.ptr())); |
172 var.Set(false); | 172 var.Set(false); |
173 EXPECT_EQ(VT_BOOL, var.type()); | 173 EXPECT_EQ(VT_BOOL, var.type()); |
174 EXPECT_EQ(VARIANT_FALSE, V_BOOL(&var)); | 174 EXPECT_EQ(VARIANT_FALSE, V_BOOL(var.ptr())); |
175 | 175 |
176 // Com interface tests | 176 // Com interface tests |
177 | 177 |
178 var.Set(static_cast<IDispatch*>(NULL)); | 178 var.Set(static_cast<IDispatch*>(NULL)); |
179 EXPECT_EQ(VT_DISPATCH, var.type()); | 179 EXPECT_EQ(VT_DISPATCH, var.type()); |
180 EXPECT_EQ(NULL, V_DISPATCH(&var)); | 180 EXPECT_EQ(NULL, V_DISPATCH(var.ptr())); |
181 var.Reset(); | 181 var.Reset(); |
182 | 182 |
183 var.Set(static_cast<IUnknown*>(NULL)); | 183 var.Set(static_cast<IUnknown*>(NULL)); |
184 EXPECT_EQ(VT_UNKNOWN, var.type()); | 184 EXPECT_EQ(VT_UNKNOWN, var.type()); |
185 EXPECT_EQ(NULL, V_UNKNOWN(&var)); | 185 EXPECT_EQ(NULL, V_UNKNOWN(var.ptr())); |
186 var.Reset(); | 186 var.Reset(); |
187 | 187 |
188 FakeComObject faker; | 188 FakeComObject faker; |
189 EXPECT_EQ(0, faker.ref_count()); | 189 EXPECT_EQ(0, faker.ref_count()); |
190 var.Set(static_cast<IDispatch*>(&faker)); | 190 var.Set(static_cast<IDispatch*>(&faker)); |
191 EXPECT_EQ(VT_DISPATCH, var.type()); | 191 EXPECT_EQ(VT_DISPATCH, var.type()); |
192 EXPECT_EQ(&faker, V_DISPATCH(&var)); | 192 EXPECT_EQ(&faker, V_DISPATCH(var.ptr())); |
193 EXPECT_EQ(1, faker.ref_count()); | 193 EXPECT_EQ(1, faker.ref_count()); |
194 var.Reset(); | 194 var.Reset(); |
195 EXPECT_EQ(0, faker.ref_count()); | 195 EXPECT_EQ(0, faker.ref_count()); |
196 | 196 |
197 var.Set(static_cast<IUnknown*>(&faker)); | 197 var.Set(static_cast<IUnknown*>(&faker)); |
198 EXPECT_EQ(VT_UNKNOWN, var.type()); | 198 EXPECT_EQ(VT_UNKNOWN, var.type()); |
199 EXPECT_EQ(&faker, V_UNKNOWN(&var)); | 199 EXPECT_EQ(&faker, V_UNKNOWN(var.ptr())); |
200 EXPECT_EQ(1, faker.ref_count()); | 200 EXPECT_EQ(1, faker.ref_count()); |
201 var.Reset(); | 201 var.Reset(); |
202 EXPECT_EQ(0, faker.ref_count()); | 202 EXPECT_EQ(0, faker.ref_count()); |
203 | 203 |
204 { | 204 { |
205 ScopedVariant disp_var(&faker); | 205 ScopedVariant disp_var(&faker); |
206 EXPECT_EQ(VT_DISPATCH, disp_var.type()); | 206 EXPECT_EQ(VT_DISPATCH, disp_var.type()); |
207 EXPECT_EQ(&faker, V_DISPATCH(&disp_var)); | 207 EXPECT_EQ(&faker, V_DISPATCH(disp_var.ptr())); |
208 EXPECT_EQ(1, faker.ref_count()); | 208 EXPECT_EQ(1, faker.ref_count()); |
209 } | 209 } |
210 EXPECT_EQ(0, faker.ref_count()); | 210 EXPECT_EQ(0, faker.ref_count()); |
211 | 211 |
212 { | 212 { |
213 ScopedVariant ref1(&faker); | 213 ScopedVariant ref1(&faker); |
214 EXPECT_EQ(1, faker.ref_count()); | 214 EXPECT_EQ(1, faker.ref_count()); |
215 ScopedVariant ref2(static_cast<const VARIANT&>(ref1)); | 215 ScopedVariant ref2(static_cast<const VARIANT&>(ref1)); |
216 EXPECT_EQ(2, faker.ref_count()); | 216 EXPECT_EQ(2, faker.ref_count()); |
217 ScopedVariant ref3; | 217 ScopedVariant ref3; |
218 ref3 = static_cast<const VARIANT&>(ref2); | 218 ref3 = static_cast<const VARIANT&>(ref2); |
219 EXPECT_EQ(3, faker.ref_count()); | 219 EXPECT_EQ(3, faker.ref_count()); |
220 } | 220 } |
221 EXPECT_EQ(0, faker.ref_count()); | 221 EXPECT_EQ(0, faker.ref_count()); |
222 | 222 |
223 { | 223 { |
224 ScopedVariant unk_var(static_cast<IUnknown*>(&faker)); | 224 ScopedVariant unk_var(static_cast<IUnknown*>(&faker)); |
225 EXPECT_EQ(VT_UNKNOWN, unk_var.type()); | 225 EXPECT_EQ(VT_UNKNOWN, unk_var.type()); |
226 EXPECT_EQ(&faker, V_UNKNOWN(&unk_var)); | 226 EXPECT_EQ(&faker, V_UNKNOWN(unk_var.ptr())); |
227 EXPECT_EQ(1, faker.ref_count()); | 227 EXPECT_EQ(1, faker.ref_count()); |
228 } | 228 } |
229 EXPECT_EQ(0, faker.ref_count()); | 229 EXPECT_EQ(0, faker.ref_count()); |
230 | 230 |
231 VARIANT raw; | 231 VARIANT raw; |
232 raw.vt = VT_UNKNOWN; | 232 raw.vt = VT_UNKNOWN; |
233 raw.punkVal = &faker; | 233 raw.punkVal = &faker; |
234 EXPECT_EQ(0, faker.ref_count()); | 234 EXPECT_EQ(0, faker.ref_count()); |
235 var.Set(raw); | 235 var.Set(raw); |
236 EXPECT_EQ(1, faker.ref_count()); | 236 EXPECT_EQ(1, faker.ref_count()); |
237 var.Reset(); | 237 var.Reset(); |
238 EXPECT_EQ(0, faker.ref_count()); | 238 EXPECT_EQ(0, faker.ref_count()); |
239 | 239 |
240 { | 240 { |
241 ScopedVariant number(123); | 241 ScopedVariant number(123); |
242 EXPECT_EQ(VT_I4, number.type()); | 242 EXPECT_EQ(VT_I4, number.type()); |
243 EXPECT_EQ(123, V_I4(&number)); | 243 EXPECT_EQ(123, V_I4(number.ptr())); |
244 } | 244 } |
245 | 245 |
246 // SAFEARRAY tests | 246 // SAFEARRAY tests |
247 var.Set(static_cast<SAFEARRAY*>(NULL)); | 247 var.Set(static_cast<SAFEARRAY*>(NULL)); |
248 EXPECT_EQ(VT_EMPTY, var.type()); | 248 EXPECT_EQ(VT_EMPTY, var.type()); |
249 | 249 |
250 SAFEARRAY* sa = ::SafeArrayCreateVector(VT_UI1, 0, 100); | 250 SAFEARRAY* sa = ::SafeArrayCreateVector(VT_UI1, 0, 100); |
251 ASSERT_TRUE(sa != NULL); | 251 ASSERT_TRUE(sa != NULL); |
252 | 252 |
253 var.Set(sa); | 253 var.Set(sa); |
254 EXPECT_TRUE(ScopedVariant::IsLeakableVarType(var.type())); | 254 EXPECT_TRUE(ScopedVariant::IsLeakableVarType(var.type())); |
255 EXPECT_EQ(VT_ARRAY | VT_UI1, var.type()); | 255 EXPECT_EQ(VT_ARRAY | VT_UI1, var.type()); |
256 EXPECT_EQ(sa, V_ARRAY(&var)); | 256 EXPECT_EQ(sa, V_ARRAY(var.ptr())); |
257 // The array is destroyed in the destructor of var. | 257 // The array is destroyed in the destructor of var. |
258 } | 258 } |
259 | 259 |
260 } // namespace win | 260 } // namespace win |
261 } // namespace base | 261 } // namespace base |
OLD | NEW |