From 25e6b7038192efee03ae26423b746ea8e5a06e2c Mon Sep 17 00:00:00 2001
From: Peter Keresztes Schmidt <carbenium@outlook.com>
Date: Sat, 10 Apr 2021 17:18:45 +0200
Subject: [PATCH] tests/comms: Add tests for ZM::UdpUnixSocket

---
 tests/zm_comms.cpp | 60 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/tests/zm_comms.cpp b/tests/zm_comms.cpp
index 00882b4c1..622ecde72 100644
--- a/tests/zm_comms.cpp
+++ b/tests/zm_comms.cpp
@@ -185,3 +185,63 @@ TEST_CASE("ZM::UdpInetSocket send/recv") {
     REQUIRE(rcv == msg);
   }
 }
+
+TEST_CASE("ZM::UdpUnixSocket basics") {
+  std::string sock_path = "/tmp/zm.unittest.sock";
+  unlink(sock_path.c_str()); // make sure the socket file does not exist
+
+  ZM::UdpUnixSocket socket;
+  REQUIRE(socket.isClosed() == true);
+  REQUIRE(socket.isOpen() == false);
+  REQUIRE(socket.isConnected() == false);
+  REQUIRE(socket.isDisconnected() == false);
+
+  SECTION("bind") {
+    REQUIRE(socket.bind(sock_path.c_str()) == true);
+    REQUIRE(socket.isOpen() == true);
+    REQUIRE(socket.isDisconnected() == true);
+    REQUIRE(socket.isClosed() == false);
+    REQUIRE(socket.isConnected() == false);
+
+    SECTION("close") {
+      REQUIRE(socket.close() == true);
+      REQUIRE(socket.isClosed() == true);
+      REQUIRE(socket.isOpen() == false);
+      REQUIRE(socket.isConnected() == false);
+      REQUIRE(socket.isDisconnected() == false);
+    }
+  }
+
+  SECTION("connect to unbound socket") {
+    REQUIRE(socket.connect(sock_path.c_str()) == false);
+  }
+}
+
+TEST_CASE("ZM::UdpUnixSocket send/recv") {
+  std::string sock_path = "/tmp/zm.unittest.sock";
+  unlink(sock_path.c_str()); // make sure the socket file does not exist
+
+  ZM::UdpUnixSocket srv_socket;
+  ZM::UdpUnixSocket client_socket;
+
+  std::array<char, 3> msg = {'a', 'b', 'c'};
+  std::array<char, msg.size()> rcv{};
+
+  SECTION("send/recv on unbound socket") {
+    REQUIRE(client_socket.send(msg.data(), msg.size()) == -1);
+    REQUIRE(srv_socket.recv(rcv.data(), rcv.size()) == -1);
+  }
+
+  SECTION("send/recv") {
+    REQUIRE(srv_socket.bind(sock_path.c_str()) == true);
+    REQUIRE(srv_socket.isOpen() == true);
+
+    REQUIRE(client_socket.connect(sock_path.c_str()) == true);
+    REQUIRE(client_socket.isConnected() == true);
+
+    REQUIRE(client_socket.send(msg.data(), msg.size()) == msg.size());
+    REQUIRE(srv_socket.recv(rcv.data(), rcv.size()) == msg.size());
+
+    REQUIRE(rcv == msg);
+  }
+}