Adding a non-templatized version of QI for times when the IID and type aren't associated.

Also fixing the ScopedComPtrVector unit test.
The problem with the test was that it expected AddRef to be called twice and Release() once when pushing a pointer to the vector.  However that only happens when a push causes a reallocation.  When there's no reallocation, there's only a single call to AddRef and no call to Release, causing the test to report failure.  The fix is to simply use std::list instead and fix the expected values.

TEST=Run *ScopedComPtr* unit tests.
BUG=none
Review URL: http://codereview.chromium.org/149345

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@20183 0039d316-1c4b-4281-b951-d872f2087c98
Esse commit está contido em:
tommi@chromium.org
2009-07-08 20:47:31 +00:00
commit 8089a74233
2 arquivos alterados com 24 adições e 7 exclusões
+7
Ver Arquivo
@@ -89,6 +89,13 @@ class ScopedComPtr : public scoped_refptr<Interface> {
return ptr_->QueryInterface(p);
}
// QI for times when the IID is not associated with the type.
HRESULT QueryInterface(const IID& iid, void** obj) {
DCHECK(obj != NULL);
DCHECK(ptr_ != NULL);
return ptr_->QueryInterface(iid, obj);
}
// Queries |other| for the interface this object wraps and returns the
// error code from the other->QueryInterface operation.
HRESULT QueryFrom(IUnknown* object) {
+17 -7
Ver Arquivo
@@ -2,9 +2,10 @@
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/scoped_comptr_win.h"
#include <shlobj.h>
#include "base/scoped_comptr_win.h"
#include "base/scoped_ptr.h"
#include "testing/gtest/include/gtest/gtest.h"
@@ -42,6 +43,12 @@ TEST(ScopedComPtrTest, ScopedComPtr) {
ScopedComPtr<IMalloc> mem_alloc;
EXPECT_TRUE(SUCCEEDED(CoGetMalloc(1, mem_alloc.Receive())));
ScopedComPtr<IUnknown> qi_test;
EXPECT_HRESULT_SUCCEEDED(mem_alloc.QueryInterface(IID_IUnknown,
reinterpret_cast<void**>(qi_test.Receive())));
EXPECT_TRUE(qi_test.get() != NULL);
qi_test.Release();
// test ScopedComPtr& constructor
ScopedComPtr<IMalloc> copy1(mem_alloc);
EXPECT_TRUE(copy1.IsSameObject(mem_alloc));
@@ -73,6 +80,7 @@ TEST(ScopedComPtrTest, ScopedComPtrVector) {
// Verify we don't get error C2558.
typedef ScopedComPtr<Dummy, &dummy_iid> Ptr;
std::vector<Ptr> bleh;
scoped_ptr<Dummy> p(new Dummy);
{
Ptr p2(p.get());
@@ -84,14 +92,16 @@ TEST(ScopedComPtrTest, ScopedComPtrVector) {
p3 = p2;
EXPECT_EQ(p->adds, 3);
EXPECT_EQ(p->releases, 1);
// To avoid hitting a reallocation.
bleh.reserve(1);
bleh.push_back(p2);
EXPECT_EQ(p->adds, 5);
EXPECT_EQ(p->releases, 2);
EXPECT_EQ(p->adds, 4);
EXPECT_EQ(p->releases, 1);
EXPECT_EQ(bleh[0], p.get());
bleh.pop_back();
EXPECT_EQ(p->adds, 5);
EXPECT_EQ(p->releases, 3);
EXPECT_EQ(p->adds, 4);
EXPECT_EQ(p->releases, 2);
}
EXPECT_EQ(p->adds, 5);
EXPECT_EQ(p->releases, 5);
EXPECT_EQ(p->adds, 4);
EXPECT_EQ(p->releases, 4);
}