tests: Convert device test to NUTS

This also adds more tests for additional test cases (aio, and
more validations of incompatible device configurations).

2 files changed, 171 insertions, 0 deletions

diff --git a/src/sp/CMakeLists.txt b/src/sp/CMakeLists.txt
index 128aac15..7ccaf051 100644
--- a/src/sp/CMakeLists.txt
+++ b/src/sp/CMakeLists.txt
@@ -19,4 +19,5 @@ nng_sources(
 )
 
 nng_test(pipe_test)
+nng_test(device_test)
 nng_test(multistress_test)
diff --git a/src/sp/device_test.c b/src/sp/device_test.c
new file mode 100644
index 00000000..6d2ff1fb
--- /dev/null
+++ b/src/sp/device_test.c
@@ -0,0 +1,170 @@
+//
+// Copyright 2024 Staysail Systems, Inc. <info@staysail.tech>
+// Copyright 2018 Capitar IT Group BV <info@capitar.com>
+//
+// This software is supplied under the terms of the MIT License, a
+// copy of which should be located in the distribution where this
+// file was obtained (LICENSE.txt).  A copy of the license may also be
+// found online at https://opensource.org/licenses/MIT.
+//
+
+#include <string.h>
+
+#include <nng/nng.h>
+#include <nng/protocol/pair0/pair.h>
+#include <nng/protocol/pair1/pair.h>
+
+#include <nuts.h>
+
+struct dev_data {
+	nng_socket s1;
+	nng_socket s2;
+};
+
+void
+dodev(void *arg)
+{
+	struct dev_data *d = arg;
+
+	nng_device(d->s1, d->s2);
+}
+
+#define SECOND(x) ((x) *1000)
+
+void
+test_device_not_cooked(void)
+{
+	nng_socket cooked;
+	nng_socket raw;
+	NUTS_PASS(nng_pair1_open(&cooked));
+	NUTS_PASS(nng_pair1_open_raw(&raw));
+	NUTS_FAIL(nng_device(cooked, cooked), NNG_EINVAL);
+	NUTS_FAIL(nng_device(raw, cooked), NNG_EINVAL);
+	NUTS_FAIL(nng_device(cooked, raw), NNG_EINVAL);
+	NUTS_CLOSE(cooked);
+	NUTS_CLOSE(raw);
+}
+
+void
+test_device_incompatible(void)
+{
+	nng_socket s1;
+	nng_socket s2;
+	NUTS_PASS(nng_pair0_open_raw(&s1));
+	NUTS_PASS(nng_pair1_open_raw(&s2));
+	NUTS_FAIL(nng_device(s1, s2), NNG_EINVAL);
+	NUTS_CLOSE(s1);
+	NUTS_CLOSE(s2);
+}
+void
+test_device_forward(void)
+{
+	nng_thread     *thr;
+	struct dev_data d;
+	nng_duration    tmo = SECOND(1);
+	nng_socket      e1, e2;
+
+	// will be a pair variant
+	NUTS_PASS(nng_pair1_open_raw(&d.s1));
+	NUTS_PASS(nng_pair1_open_raw(&d.s2));
+	NUTS_PASS(nng_pair1_open(&e1));
+	NUTS_PASS(nng_pair1_open(&e2));
+	NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_RECVTIMEO, tmo));
+	NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_RECVTIMEO, tmo));
+	NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_SENDTIMEO, tmo));
+	NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_SENDTIMEO, tmo));
+
+	NUTS_MARRY(e1, d.s1);
+	NUTS_MARRY(e2, d.s2);
+	NUTS_PASS(nng_thread_create(&thr, dodev, &d));
+	nng_thread_set_name(thr, "device thread");
+
+	NUTS_SEND(e1, "ping");
+	NUTS_RECV(e2, "ping");
+	NUTS_SEND(e2, "pong");
+	NUTS_RECV(e1, "pong");
+
+	NUTS_CLOSE(e1);
+	NUTS_CLOSE(e2);
+	NUTS_CLOSE(d.s1);
+	NUTS_CLOSE(d.s2);
+	nng_thread_destroy(thr);
+}
+
+void
+test_device_reflect(void)
+{
+	nng_thread     *thr;
+	struct dev_data d;
+	nng_duration    tmo = SECOND(1);
+	nng_socket      e1;
+
+	// will be a pair variant
+	NUTS_PASS(nng_pair1_open_raw(&d.s1));
+	d.s2 = d.s1;
+	NUTS_PASS(nng_pair1_open(&e1));
+	NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_RECVTIMEO, tmo));
+	NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_SENDTIMEO, tmo));
+
+	NUTS_MARRY(e1, d.s1);
+	NUTS_PASS(nng_thread_create(&thr, dodev, &d));
+	nng_thread_set_name(thr, "device thread");
+
+	NUTS_SEND(e1, "ping");
+	NUTS_RECV(e1, "ping");
+	NUTS_SEND(e1, "pong");
+	NUTS_RECV(e1, "pong");
+
+	NUTS_CLOSE(e1);
+	NUTS_CLOSE(d.s1);
+	nng_thread_destroy(thr);
+}
+
+void
+test_device_aio(void)
+{
+	struct dev_data d;
+	nng_duration    tmo = SECOND(1);
+	nng_socket      e1, e2;
+	nng_aio        *aio;
+
+	// will be a pair variant
+	NUTS_PASS(nng_pair1_open_raw(&d.s1));
+	NUTS_PASS(nng_pair1_open_raw(&d.s2));
+	NUTS_PASS(nng_pair1_open(&e1));
+	NUTS_PASS(nng_pair1_open(&e2));
+	NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_RECVTIMEO, tmo));
+	NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_RECVTIMEO, tmo));
+	NUTS_PASS(nng_socket_set_ms(e1, NNG_OPT_SENDTIMEO, tmo));
+	NUTS_PASS(nng_socket_set_ms(e2, NNG_OPT_SENDTIMEO, tmo));
+
+	NUTS_MARRY(e1, d.s1);
+	NUTS_MARRY(e2, d.s2);
+	// cancellation of this aio is how we stop it
+	NUTS_PASS(nng_aio_alloc(&aio, NULL, NULL));
+	nng_device_aio(aio, d.s1, d.s2);
+
+	NUTS_SEND(e1, "ping");
+	NUTS_RECV(e2, "ping");
+	NUTS_SEND(e2, "pong");
+	NUTS_RECV(e1, "pong");
+
+	nng_aio_cancel(aio);
+	nng_aio_wait(aio);
+	NUTS_FAIL(nng_aio_result(aio), NNG_ECANCELED);
+	nng_aio_free(aio);
+
+	NUTS_CLOSE(e1);
+	NUTS_CLOSE(e2);
+	NUTS_CLOSE(d.s1);
+	NUTS_CLOSE(d.s2);
+}
+
+NUTS_TESTS = {
+	{ "device not cooked", test_device_not_cooked },
+	{ "device incompatible", test_device_incompatible },
+	{ "device forward", test_device_forward },
+	{ "device reflect", test_device_reflect },
+	{ "device aio", test_device_aio },
+	{ NULL, NULL },
+};