1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/win/com_init_util.h"
#include "base/test/gtest_util.h"
#include "base/win/scoped_com_initializer.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace base {
namespace win {
TEST(ComInitUtil, AssertNotInitialized) {
if (DCHECK_IS_ON()) {
EXPECT_NOTREACHED_DEATH(AssertComInitialized());
}
}
TEST(ComInitUtil, AssertUninitialized) {
// When COM is uninitialized, the TLS data will remain, but the apartment
// status will be updated. This covers that case.
{
ScopedCOMInitializer com_initializer;
ASSERT_TRUE(com_initializer.Succeeded());
}
if (DCHECK_IS_ON()) {
EXPECT_NOTREACHED_DEATH(AssertComInitialized());
}
}
TEST(ComInitUtil, AssertSTAInitialized) {
ScopedCOMInitializer com_initializer;
ASSERT_TRUE(com_initializer.Succeeded());
AssertComInitialized();
}
TEST(ComInitUtil, AssertMTAInitialized) {
ScopedCOMInitializer com_initializer(ScopedCOMInitializer::kMTA);
ASSERT_TRUE(com_initializer.Succeeded());
AssertComInitialized();
}
TEST(ComInitUtil, AssertNoneApartmentType) {
AssertComApartmentType(ComApartmentType::NONE);
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::STA));
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::MTA));
}
TEST(ComInitUtil, AssertNoneApartmentTypeUninitialized) {
// When COM is uninitialized, the TLS data will remain, but the apartment
// status will be updated. This covers that case.
{
ScopedCOMInitializer com_initializer;
ASSERT_TRUE(com_initializer.Succeeded());
}
AssertComApartmentType(ComApartmentType::NONE);
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::STA));
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::MTA));
}
TEST(ComInitUtil, AssertSTAApartmentType) {
ScopedCOMInitializer com_initializer;
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::NONE));
AssertComApartmentType(ComApartmentType::STA);
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::MTA));
}
TEST(ComInitUtil, AssertMTAApartmentType) {
ScopedCOMInitializer com_initializer(ScopedCOMInitializer::kMTA);
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::NONE));
EXPECT_DCHECK_DEATH(AssertComApartmentType(ComApartmentType::STA));
AssertComApartmentType(ComApartmentType::MTA);
}
} // namespace win
} // namespace base
|
