blob: 815bf35a43f5e2ee19d3efa1aa53dceb31b78962 [file] [log] [blame]
/*
* Copyright (C) 2017 The Android Open foo Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "gmock/gmock.h"
#include "gtest/gtest.h"
#include "ipc/client.h"
#include "ipc/host.h"
#include "perfetto_base/test/test_task_runner.h"
#include "ipc/src/test/greeter_service.ipc.h"
#include "ipc/src/test/greeter_service.pb.h"
namespace ipc_test {
namespace {
using ::testing::_;
using ::testing::Invoke;
using ::perfetto::ipc::AsyncResult;
using ::perfetto::ipc::Client;
using ::perfetto::ipc::Deferred;
using ::perfetto::ipc::Host;
using ::perfetto::ipc::Service;
using ::perfetto::ipc::ServiceProxy;
constexpr char kSockName[] = "/tmp/perfetto_ipc_test.sock";
class MockEventListener : public ServiceProxy::EventListener {
public:
MOCK_METHOD0(OnConnect, void());
MOCK_METHOD0(OnDisconnect, void());
};
class MockGreeterService : public ipc_test::Greeter {
public:
MOCK_METHOD2(OnSayHello,
void(const GreeterRequestMsg&, DeferredGreeterReplyMsg*));
void SayHello(const GreeterRequestMsg& request,
DeferredGreeterReplyMsg reply) override {
OnSayHello(request, &reply);
}
MOCK_METHOD2(OnWaveGoodbye,
void(const GreeterRequestMsg&, DeferredGreeterReplyMsg*));
void WaveGoodbye(const GreeterRequestMsg& request,
DeferredGreeterReplyMsg reply) override {
OnWaveGoodbye(request, &reply);
}
};
class IPCIntegrationTest : public ::testing::Test {
protected:
void SetUp() override { unlink(kSockName); }
void TearDown() override { unlink(kSockName); }
perfetto::base::TestTaskRunner task_runner_;
MockEventListener svc_proxy_events_;
};
TEST_F(IPCIntegrationTest, SayHelloWaveGoodbye) {
std::unique_ptr<Host> host = Host::CreateInstance(kSockName, &task_runner_);
ASSERT_TRUE(host);
MockGreeterService* svc = new MockGreeterService();
ASSERT_TRUE(host->ExposeService(std::unique_ptr<Service>(svc)));
auto on_connect = task_runner_.CreateCheckpoint("on_connect");
EXPECT_CALL(svc_proxy_events_, OnConnect()).WillOnce(Invoke(on_connect));
std::unique_ptr<Client> cli =
Client::CreateInstance(kSockName, &task_runner_);
std::unique_ptr<GreeterProxy> svc_proxy(new GreeterProxy(&svc_proxy_events_));
cli->BindService(svc_proxy->GetWeakPtr());
task_runner_.RunUntilCheckpoint("on_connect");
{
GreeterRequestMsg req;
req.set_name("Mr Bojangles");
auto on_reply = task_runner_.CreateCheckpoint("on_hello_reply");
Deferred<GreeterReplyMsg> deferred_reply(
[on_reply](AsyncResult<GreeterReplyMsg> reply) {
ASSERT_TRUE(reply.success());
ASSERT_FALSE(reply.has_more());
ASSERT_EQ("Hello Mr Bojangles", reply->message());
on_reply();
});
EXPECT_CALL(*svc, OnSayHello(_, _))
.WillOnce(Invoke([](const GreeterRequestMsg& host_req,
Deferred<GreeterReplyMsg>* host_reply) {
auto reply = AsyncResult<GreeterReplyMsg>::Create();
reply->set_message("Hello " + host_req.name());
host_reply->Resolve(std::move(reply));
}));
svc_proxy->SayHello(req, std::move(deferred_reply));
task_runner_.RunUntilCheckpoint("on_hello_reply");
}
{
GreeterRequestMsg req;
req.set_name("Mrs Bojangles");
auto on_reply = task_runner_.CreateCheckpoint("on_goodbye_reply");
Deferred<GreeterReplyMsg> deferred_reply(
[on_reply](AsyncResult<GreeterReplyMsg> reply) {
ASSERT_TRUE(reply.success());
ASSERT_FALSE(reply.has_more());
ASSERT_EQ("Goodbye Mrs Bojangles", reply->message());
on_reply();
});
EXPECT_CALL(*svc, OnWaveGoodbye(_, _))
.WillOnce(Invoke([](const GreeterRequestMsg& host_req,
Deferred<GreeterReplyMsg>* host_reply) {
auto reply = AsyncResult<GreeterReplyMsg>::Create();
reply->set_message("Goodbye " + host_req.name());
host_reply->Resolve(std::move(reply));
}));
svc_proxy->WaveGoodbye(req, std::move(deferred_reply));
task_runner_.RunUntilCheckpoint("on_goodbye_reply");
}
}
} // namespace
} // namespace ipc_test