base/win/scoped_comptr_unittest.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "base/win/scoped_comptr.h"
   6
   7 #include <shlobj.h>
   8
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/win/scoped_com_initializer.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace base {
  14 namespace win {
  15
  16 namespace {
  17
  18 struct Dummy {
  19   Dummy() : adds(0), releases(0) { }
  20   void AddRef() { ++adds; }
  21   void Release() { ++releases; }
  22
  23   int adds;
  24   int releases;
  25 };
  26
  27 extern const IID dummy_iid;
  28 const IID dummy_iid = { 0x12345678u, 0x1234u, 0x5678u, 01, 23, 45, 67, 89,
  29                         01, 23, 45 };
  30
  31 }  // namespace
  32
  33 TEST(ScopedComPtrTest, ScopedComPtr) {
  34   EXPECT_TRUE(memcmp(&ScopedComPtr<IUnknown>::iid(), &IID_IUnknown,
  35                      sizeof(IID)) == 0);
  36
  37   base::win::ScopedCOMInitializer com_initializer;
  38   EXPECT_TRUE(com_initializer.succeeded());
  39
  40   ScopedComPtr<IUnknown> unk;
  41   EXPECT_TRUE(SUCCEEDED(unk.CreateInstance(CLSID_ShellLink)));
  42   ScopedComPtr<IUnknown> unk2;
  43   unk2.Attach(unk.Detach());
  44   EXPECT_TRUE(unk.get() == NULL);
  45   EXPECT_TRUE(unk2.get() != NULL);
  46
  47   ScopedComPtr<IMalloc> mem_alloc;
  48   EXPECT_TRUE(SUCCEEDED(CoGetMalloc(1, mem_alloc.Receive())));
  49
  50   ScopedComPtr<IUnknown> qi_test;
  51   EXPECT_HRESULT_SUCCEEDED(mem_alloc.QueryInterface(IID_IUnknown,
  52       reinterpret_cast<void**>(qi_test.Receive())));
  53   EXPECT_TRUE(qi_test.get() != NULL);
  54   qi_test.Release();
  55
  56   // test ScopedComPtr& constructor
  57   ScopedComPtr<IMalloc> copy1(mem_alloc);
  58   EXPECT_TRUE(copy1.IsSameObject(mem_alloc.get()));
  59   EXPECT_FALSE(copy1.IsSameObject(unk2.get()));  // unk2 is valid but different
  60   EXPECT_FALSE(copy1.IsSameObject(unk.get()));  // unk is NULL
  61
  62   IMalloc* naked_copy = copy1.Detach();
  63   copy1 = naked_copy;  // Test the =(T*) operator.
  64   naked_copy->Release();
  65
  66   copy1.Release();
  67   EXPECT_FALSE(copy1.IsSameObject(unk2.get()));  // unk2 is valid, copy1 is not
  68
  69   // test Interface* constructor
  70   ScopedComPtr<IMalloc> copy2(static_cast<IMalloc*>(mem_alloc.get()));
  71   EXPECT_TRUE(copy2.IsSameObject(mem_alloc.get()));
  72
  73   EXPECT_TRUE(SUCCEEDED(unk.QueryFrom(mem_alloc.get())));
  74   EXPECT_TRUE(unk.get() != NULL);
  75   unk.Release();
  76   EXPECT_TRUE(unk.get() == NULL);
  77   EXPECT_TRUE(unk.IsSameObject(copy1.get()));  // both are NULL
  78 }
  79
  80 TEST(ScopedComPtrTest, ScopedComPtrVector) {
  81   // Verify we don't get error C2558.
  82   typedef ScopedComPtr<Dummy, &dummy_iid> Ptr;
  83   std::vector<Ptr> bleh;
  84
  85   scoped_ptr<Dummy> p(new Dummy);
  86   {
  87     Ptr p2(p.get());
  88     EXPECT_EQ(p->adds, 1);
  89     EXPECT_EQ(p->releases, 0);
  90     Ptr p3 = p2;
  91     EXPECT_EQ(p->adds, 2);
  92     EXPECT_EQ(p->releases, 0);
  93     p3 = p2;
  94     EXPECT_EQ(p->adds, 3);
  95     EXPECT_EQ(p->releases, 1);
  96     // To avoid hitting a reallocation.
  97     bleh.reserve(1);
  98     bleh.push_back(p2);
  99     EXPECT_EQ(p->adds, 4);
 100     EXPECT_EQ(p->releases, 1);
 101     EXPECT_EQ(bleh[0].get(), p.get());
 102     bleh.pop_back();
 103     EXPECT_EQ(p->adds, 4);
 104     EXPECT_EQ(p->releases, 2);
 105   }
 106   EXPECT_EQ(p->adds, 4);
 107   EXPECT_EQ(p->releases, 4);
 108 }
 109
 110 }  // namespace win
 111 }  // namespace base