223 lines
6.7 KiB
C++
223 lines
6.7 KiB
C++
/*
|
|
* Copyright (c) 2012 by Michael Berlin, Zuse Institute Berlin
|
|
*
|
|
* Licensed under the BSD License, see LICENSE file for details.
|
|
*
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <boost/scoped_ptr.hpp>
|
|
#include <string>
|
|
|
|
#include "libxtreemfs/pbrpc_url.h"
|
|
#include "libxtreemfs/xtreemfs_exception.h"
|
|
#include "util/logging.h"
|
|
#include "xtreemfs/GlobalTypes.pb.h"
|
|
|
|
using namespace std;
|
|
using namespace xtreemfs::pbrpc;
|
|
using namespace xtreemfs::util;
|
|
|
|
namespace xtreemfs {
|
|
|
|
class PBRPCURLTest : public ::testing::Test {
|
|
protected:
|
|
virtual void SetUp() {
|
|
initialize_logger(LEVEL_WARN);
|
|
|
|
volume_name = "test";
|
|
default_scheme = PBRPCURL::GetSchemePBRPC();
|
|
default_port = DIR_PBRPC_PORT_DEFAULT;
|
|
|
|
servers.push_back("localhost");
|
|
servers.push_back("127.0.0.1");
|
|
servers.push_back("somehost");
|
|
|
|
ports.push_back(1234);
|
|
ports.push_back(4242);
|
|
ports.push_back(31337);
|
|
|
|
pbrpc_url_.reset(new PBRPCURL());
|
|
}
|
|
|
|
virtual void TearDown() {
|
|
shutdown_logger();
|
|
}
|
|
|
|
typedef list<string> ServerList;
|
|
typedef list<uint16_t> PortList;
|
|
|
|
ServerList servers;
|
|
PortList ports;
|
|
string volume_name;
|
|
string default_scheme;
|
|
uint16_t default_port;
|
|
|
|
boost::scoped_ptr<PBRPCURL> pbrpc_url_;
|
|
};
|
|
|
|
TEST_F(PBRPCURLTest, URLWithOneAddressAndVolume) {
|
|
stringstream url_to_parse;
|
|
url_to_parse << servers.front() << '/' << volume_name;
|
|
pbrpc_url_->ParseURL(url_to_parse.str(), default_scheme, default_port);
|
|
|
|
ServiceAddresses addresses = pbrpc_url_->GetAddresses();
|
|
|
|
EXPECT_FALSE(addresses.empty());
|
|
EXPECT_EQ(volume_name, pbrpc_url_->volume());
|
|
EXPECT_EQ(PBRPCURL::GetSchemePBRPC(), pbrpc_url_->scheme());
|
|
|
|
stringstream expected_address;
|
|
expected_address << "localhost:" << DIR_PBRPC_PORT_DEFAULT;
|
|
EXPECT_EQ(expected_address.str(), addresses.GetAddresses().front());
|
|
}
|
|
|
|
TEST_F(PBRPCURLTest, URLWithMultipleAddressesAndVolume) {
|
|
// Build an URL to parse
|
|
stringstream url_to_parse;
|
|
ServerList::const_iterator servers_it = servers.begin();
|
|
|
|
for (; servers_it != servers.end(); ++servers_it) {
|
|
if(servers_it != servers.begin()) {
|
|
url_to_parse << ',';
|
|
}
|
|
url_to_parse << *servers_it;
|
|
}
|
|
url_to_parse << '/' << volume_name;
|
|
|
|
// Parse URL and get addresses
|
|
pbrpc_url_->ParseURL(url_to_parse.str(), default_scheme, default_port);
|
|
ServiceAddresses addresses = pbrpc_url_->GetAddresses();
|
|
|
|
// Check expectations
|
|
EXPECT_EQ(servers.size(), addresses.size());
|
|
EXPECT_EQ(volume_name, pbrpc_url_->volume());
|
|
EXPECT_EQ(default_scheme, pbrpc_url_->scheme());
|
|
|
|
servers_it = servers.begin();
|
|
stringstream expected_address;
|
|
ServiceAddresses::Addresses services = addresses.GetAddresses();
|
|
for(ServiceAddresses::Addresses::iterator it = services.begin();
|
|
it != services.end();
|
|
++it, ++servers_it) {
|
|
expected_address.str("");
|
|
expected_address << *servers_it << ':' << default_port;
|
|
EXPECT_EQ(expected_address.str(), *it);
|
|
}
|
|
}
|
|
|
|
TEST_F(PBRPCURLTest, URLWithMultipleAddressesPortsAndVolume) {
|
|
// Build an URL to parse
|
|
stringstream url_to_parse;
|
|
ServerList::const_iterator servers_it = servers.begin();
|
|
PortList::const_iterator ports_it = ports.begin();
|
|
|
|
for (; servers_it != servers.end(); ++servers_it, ++ports_it) {
|
|
if(servers_it != servers.begin()) {
|
|
url_to_parse << ',';
|
|
}
|
|
url_to_parse << *servers_it << ':' << *ports_it;
|
|
}
|
|
url_to_parse << '/' << volume_name;
|
|
|
|
// Parse URL and get addresses
|
|
pbrpc_url_->ParseURL(url_to_parse.str(), default_scheme, default_port);
|
|
ServiceAddresses addresses = pbrpc_url_->GetAddresses();
|
|
|
|
// Check expectations
|
|
EXPECT_EQ(servers.size(), addresses.size());
|
|
EXPECT_EQ(volume_name, pbrpc_url_->volume());
|
|
EXPECT_EQ(default_scheme, pbrpc_url_->scheme());
|
|
|
|
servers_it = servers.begin();
|
|
ports_it = ports.begin();
|
|
stringstream expected_address;
|
|
ServiceAddresses::Addresses services = addresses.GetAddresses();
|
|
for(ServiceAddresses::Addresses::iterator it = services.begin();
|
|
it != services.end();
|
|
++it, ++servers_it, ++ports_it) {
|
|
expected_address.str("");
|
|
expected_address << *servers_it << ':' << *ports_it;
|
|
EXPECT_EQ(expected_address.str(), *it);
|
|
}
|
|
}
|
|
|
|
TEST_F(PBRPCURLTest, URLWithMultipleAddressesProtocolsPortsAndVolume) {
|
|
// Build an URL to parse
|
|
stringstream url_to_parse;
|
|
ServerList::const_iterator servers_it = servers.begin();
|
|
PortList::const_iterator ports_it = ports.begin();
|
|
|
|
for (; servers_it != servers.end(); ++servers_it, ++ports_it) {
|
|
if(servers_it != servers.begin()) {
|
|
url_to_parse << ',';
|
|
}
|
|
url_to_parse << default_scheme << "://" << *servers_it << ':' << *ports_it;
|
|
}
|
|
url_to_parse << '/' << volume_name;
|
|
// Parse URL and get addresses
|
|
pbrpc_url_->ParseURL(url_to_parse.str(), default_scheme, default_port);
|
|
ServiceAddresses addresses = pbrpc_url_->GetAddresses();
|
|
|
|
// Check expectations
|
|
EXPECT_EQ(servers.size(), addresses.size());
|
|
EXPECT_EQ(volume_name, pbrpc_url_->volume());
|
|
EXPECT_EQ(default_scheme, pbrpc_url_->scheme());
|
|
|
|
servers_it = servers.begin();
|
|
ports_it = ports.begin();
|
|
stringstream expected_address;
|
|
ServiceAddresses::Addresses services = addresses.GetAddresses();
|
|
for(ServiceAddresses::Addresses::iterator it = services.begin();
|
|
it != services.end();
|
|
++it, ++servers_it, ++ports_it) {
|
|
expected_address.str("");
|
|
expected_address << *servers_it << ':' << *ports_it;
|
|
EXPECT_EQ(expected_address.str(), *it);
|
|
}
|
|
}
|
|
|
|
TEST_F(PBRPCURLTest, URLWithMultipleAddressesProtocolsAndVolume) {
|
|
// Build an URL to parse
|
|
stringstream url_to_parse;
|
|
ServerList::const_iterator servers_it = servers.begin();
|
|
|
|
for (; servers_it != servers.end(); ++servers_it) {
|
|
if(servers_it != servers.begin()) {
|
|
url_to_parse << ',';
|
|
}
|
|
url_to_parse << default_scheme << "://" << *servers_it;
|
|
}
|
|
url_to_parse << '/' << volume_name;
|
|
// Parse URL and get addresses
|
|
pbrpc_url_->ParseURL(url_to_parse.str(), default_scheme, default_port);
|
|
ServiceAddresses addresses = pbrpc_url_->GetAddresses();
|
|
|
|
// Check expectations
|
|
EXPECT_EQ(servers.size(), addresses.size());
|
|
EXPECT_EQ(volume_name, pbrpc_url_->volume());
|
|
EXPECT_EQ(default_scheme, pbrpc_url_->scheme());
|
|
|
|
servers_it = servers.begin();
|
|
stringstream expected_address;
|
|
ServiceAddresses::Addresses services = addresses.GetAddresses();
|
|
for(ServiceAddresses::Addresses::iterator it = services.begin();
|
|
it != services.end();
|
|
++it, ++servers_it) {
|
|
expected_address.str("");
|
|
expected_address << *servers_it << ':' << default_port;
|
|
EXPECT_EQ(expected_address.str(), *it);
|
|
}
|
|
}
|
|
|
|
TEST_F(PBRPCURLTest, URLAddressesWithDifferentProtocols) {
|
|
EXPECT_THROW({
|
|
pbrpc_url_->ParseURL("pbrpc://localhost,pbrpcg://remote/" + volume_name,
|
|
PBRPCURL::GetSchemePBRPC(),
|
|
DIR_PBRPC_PORT_DEFAULT);
|
|
}, InvalidURLException);
|
|
}
|
|
|
|
} // namespace xtreemfs
|