mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 08:47:12 +03:00
13da025c6c
Summary: This diff adds `ServiceAddress` to utils. This is a class that encapsulates an address of a service that could be a traditional hostname & port pair as well as an SMC tier name. This eliminates the needs of the manually resolving network address for remote services. Reviewed By: chadaustin Differential Revision: D14845257 fbshipit-source-id: 9fe9847cca4bba0170be94b9c209247342708574
97 lines
2.7 KiB
C++
97 lines
2.7 KiB
C++
/*
|
|
* Copyright (c) 2018-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree. An additional grant
|
|
* of patent rights can be found in the PATENTS file in the same directory.
|
|
*
|
|
*/
|
|
#include "eden/fs/utils/ServiceAddress.h"
|
|
|
|
#include <folly/SocketAddress.h>
|
|
#include <folly/logging/xlog.h>
|
|
#include <gtest/gtest.h>
|
|
#include "eden/fs/eden-config.h"
|
|
|
|
#ifdef EDEN_HAVE_SERVICEROUTER
|
|
#include <servicerouter/client/cpp2/ServiceRouter.h>
|
|
#endif
|
|
|
|
using namespace facebook::eden;
|
|
|
|
TEST(ServiceAddressTest, fromHostnameAndPort) {
|
|
auto hostname = "::1";
|
|
auto svc = ServiceAddress{hostname, 1234};
|
|
auto result = svc.getSocketAddressBlocking();
|
|
|
|
EXPECT_EQ(result->first.getAddressStr(), "::1");
|
|
EXPECT_EQ(result->first.getPort(), 1234);
|
|
EXPECT_EQ(result->second, "::1");
|
|
}
|
|
|
|
TEST(ServiceAddressTest, nonexistentHostname) {
|
|
auto hostname = "this-hostname-should-never-exist";
|
|
auto svc = ServiceAddress{hostname, 1234};
|
|
EXPECT_THROW(svc.getSocketAddressBlocking(), std::system_error);
|
|
}
|
|
|
|
#ifdef EDEN_HAVE_SERVICEROUTER
|
|
|
|
namespace {
|
|
using namespace facebook::servicerouter;
|
|
|
|
class MockServiceCacheIf : public ServiceCacheIf {
|
|
public:
|
|
virtual Selection getSelection(
|
|
const std::string& serviceName,
|
|
const ServiceOptions& /* options */,
|
|
const ConnConfigs& /* overrides */) override {
|
|
Selection selection;
|
|
|
|
if (serviceName == "mononoke-apiserver") {
|
|
auto location = std::make_shared<HostInfoLocation>("::1", 1234);
|
|
location->setHostname("some-hostname");
|
|
|
|
selection.hosts.push_back(std::make_shared<HostInfo>(
|
|
std::make_unique<HostInfoProperties>(), std::move(location)));
|
|
}
|
|
|
|
return selection;
|
|
}
|
|
|
|
virtual void getSelectionAsync(
|
|
const std::string& /* serviceName */,
|
|
DebugContext&& /* dbgCtx */,
|
|
SelectionCacheCallback&& /* callback */,
|
|
folly::EventBase* /* eventBase */,
|
|
ServiceOptions&& /* options */,
|
|
ConnConfigs&& /* overrides */) override {}
|
|
|
|
bool invalidateSelection(
|
|
const string& /* serviceName */,
|
|
const Config& /* cfg */) override {
|
|
return true;
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
TEST(ServiceAddressTest, fromSMCTier) {
|
|
auto tier = "mononoke-apiserver";
|
|
auto svc = ServiceAddress{tier};
|
|
auto result = svc.addressFromSMCTier(std::make_shared<MockServiceCacheIf>());
|
|
|
|
EXPECT_EQ(result->first.getAddressStr(), "::1");
|
|
EXPECT_EQ(result->first.getPort(), 1234);
|
|
EXPECT_EQ(result->second, "some-hostname");
|
|
}
|
|
|
|
TEST(ServiceAddressTest, failFromSMCTier) {
|
|
auto tier = "nonexistent-tier";
|
|
auto svc = ServiceAddress{tier};
|
|
auto result = svc.addressFromSMCTier(std::make_shared<MockServiceCacheIf>());
|
|
EXPECT_EQ(result, std::nullopt);
|
|
}
|
|
|
|
#endif
|