Index: base/win/enum_variant_unittest.cc |
=================================================================== |
--- base/win/enum_variant_unittest.cc (revision 0) |
+++ base/win/enum_variant_unittest.cc (revision 0) |
@@ -0,0 +1,118 @@ |
+// 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 "base/win/enum_variant.h" |
+ |
+#include "base/win/scoped_com_initializer.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace base { |
+namespace win { |
+ |
+TEST(EnumVariantTest, EmptyEnumVariant) { |
+ ScopedCOMInitializer com_initializer; |
+ |
+ EnumVariant* ev = new EnumVariant(0); |
+ ev->AddRef(); |
+ |
+ IUnknown* iunknown; |
+ EXPECT_TRUE(SUCCEEDED( |
+ ev->QueryInterface(IID_IUnknown, reinterpret_cast<void**>(&iunknown)))); |
+ iunknown->Release(); |
+ |
+ IEnumVARIANT* ienumvariant; |
+ EXPECT_TRUE(SUCCEEDED( |
+ ev->QueryInterface(IID_IEnumVARIANT, |
+ reinterpret_cast<void**>(&ienumvariant)))); |
+ EXPECT_EQ(ev, ienumvariant); |
+ ienumvariant->Release(); |
+ |
+ VARIANT out_element; |
+ ULONG out_received = 0; |
+ EXPECT_EQ(S_FALSE, ev->Next(1, &out_element, &out_received)); |
+ EXPECT_EQ(0, out_received); |
+ |
+ EXPECT_EQ(S_FALSE, ev->Skip(1)); |
+ |
+ EXPECT_EQ(S_OK, ev->Reset()); |
+ |
+ IEnumVARIANT* ev2 = NULL; |
+ EXPECT_EQ(S_OK, ev->Clone(&ev2)); |
+ |
+ EXPECT_NE(static_cast<IEnumVARIANT*>(NULL), ev2); |
+ EXPECT_NE(ev, ev2); |
+ EXPECT_EQ(S_FALSE, ev2->Skip(1)); |
+ EXPECT_EQ(S_OK, ev2->Reset()); |
+ |
+ ULONG ev2_finalrefcount = ev2->Release(); |
+ EXPECT_EQ(0, ev2_finalrefcount); |
+ |
+ ULONG ev_finalrefcount = ev->Release(); |
+ EXPECT_EQ(0, ev_finalrefcount); |
+} |
+ |
+TEST(EnumVariantTest, SimpleEnumVariant) { |
+ ScopedCOMInitializer com_initializer; |
+ |
+ EnumVariant* ev = new EnumVariant(3); |
+ ev->AddRef(); |
+ ev->ItemAt(0)->vt = VT_I4; |
+ ev->ItemAt(0)->lVal = 10; |
+ ev->ItemAt(1)->vt = VT_I4; |
+ ev->ItemAt(1)->lVal = 20; |
+ ev->ItemAt(2)->vt = VT_I4; |
+ ev->ItemAt(2)->lVal = 30; |
+ |
+ // Get elements one at a time. |
+ VARIANT out_element; |
+ ULONG out_received = 0; |
+ EXPECT_EQ(S_OK, ev->Next(1, &out_element, &out_received)); |
+ EXPECT_EQ(1, out_received); |
+ EXPECT_EQ(VT_I4, out_element.vt); |
+ EXPECT_EQ(10, out_element.lVal); |
+ EXPECT_EQ(S_OK, ev->Skip(1)); |
+ EXPECT_EQ(S_OK, ev->Next(1, &out_element, &out_received)); |
+ EXPECT_EQ(1, out_received); |
+ EXPECT_EQ(VT_I4, out_element.vt); |
+ EXPECT_EQ(30, out_element.lVal); |
+ EXPECT_EQ(S_FALSE, ev->Next(1, &out_element, &out_received)); |
+ |
+ // Reset and get all elements at once. |
+ VARIANT out_elements[3]; |
+ EXPECT_EQ(S_OK, ev->Reset()); |
+ EXPECT_EQ(S_OK, ev->Next(3, out_elements, &out_received)); |
+ EXPECT_EQ(3, out_received); |
+ EXPECT_EQ(VT_I4, out_elements[0].vt); |
+ EXPECT_EQ(10, out_elements[0].lVal); |
+ EXPECT_EQ(VT_I4, out_elements[1].vt); |
+ EXPECT_EQ(20, out_elements[1].lVal); |
+ EXPECT_EQ(VT_I4, out_elements[2].vt); |
+ EXPECT_EQ(30, out_elements[2].lVal); |
+ EXPECT_EQ(S_FALSE, ev->Next(1, &out_element, &out_received)); |
+ |
+ // Clone it. |
+ IEnumVARIANT* ev2 = NULL; |
+ EXPECT_EQ(S_OK, ev->Clone(&ev2)); |
+ EXPECT_TRUE(ev2 != NULL); |
+ EXPECT_EQ(S_FALSE, ev->Next(1, &out_element, &out_received)); |
+ EXPECT_EQ(S_OK, ev2->Reset()); |
+ EXPECT_EQ(S_OK, ev2->Next(3, out_elements, &out_received)); |
+ EXPECT_EQ(3, out_received); |
+ EXPECT_EQ(VT_I4, out_elements[0].vt); |
+ EXPECT_EQ(10, out_elements[0].lVal); |
+ EXPECT_EQ(VT_I4, out_elements[1].vt); |
+ EXPECT_EQ(20, out_elements[1].lVal); |
+ EXPECT_EQ(VT_I4, out_elements[2].vt); |
+ EXPECT_EQ(30, out_elements[2].lVal); |
+ EXPECT_EQ(S_FALSE, ev2->Next(1, &out_element, &out_received)); |
+ |
+ ULONG ev2_finalrefcount = ev2->Release(); |
+ EXPECT_EQ(0, ev2_finalrefcount); |
+ |
+ ULONG ev_finalrefcount = ev->Release(); |
+ EXPECT_EQ(0, ev_finalrefcount); |
+} |
+ |
+} // namespace win |
+} // namespace base |