1 /*
2 * Copyright (C) 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include "perfetto/ext/ipc/client.h"
18 #include "perfetto/ext/ipc/host.h"
19 #include "src/base/test/test_task_runner.h"
20 #include "src/ipc/test/test_socket.h"
21 #include "test/gtest_and_gmock.h"
22
23 #include "src/ipc/test/greeter_service.gen.h"
24 #include "src/ipc/test/greeter_service.ipc.h"
25
26 namespace ipc_test {
27 namespace {
28
29 using ::perfetto::ipc::AsyncResult;
30 using ::perfetto::ipc::Client;
31 using ::perfetto::ipc::Deferred;
32 using ::perfetto::ipc::Host;
33 using ::perfetto::ipc::Service;
34 using ::perfetto::ipc::ServiceProxy;
35 using ::testing::_;
36 using ::testing::Invoke;
37
38 using namespace ::ipc_test::gen;
39
40 ::perfetto::ipc::TestSocket kTestSocket{"ipc_integrationtest"};
41
42 class MockEventListener : public ServiceProxy::EventListener {
43 public:
44 MOCK_METHOD(void, OnConnect, (), (override));
45 MOCK_METHOD(void, OnDisconnect, (), (override));
46 };
47
48 class MockGreeterService : public ::ipc_test::gen::Greeter {
49 public:
50 MOCK_METHOD(void,
51 OnSayHello,
52 (const GreeterRequestMsg&, DeferredGreeterReplyMsg*));
SayHello(const GreeterRequestMsg & request,DeferredGreeterReplyMsg reply)53 void SayHello(const GreeterRequestMsg& request,
54 DeferredGreeterReplyMsg reply) override {
55 OnSayHello(request, &reply);
56 }
57
58 MOCK_METHOD(void,
59 OnWaveGoodbye,
60 (const GreeterRequestMsg&, DeferredGreeterReplyMsg*));
WaveGoodbye(const GreeterRequestMsg & request,DeferredGreeterReplyMsg reply)61 void WaveGoodbye(const GreeterRequestMsg& request,
62 DeferredGreeterReplyMsg reply) override {
63 OnWaveGoodbye(request, &reply);
64 }
65 };
66
67 class IPCIntegrationTest : public ::testing::Test {
68 protected:
SetUp()69 void SetUp() override { kTestSocket.Destroy(); }
TearDown()70 void TearDown() override { kTestSocket.Destroy(); }
71
72 perfetto::base::TestTaskRunner task_runner_;
73 MockEventListener svc_proxy_events_;
74 };
75
TEST_F(IPCIntegrationTest,SayHelloWaveGoodbye)76 TEST_F(IPCIntegrationTest, SayHelloWaveGoodbye) {
77 #if PERFETTO_BUILDFLAG(PERFETTO_OS_FUCHSIA)
78 std::unique_ptr<Host> host = Host::CreateInstance_Fuchsia(&task_runner_);
79 #else
80 std::unique_ptr<Host> host =
81 Host::CreateInstance(kTestSocket.name(), &task_runner_);
82 #endif // PERFETTO_BUILDFLAG(PERFETTO_OS_FUCHSIA)
83 ASSERT_TRUE(host);
84
85 MockGreeterService* svc = new MockGreeterService();
86 ASSERT_TRUE(host->ExposeService(std::unique_ptr<Service>(svc)));
87
88 auto on_connect = task_runner_.CreateCheckpoint("on_connect");
89 EXPECT_CALL(svc_proxy_events_, OnConnect()).WillOnce(Invoke(on_connect));
90
91 #if PERFETTO_BUILDFLAG(PERFETTO_OS_FUCHSIA)
92 auto socket_pair = perfetto::base::UnixSocketRaw::CreatePairPosix(
93 perfetto::base::SockFamily::kUnix, perfetto::base::SockType::kStream);
94
95 std::unique_ptr<Client> cli = Client::CreateInstance(
96 Client::ConnArgs(
97 perfetto::base::ScopedSocketHandle(socket_pair.first.ReleaseFd())),
98 &task_runner_);
99 host->AdoptConnectedSocket_Fuchsia(
100 perfetto::base::ScopedSocketHandle(socket_pair.second.ReleaseFd()),
101 [](int) { return false; });
102 #else
103 std::unique_ptr<Client> cli = Client::CreateInstance(
104 {kTestSocket.name(), /*retry=*/false}, &task_runner_);
105 #endif // PERFETTO_BUILDFLAG(PERFETTO_OS_FUCHSIA)
106 std::unique_ptr<GreeterProxy> svc_proxy(new GreeterProxy(&svc_proxy_events_));
107 cli->BindService(svc_proxy->GetWeakPtr());
108 task_runner_.RunUntilCheckpoint("on_connect");
109
110 {
111 GreeterRequestMsg req;
112 req.set_name("Mr Bojangles");
113 auto on_reply = task_runner_.CreateCheckpoint("on_hello_reply");
114 Deferred<GreeterReplyMsg> deferred_reply(
115 [on_reply](AsyncResult<GreeterReplyMsg> reply) {
116 ASSERT_TRUE(reply.success());
117 ASSERT_FALSE(reply.has_more());
118 ASSERT_EQ("Hello Mr Bojangles", reply->message());
119 on_reply();
120 });
121
122 EXPECT_CALL(*svc, OnSayHello(_, _))
123 .WillOnce(Invoke([](const GreeterRequestMsg& host_req,
124 Deferred<GreeterReplyMsg>* host_reply) {
125 auto reply = AsyncResult<GreeterReplyMsg>::Create();
126 reply->set_message("Hello " + host_req.name());
127 host_reply->Resolve(std::move(reply));
128 }));
129 svc_proxy->SayHello(req, std::move(deferred_reply));
130 task_runner_.RunUntilCheckpoint("on_hello_reply");
131 }
132
133 {
134 GreeterRequestMsg req;
135 req.set_name("Mrs Bojangles");
136 auto on_reply = task_runner_.CreateCheckpoint("on_goodbye_reply");
137 Deferred<GreeterReplyMsg> deferred_reply(
138 [on_reply](AsyncResult<GreeterReplyMsg> reply) {
139 ASSERT_TRUE(reply.success());
140 ASSERT_FALSE(reply.has_more());
141 ASSERT_EQ("Goodbye Mrs Bojangles", reply->message());
142 on_reply();
143 });
144
145 EXPECT_CALL(*svc, OnWaveGoodbye(_, _))
146 .WillOnce(Invoke([](const GreeterRequestMsg& host_req,
147 Deferred<GreeterReplyMsg>* host_reply) {
148 auto reply = AsyncResult<GreeterReplyMsg>::Create();
149 reply->set_message("Goodbye " + host_req.name());
150 host_reply->Resolve(std::move(reply));
151 }));
152 svc_proxy->WaveGoodbye(req, std::move(deferred_reply));
153 task_runner_.RunUntilCheckpoint("on_goodbye_reply");
154 }
155 }
156
157 } // namespace
158 } // namespace ipc_test
159