111 lines
3.1 KiB
C++
111 lines
3.1 KiB
C++
/*
|
|
* IXWebSocketServerTest.cpp
|
|
* Author: Benjamin Sergeant
|
|
* Copyright (c) 2019 Machine Zone. All rights reserved.
|
|
*/
|
|
|
|
#include "IXTest.h"
|
|
#include "catch.hpp"
|
|
#include <iostream>
|
|
#include <ixwebsocket/IXSocket.h>
|
|
#include <ixwebsocket/IXSocketFactory.h>
|
|
#include <ixwebsocket/IXWebSocket.h>
|
|
#include <ixwebsocket/IXWebSocketServer.h>
|
|
|
|
using namespace ix;
|
|
|
|
bool startServer(ix::WebSocketServer& server, std::string& subProtocols)
|
|
{
|
|
server.setOnClientMessageCallback(
|
|
[&server, &subProtocols](std::shared_ptr<ConnectionState> connectionState,
|
|
ConnectionInfo& connectionInfo,
|
|
WebSocket& webSocket,
|
|
const ix::WebSocketMessagePtr& msg) {
|
|
auto remoteIp = connectionInfo.remoteIp;
|
|
if (msg->type == ix::WebSocketMessageType::Open)
|
|
{
|
|
TLogger() << "New connection";
|
|
TLogger() << "remote ip: " << remoteIp;
|
|
TLogger() << "id: " << connectionState->getId();
|
|
TLogger() << "Uri: " << msg->openInfo.uri;
|
|
TLogger() << "Headers:";
|
|
for (auto it : msg->openInfo.headers)
|
|
{
|
|
TLogger() << it.first << ": " << it.second;
|
|
}
|
|
|
|
subProtocols = msg->openInfo.headers["Sec-WebSocket-Protocol"];
|
|
}
|
|
else if (msg->type == ix::WebSocketMessageType::Close)
|
|
{
|
|
log("Closed connection");
|
|
}
|
|
else if (msg->type == ix::WebSocketMessageType::Message)
|
|
{
|
|
for (auto&& client : server.getClients())
|
|
{
|
|
if (client.get() != &webSocket)
|
|
{
|
|
client->sendBinary(msg->str);
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
auto res = server.listen();
|
|
if (!res.first)
|
|
{
|
|
log(res.second);
|
|
return false;
|
|
}
|
|
|
|
server.start();
|
|
return true;
|
|
}
|
|
|
|
TEST_CASE("subprotocol", "[websocket_subprotocol]")
|
|
{
|
|
SECTION("Connect to the server")
|
|
{
|
|
int port = getFreePort();
|
|
ix::WebSocketServer server(port);
|
|
|
|
std::string subProtocols;
|
|
startServer(server, subProtocols);
|
|
|
|
std::atomic<bool> connected(false);
|
|
ix::WebSocket webSocket;
|
|
webSocket.setOnMessageCallback([&connected](const ix::WebSocketMessagePtr& msg) {
|
|
if (msg->type == ix::WebSocketMessageType::Open)
|
|
{
|
|
connected = true;
|
|
log("Client connected");
|
|
}
|
|
});
|
|
|
|
webSocket.addSubProtocol("json");
|
|
webSocket.addSubProtocol("msgpack");
|
|
|
|
std::string url;
|
|
std::stringstream ss;
|
|
ss << "ws://127.0.0.1:" << port;
|
|
url = ss.str();
|
|
|
|
webSocket.setUrl(url);
|
|
webSocket.start();
|
|
|
|
// Give us 3 seconds to connect
|
|
int attempts = 0;
|
|
while (!connected)
|
|
{
|
|
REQUIRE(attempts++ < 300);
|
|
ix::msleep(10);
|
|
}
|
|
|
|
webSocket.stop();
|
|
server.stop();
|
|
|
|
REQUIRE(subProtocols == "json,msgpack");
|
|
}
|
|
}
|