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
80
81
82
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "mojo/core/test/scoped_mojo_support.h"
#include "base/base_switches.h"
#include "base/command_line.h"
#include "base/feature_list.h"
#include "base/synchronization/waitable_event.h"
#include "base/test/bind.h"
#include "base/test/test_io_thread.h"
#include "mojo/core/embedder/configuration.h"
#include "mojo/core/embedder/embedder.h"
#include "mojo/core/embedder/scoped_ipc_support.h"
#include "mojo/core/test/mojo_test_base.h"
#include "mojo/core/test/test_support_impl.h"
#include "mojo/core/test/test_switches.h"
#include "mojo/public/tests/test_support_private.h"
namespace mojo::core::test {
namespace {
// Command-line switch to disable internal Mojo Channel capability
// advertisement, used to test skew between client versions.
const char kDisableAllCapabilities[] = "disable-all-capabilities";
class TestSupportInitializer {
public:
TestSupportInitializer() {
mojo::test::TestSupport::Init(new mojo::core::test::TestSupportImpl());
}
};
} // namespace
class ScopedMojoSupport::CoreInstance {
public:
CoreInstance() {
mojo::core::Configuration mojo_config;
// A relatively low limit to make it easier to test behavior at the limit.
mojo_config.max_message_num_bytes =
mojo::core::test::MojoTestBase::kMaxMessageSizeInTests;
if (!base::CommandLine::ForCurrentProcess()->HasSwitch(
switches::kTestChildProcess) ||
base::CommandLine::ForCurrentProcess()->HasSwitch(
test_switches::kMojoIsBroker)) {
mojo_config.is_broker_process = true;
}
if (base::CommandLine::ForCurrentProcess()->HasSwitch(
kDisableAllCapabilities)) {
mojo_config.dont_advertise_capabilities = true;
}
mojo::core::InitFeatures();
mojo::core::Init(mojo_config);
static TestSupportInitializer initializer;
}
~CoreInstance() { mojo::core::ShutDown(); }
};
ScopedMojoSupport::ScopedMojoSupport()
: core_(std::make_unique<CoreInstance>()) {
// IO thread initialization can race to modify globals which other base object
// initializations (e.g. TaskEnvironment) might touch on the main thread as a
// side effect of any test that might run. Ensure the thread is fully started
// before we proceed. See https://crbug.com/1364731.
base::WaitableEvent io_thread_initialized;
test_io_thread_.PostTask(FROM_HERE, base::BindLambdaForTesting([&] {
io_thread_initialized.Signal();
}));
io_thread_initialized.Wait();
}
ScopedMojoSupport::~ScopedMojoSupport() = default;
} // namespace mojo::core::test
|
