poller: Add unittest

3 files changed, 200 insertions, 0 deletions

diff --git a/meson.build b/meson.build
index 33eaa98..fa000c1 100644
--- a/meson.build
+++ b/meson.build
@@ -183,6 +183,14 @@ test(
                 include_directories: [include_directories('.', 'include')],
         )
 )
+test(
+        'poller',
+        executable(
+                'poller_test',
+                ['common/linked_list.c', 'seatd/poller.c', 'tests/poller.c'],
+                include_directories: [include_directories('.', 'include')],
+        )
+)
 
 scdoc = dependency('scdoc', required: get_option('man-pages'), version: '>= 1.9.7', native: true)
 
diff --git a/seatd/poller.c b/seatd/poller.c
index 1d1b9b4..267929b 100644
--- a/seatd/poller.c
+++ b/seatd/poller.c
@@ -111,6 +111,7 @@ int poller_finish(struct poller *poller) {
 		free(bps);
 	}
 	free(poller->pollfds);
+	global_poller = NULL;
 	return 0;
 }
 
diff --git a/tests/poller.c b/tests/poller.c
new file mode 100644
index 0000000..382d9d2
--- /dev/null
+++ b/tests/poller.c
@@ -0,0 +1,191 @@
+#include <signal.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/types.h>
+#include <unistd.h>
+
+#include "poller.h"
+#include "test.h"
+
+static void test_poller_init(void) {
+	struct poller poller;
+	test_assert(poller_init(&poller) == 0);
+	poller_finish(&poller);
+}
+
+struct test_fd {
+	int fd;
+	uint32_t events;
+};
+
+static int test_fd_event(int fd, uint32_t mask, void *data) {
+	struct test_fd *d = data;
+	d->fd = fd;
+	d->events = mask;
+	return 0;
+}
+
+static void test_poller_single_fd(void) {
+	struct poller poller;
+	test_assert(poller_init(&poller) == 0);
+
+	int fds[2];
+	test_assert(pipe(fds) == 0);
+
+	struct test_fd evd;
+	struct event_source_fd *ev =
+		poller_add_fd(&poller, fds[0], EVENT_READABLE, test_fd_event, &evd);
+	test_assert(ev != NULL);
+
+	evd.fd = 0;
+	evd.events = 0;
+	test_assert(write(fds[1], "\0", 1) == 1);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(evd.fd == fds[0]);
+	test_assert(evd.events == EVENT_READABLE);
+
+	evd.fd = 0;
+	evd.events = 0;
+	test_assert(write(fds[1], "\0", 1) == 1);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(evd.fd == fds[0]);
+	test_assert(evd.events == EVENT_READABLE);
+
+	close(fds[0]);
+	close(fds[1]);
+	poller_finish(&poller);
+}
+
+static void test_poller_multi_fd(void) {
+	struct poller poller;
+	test_assert(poller_init(&poller) == 0);
+
+	char dummy[8];
+	int fdsa[2], fdsb[2];
+	test_assert(pipe(fdsa) == 0);
+	test_assert(pipe(fdsb) == 0);
+
+	struct test_fd evd1, evd2;
+	struct event_source_fd *ev1 =
+		poller_add_fd(&poller, fdsa[0], EVENT_READABLE, test_fd_event, &evd1);
+	struct event_source_fd *ev2 =
+		poller_add_fd(&poller, fdsb[0], EVENT_READABLE, test_fd_event, &evd2);
+	test_assert(ev1 != NULL);
+	test_assert(ev2 != NULL);
+
+	evd1.fd = evd2.fd = 0;
+	evd1.events = evd2.events = 0;
+	test_assert(write(fdsa[1], "\0", 1) == 1);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(read(fdsa[0], &dummy, sizeof dummy) == 1);
+	test_assert(evd1.fd == fdsa[0]);
+	test_assert(evd1.events == EVENT_READABLE);
+	test_assert(evd2.fd == 0);
+	test_assert(evd2.events == 0);
+
+	evd1.fd = evd2.fd = 0;
+	evd1.events = evd2.events = 0;
+	test_assert(write(fdsb[1], "\0", 1) == 1);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(read(fdsb[0], &dummy, sizeof dummy) == 1);
+	test_assert(evd1.fd == 0);
+	test_assert(evd1.events == 0);
+	test_assert(evd2.fd == fdsb[0]);
+	test_assert(evd2.events == EVENT_READABLE);
+
+	evd1.fd = evd2.fd = 0;
+	evd1.events = evd2.events = 0;
+	test_assert(write(fdsa[1], "\0", 1) == 1);
+	test_assert(write(fdsb[1], "\0", 1) == 1);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(read(fdsa[0], &dummy, sizeof dummy) == 1);
+	test_assert(read(fdsb[0], &dummy, sizeof dummy) == 1);
+	test_assert(evd1.fd == fdsa[0]);
+	test_assert(evd1.events == EVENT_READABLE);
+	test_assert(evd2.fd == fdsb[0]);
+	test_assert(evd2.events == EVENT_READABLE);
+
+	close(fdsa[0]);
+	close(fdsa[1]);
+	close(fdsb[0]);
+	close(fdsb[1]);
+	poller_finish(&poller);
+}
+
+struct test_signal {
+	int signal;
+};
+
+static int test_signal_event(int signal, void *data) {
+	struct test_signal *d = data;
+	d->signal = signal;
+	return 0;
+}
+
+static void test_poller_single_signal(void) {
+	struct poller poller;
+	test_assert(poller_init(&poller) == 0);
+
+	struct test_signal evd;
+	struct event_source_signal *ev =
+		poller_add_signal(&poller, SIGRTMIN, test_signal_event, &evd);
+	test_assert(ev != NULL);
+
+	evd.signal = 0;
+	test_assert(kill(getpid(), SIGRTMIN) == 0);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(evd.signal == SIGRTMIN);
+
+	evd.signal = 0;
+	test_assert(kill(getpid(), SIGRTMIN) == 0);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(evd.signal == SIGRTMIN);
+
+	poller_finish(&poller);
+}
+
+static void test_poller_multi_signal(void) {
+	struct poller poller;
+	test_assert(poller_init(&poller) == 0);
+
+	struct test_signal evd1, evd2;
+	struct event_source_signal *ev1 =
+		poller_add_signal(&poller, SIGRTMIN, test_signal_event, &evd1);
+	struct event_source_signal *ev2 =
+		poller_add_signal(&poller, SIGRTMIN + 1, test_signal_event, &evd2);
+	test_assert(ev1 != NULL);
+	test_assert(ev2 != NULL);
+
+	evd1.signal = evd2.signal = 0;
+	test_assert(kill(getpid(), SIGRTMIN) == 0);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(evd1.signal == SIGRTMIN);
+	test_assert(evd2.signal == 0);
+
+	evd1.signal = evd2.signal = 0;
+	test_assert(kill(getpid(), SIGRTMIN + 1) == 0);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(evd1.signal == 0);
+	test_assert(evd2.signal == SIGRTMIN + 1);
+
+	evd1.signal = evd2.signal = 0;
+	test_assert(kill(getpid(), SIGRTMIN) == 0);
+	test_assert(kill(getpid(), SIGRTMIN + 1) == 0);
+	test_assert(poller_poll(&poller) == 0);
+	test_assert(evd1.signal == SIGRTMIN);
+	test_assert(evd2.signal == SIGRTMIN + 1);
+
+	poller_finish(&poller);
+}
+
+int main(int argc, char *argv[]) {
+	(void)argc;
+	(void)argv;
+
+	test_run(test_poller_init);
+	test_run(test_poller_single_fd);
+	test_run(test_poller_multi_fd);
+	test_run(test_poller_single_signal);
+	test_run(test_poller_multi_signal);
+}