Adding a Set() method for copying a variant over to the ScopedVariant....

base/scoped_variant_win_unittest.cc

 // Copyright (c) 2009 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_variant_win.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 static const wchar_t kTestString1[] = L"Used to create BSTRs";
@@ skipping 180 matching lines @@
   var.Reset();
   EXPECT_EQ(0, faker.ref_count());
 
   var.Set(static_cast<IUnknown*>(&faker));
   EXPECT_EQ(VT_UNKNOWN, var.type());
   EXPECT_EQ(&faker, V_UNKNOWN(&var));
   EXPECT_EQ(1, faker.ref_count());
   var.Reset();
   EXPECT_EQ(0, faker.ref_count());
 
+  {
+    ScopedVariant disp_var(&faker);
+    EXPECT_EQ(VT_DISPATCH, disp_var.type());
+    EXPECT_EQ(&faker, V_DISPATCH(&disp_var));
+    EXPECT_EQ(1, faker.ref_count());
+  }
+  EXPECT_EQ(0, faker.ref_count());
+
+  {
+    ScopedVariant ref1(&faker);
+    EXPECT_EQ(1, faker.ref_count());
+    ScopedVariant ref2(static_cast<const VARIANT&>(ref1));
+    EXPECT_EQ(2, faker.ref_count());
+    ScopedVariant ref3;
+    ref3 = static_cast<const VARIANT&>(ref2);
+    EXPECT_EQ(3, faker.ref_count());
+  }
+  EXPECT_EQ(0, faker.ref_count());
+
+  {
+    ScopedVariant unk_var(static_cast<IUnknown*>(&faker));
+    EXPECT_EQ(VT_UNKNOWN, unk_var.type());
+    EXPECT_EQ(&faker, V_UNKNOWN(&unk_var));
+    EXPECT_EQ(1, faker.ref_count());
+  }
+  EXPECT_EQ(0, faker.ref_count());
+
+  VARIANT raw;
+  raw.vt = VT_UNKNOWN;
+  raw.punkVal = &faker;
+  EXPECT_EQ(0, faker.ref_count());
+  var.Set(raw);
+  EXPECT_EQ(1, faker.ref_count());
+  var.Reset();
+  EXPECT_EQ(0, faker.ref_count());
+
+  {
+    ScopedVariant number(123);
+    EXPECT_EQ(VT_I4, number.type());
+    EXPECT_EQ(123, V_I4(&number));
+  }
+
   // SAFEARRAY tests
   var.Set(static_cast<SAFEARRAY*>(NULL));
   EXPECT_EQ(VT_EMPTY, var.type());
 
   SAFEARRAY* sa = ::SafeArrayCreateVector(VT_UI1, 0, 100);
   ASSERT_TRUE(sa != NULL);
 
   var.Set(sa);
 #ifndef OFFICIAL_BUILD
   EXPECT_TRUE(ScopedVariant::IsLeakableVarType(var.type()));
 #endif
   EXPECT_EQ(VT_ARRAY | VT_UI1, var.type());
   EXPECT_EQ(sa, V_ARRAY(&var));
   // The array is destroyed in the destructor of var.
 }