123 lines
4.2 KiB
C++
123 lines
4.2 KiB
C++
#include <boost/test/unit_test.hpp>
|
|
|
|
#include <fc/network/http/websocket.hpp>
|
|
|
|
#include <iostream>
|
|
#include <fc/log/logger.hpp>
|
|
#include <fc/log/console_appender.hpp>
|
|
|
|
BOOST_AUTO_TEST_SUITE(fc_network)
|
|
|
|
BOOST_AUTO_TEST_CASE(websocket_test)
|
|
{
|
|
// set up logging
|
|
fc::shared_ptr<fc::console_appender> ca(new fc::console_appender);
|
|
fc::logger l = fc::logger::get("rpc");
|
|
l.add_appender( ca );
|
|
|
|
fc::http::websocket_client client;
|
|
fc::http::websocket_connection_ptr s_conn, c_conn;
|
|
int port;
|
|
{
|
|
fc::http::websocket_server server("MyProxyHeaderKey");
|
|
server.on_connection([&]( const fc::http::websocket_connection_ptr& c ){
|
|
s_conn = c;
|
|
c->on_message_handler([&](const std::string& s){
|
|
c->send_message("echo: " + s);
|
|
});
|
|
});
|
|
|
|
server.listen( 0 );
|
|
port = server.get_listening_port();
|
|
|
|
server.start_accept();
|
|
|
|
std::string echo;
|
|
c_conn = client.connect( "ws://localhost:" + fc::to_string(port) );
|
|
c_conn->on_message_handler([&](const std::string& s){
|
|
echo = s;
|
|
});
|
|
c_conn->send_message( "hello world" );
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_EQUAL("echo: hello world", echo);
|
|
c_conn->send_message( "again" );
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_EQUAL("echo: again", echo);
|
|
|
|
s_conn->close(0, "test");
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_THROW(c_conn->send_message( "again" ), fc::exception);
|
|
|
|
c_conn = client.connect( "ws://localhost:" + fc::to_string(port) );
|
|
c_conn->on_message_handler([&](const std::string& s){
|
|
echo = s;
|
|
});
|
|
c_conn->send_message( "hello world" );
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_EQUAL("echo: hello world", echo);
|
|
}
|
|
|
|
BOOST_CHECK_THROW(c_conn->send_message( "again" ), fc::assert_exception);
|
|
BOOST_CHECK_THROW(client.connect( "ws://localhost:" + fc::to_string(port) ), fc::exception);
|
|
l.remove_appender(ca);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(websocket_test_with_proxy_header)
|
|
{
|
|
// set up logging
|
|
fc::shared_ptr<fc::console_appender> ca(new fc::console_appender);
|
|
fc::logger l = fc::logger::get("rpc");
|
|
l.add_appender( ca );
|
|
|
|
fc::http::websocket_client client;
|
|
// add the proxy header element
|
|
client.append_header("MyProxyHeaderKey", "MyServer:8080");
|
|
|
|
fc::http::websocket_connection_ptr s_conn, c_conn;
|
|
int port;
|
|
{
|
|
// the server will be on the lookout for the key in the header
|
|
fc::http::websocket_server server("MyProxyHeaderKey");
|
|
server.on_connection([&]( const fc::http::websocket_connection_ptr& c ){
|
|
s_conn = c;
|
|
c->on_message_handler([&](const std::string& s){
|
|
c->send_message("echo: " + s);
|
|
});
|
|
});
|
|
|
|
server.listen( 0 );
|
|
port = server.get_listening_port();
|
|
|
|
server.start_accept();
|
|
|
|
std::string echo;
|
|
c_conn = client.connect( "ws://localhost:" + fc::to_string(port) );
|
|
c_conn->on_message_handler([&](const std::string& s){
|
|
echo = s;
|
|
});
|
|
c_conn->send_message( "hello world" );
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_EQUAL("echo: hello world", echo);
|
|
c_conn->send_message( "again" );
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_EQUAL("echo: again", echo);
|
|
|
|
s_conn->close(0, "test");
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_THROW(c_conn->send_message( "again" ), fc::exception);
|
|
|
|
c_conn = client.connect( "ws://localhost:" + fc::to_string(port) );
|
|
c_conn->on_message_handler([&](const std::string& s){
|
|
echo = s;
|
|
});
|
|
c_conn->send_message( "hello world" );
|
|
fc::usleep( fc::milliseconds(100) );
|
|
BOOST_CHECK_EQUAL("echo: hello world", echo);
|
|
}
|
|
|
|
BOOST_CHECK_THROW(c_conn->send_message( "again" ), fc::assert_exception);
|
|
BOOST_CHECK_THROW(client.connect( "ws://localhost:" + fc::to_string(port) ), fc::exception);
|
|
l.remove_appender(ca);
|
|
}
|
|
|
|
BOOST_AUTO_TEST_SUITE_END()
|