blob: 576e4458d8838746287de65c070d684f316ebd2f [file] [log] [blame]
// Copyright 2021 The Pigweed Authors
//
// 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
//
// https://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 "pw_rpc/integration_testing.h"
#include <limits>
#include "gtest/gtest.h"
#include "pw_log/log.h"
#include "pw_rpc/integration_test_socket_client.h"
#include "pw_unit_test/logging_event_handler.h"
namespace pw::rpc::integration_test {
namespace {
SocketClientContext<512> context;
unit_test::LoggingEventHandler log_test_events;
} // namespace
Client& client() { return context.client(); }
int GetClientSocketFd() { return context.GetSocketFd(); }
void SetEgressChannelManipulator(ChannelManipulator* new_channel_manipulator) {
context.SetEgressChannelManipulator(new_channel_manipulator);
}
void SetIngressChannelManipulator(ChannelManipulator* new_channel_manipulator) {
context.SetIngressChannelManipulator(new_channel_manipulator);
}
Status InitializeClient(int port) {
unit_test::RegisterEventHandler(&log_test_events);
if (port <= 0 || port > std::numeric_limits<uint16_t>::max()) {
PW_LOG_CRITICAL("Port numbers must be between 1 and 65535; %d is invalid",
port);
return Status::InvalidArgument();
}
PW_LOG_INFO("Connecting to pw_rpc client at localhost:%d", port);
return context.Start(port);
}
void TerminateClient() { context.Terminate(); }
Status InitializeClient(int argc, char* argv[], const char* usage_args) {
if (argc < 2) {
PW_LOG_INFO("Usage: %s %s", argv[0], usage_args);
return Status::InvalidArgument();
}
const int port = std::atoi(argv[1]);
return InitializeClient(port);
}
} // namespace pw::rpc::integration_test