Team2890/allwpilib
4
0
Fork 0
mirror of https://github.com/wpilibsuite/allwpilib synced 2026-06-23 01:21:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

[wpinet] WebSocket: Send pong in response to ping (#5498)

Browse Source 
This is required by the spec (RFC 6455 section 5.5.2).
This commit is contained in:
Peter Johnson
2023-08-04 16:27:08 -07:00
committed by GitHub
parent e2d17a24a6
commit c52dad609e
3 changed files with 56 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
wpinet/src/test/native/cpp/WebSocketIntegrationTest.cpp
View File

@@ -147,4 +147,45 @@ TEST_F(WebSocketIntegrationTest, ClientSendText) {
ASSERT_EQ(gotData, 1);
}
TEST_F(WebSocketIntegrationTest, ServerSendPing) {
int gotPing = 0;
int gotPong = 0;
serverPipe->Listen([&]() {
auto conn = serverPipe->Accept();
auto server = WebSocketServer::Create(*conn);
server->connected.connect([&](std::string_view, WebSocket& ws) {
ws.SendPing({uv::Buffer{"\x03\x04", 2}}, [&](auto, uv::Error) {});
ws.pong.connect([&](auto data) {
++gotPong;
std::vector<uint8_t> recvData{data.begin(), data.end()};
std::vector<uint8_t> expectData{0x03, 0x04};
ASSERT_EQ(recvData, expectData);
ws.Close();
});
});
});
clientPipe->Connect(pipeName, [&] {
auto ws = WebSocket::CreateClient(*clientPipe, "/test", pipeName);
ws->closed.connect([&](uint16_t code, std::string_view reason) {
Finish();
if (code != 1005 && code != 1006) {
FAIL() << "Code: " << code << " Reason: " << reason;
}
});
ws->ping.connect([&](auto data) {
++gotPing;
std::vector<uint8_t> recvData{data.begin(), data.end()};
std::vector<uint8_t> expectData{0x03, 0x04};
ASSERT_EQ(recvData, expectData);
});
});
loop->Run();
ASSERT_EQ(gotPing, 1);
ASSERT_EQ(gotPong, 1);
}
} // namespace wpi