1 // Copyright 2018 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 <memory>
6 #include <utility>
7
8 #include "base/bind.h"
9 #include "base/macros.h"
10 #include "base/run_loop.h"
11 #include "base/test/task_environment.h"
12 #include "mojo/public/cpp/bindings/receiver.h"
13 #include "mojo/public/cpp/bindings/remote.h"
14 #include "services/tracing/public/cpp/perfetto/perfetto_config.h"
15 #include "services/tracing/public/mojom/perfetto_service.mojom.h"
16 #include "services/tracing/public/mojom/tracing_service.mojom.h"
17 #include "services/tracing/tracing_service.h"
18 #include "testing/gtest/include/gtest/gtest.h"
19
20 namespace tracing {
21
22 class TracingServiceTest : public testing::Test {
23 public:
24 TracingServiceTest() = default;
25 ~TracingServiceTest() override = default;
26
27 protected:
service()28 mojom::TracingService* service() { return &service_; }
29
30 private:
31 base::test::TaskEnvironment task_environment_;
32 TracingService service_;
33
34 DISALLOW_COPY_AND_ASSIGN(TracingServiceTest);
35 };
36
37 class TestTracingClient : public mojom::TracingSessionClient {
38 public:
StartTracing(mojom::TracingService * service,base::OnceClosure on_tracing_enabled)39 void StartTracing(mojom::TracingService* service,
40 base::OnceClosure on_tracing_enabled) {
41 service->BindConsumerHost(consumer_host_.BindNewPipeAndPassReceiver());
42
43 perfetto::TraceConfig perfetto_config =
44 tracing::GetDefaultPerfettoConfig(base::trace_event::TraceConfig(""),
45 /*privacy_filtering_enabled=*/false);
46
47 consumer_host_->EnableTracing(
48 tracing_session_host_.BindNewPipeAndPassReceiver(),
49 receiver_.BindNewPipeAndPassRemote(), std::move(perfetto_config),
50 tracing::mojom::TracingClientPriority::kUserInitiated);
51
52 tracing_session_host_->RequestBufferUsage(
53 base::BindOnce([](base::OnceClosure on_response, bool, float,
54 bool) { std::move(on_response).Run(); },
55 std::move(on_tracing_enabled)));
56 }
57
58 // tracing::mojom::TracingSessionClient implementation:
OnTracingEnabled()59 void OnTracingEnabled() override {}
OnTracingDisabled()60 void OnTracingDisabled() override {}
61
62 private:
63 mojo::Remote<mojom::ConsumerHost> consumer_host_;
64 mojo::Remote<mojom::TracingSessionHost> tracing_session_host_;
65 mojo::Receiver<mojom::TracingSessionClient> receiver_{this};
66 };
67
TEST_F(TracingServiceTest,TracingServiceInstantiate)68 TEST_F(TracingServiceTest, TracingServiceInstantiate) {
69 TestTracingClient tracing_client;
70
71 base::RunLoop tracing_started;
72 tracing_client.StartTracing(service(), tracing_started.QuitClosure());
73 tracing_started.Run();
74 }
75
76 } // namespace tracing
77