OLD | NEW |
| (Empty) |
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_EQ(memcmp(&ScopedComPtr<IUnknown>::iid(), &IID_IUnknown, sizeof(IID)), | |
35 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 | |
OLD | NEW |