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/iunknown_impl.h" | |
6 | |
7 #include "base/win/scoped_com_initializer.h" | |
8 #include "testing/gtest/include/gtest/gtest.h" | |
9 | |
10 namespace base { | |
11 namespace win { | |
12 | |
13 class TestIUnknownImplSubclass : public IUnknownImpl { | |
14 public: | |
15 TestIUnknownImplSubclass() { | |
16 ++instance_count; | |
17 } | |
18 ~TestIUnknownImplSubclass() override { --instance_count; } | |
19 static int instance_count; | |
20 }; | |
21 | |
22 // static | |
23 int TestIUnknownImplSubclass::instance_count = 0; | |
24 | |
25 TEST(IUnknownImplTest, IUnknownImpl) { | |
26 ScopedCOMInitializer com_initializer; | |
27 | |
28 EXPECT_EQ(0, TestIUnknownImplSubclass::instance_count); | |
29 IUnknown* u = new TestIUnknownImplSubclass(); | |
30 | |
31 EXPECT_EQ(1, TestIUnknownImplSubclass::instance_count); | |
32 | |
33 EXPECT_EQ(1, u->AddRef()); | |
34 EXPECT_EQ(1, u->AddRef()); | |
35 | |
36 IUnknown* other = NULL; | |
37 EXPECT_EQ(E_NOINTERFACE, u->QueryInterface( | |
38 IID_IDispatch, reinterpret_cast<void**>(&other))); | |
39 EXPECT_EQ(S_OK, u->QueryInterface( | |
40 IID_IUnknown, reinterpret_cast<void**>(&other))); | |
41 other->Release(); | |
42 | |
43 EXPECT_EQ(1, u->Release()); | |
44 EXPECT_EQ(0, u->Release()); | |
45 EXPECT_EQ(0, TestIUnknownImplSubclass::instance_count); | |
46 } | |
47 | |
48 } // namespace win | |
49 } // namespace base | |
OLD | NEW |