summaryrefslogtreecommitdiffstats
path: root/Utilities/cmcppdap/src/network_test.cpp
diff options
context:
space:
mode:
authorGlen Chung <kuchung@microsoft.com>2023-05-26 13:33:57 (GMT)
committerBrad King <brad.king@kitware.com>2023-05-26 13:36:03 (GMT)
commit3381e6bd5cbfa1694f07c97a19747adda10b06f2 (patch)
treea99c177695b41b4f6576f74a7a59a7721de96687 /Utilities/cmcppdap/src/network_test.cpp
parentbd58bc7817e9546cb6726a52657f54b1cde59b0c (diff)
parent1daeefc37856c209718b406dcb86d44519393dd1 (diff)
downloadCMake-3381e6bd5cbfa1694f07c97a19747adda10b06f2.zip
CMake-3381e6bd5cbfa1694f07c97a19747adda10b06f2.tar.gz
CMake-3381e6bd5cbfa1694f07c97a19747adda10b06f2.tar.bz2
Merge branch 'upstream-cppdap' into import-cppdap
* upstream-cppdap: cppdap 2023-05-26 (03cc1867)
Diffstat (limited to 'Utilities/cmcppdap/src/network_test.cpp')
-rw-r--r--Utilities/cmcppdap/src/network_test.cpp110
1 files changed, 110 insertions, 0 deletions
diff --git a/Utilities/cmcppdap/src/network_test.cpp b/Utilities/cmcppdap/src/network_test.cpp
new file mode 100644
index 0000000..57bb0a9
--- /dev/null
+++ b/Utilities/cmcppdap/src/network_test.cpp
@@ -0,0 +1,110 @@
+// Copyright 2019 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// https://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "dap/network.h"
+#include "dap/io.h"
+
+#include "chan.h"
+
+#include "gmock/gmock.h"
+#include "gtest/gtest.h"
+
+#include <chrono>
+#include <thread>
+
+namespace {
+
+constexpr int port = 19021;
+
+bool write(const std::shared_ptr<dap::Writer>& w, const std::string& s) {
+ return w->write(s.data(), s.size()) && w->write("\0", 1);
+}
+
+std::string read(const std::shared_ptr<dap::Reader>& r) {
+ char c;
+ std::string s;
+ while (r->read(&c, sizeof(c)) > 0) {
+ if (c == '\0') {
+ return s;
+ }
+ s += c;
+ }
+ return r->isOpen() ? "<read failed>" : "<stream closed>";
+}
+
+} // anonymous namespace
+
+TEST(Network, ClientServer) {
+ dap::Chan<bool> done;
+ auto server = dap::net::Server::create();
+ if (!server->start(
+ port,
+ [&](const std::shared_ptr<dap::ReaderWriter>& rw) {
+ ASSERT_EQ(read(rw), "client to server");
+ ASSERT_TRUE(write(rw, "server to client"));
+ done.put(true);
+ },
+ [&](const char* err) { FAIL() << "Server error: " << err; })) {
+ FAIL() << "Couldn't start server";
+ return;
+ }
+
+ for (int i = 0; i < 5; i++) {
+ auto client = dap::net::connect("localhost", port);
+ ASSERT_NE(client, nullptr) << "Failed to connect client " << i;
+ ASSERT_TRUE(write(client, "client to server"));
+ ASSERT_EQ(read(client), "server to client");
+ done.take();
+ std::this_thread::sleep_for(std::chrono::seconds(1));
+ }
+
+ server.reset();
+}
+
+TEST(Network, ServerRepeatStopAndRestart) {
+ dap::Chan<bool> done;
+ auto onConnect = [&](const std::shared_ptr<dap::ReaderWriter>& rw) {
+ ASSERT_EQ(read(rw), "client to server");
+ ASSERT_TRUE(write(rw, "server to client"));
+ done.put(true);
+ };
+ auto onError = [&](const char* err) { FAIL() << "Server error: " << err; };
+
+ auto server = dap::net::Server::create();
+ if (!server->start(port, onConnect, onError)) {
+ FAIL() << "Couldn't start server";
+ return;
+ }
+
+ server->stop();
+ server->stop();
+ server->stop();
+
+ if (!server->start(port, onConnect, onError)) {
+ FAIL() << "Couldn't restart server";
+ return;
+ }
+
+ auto client = dap::net::connect("localhost", port);
+ ASSERT_NE(client, nullptr) << "Failed to connect";
+ ASSERT_TRUE(write(client, "client to server"));
+ ASSERT_EQ(read(client), "server to client");
+ done.take();
+
+ server->stop();
+ server->stop();
+ server->stop();
+
+ server.reset();
+}