y6 follow-up: tolerate unsupported epoll_pwait2 kernels

1 files changed, 33 insertions, 0 deletions

diff --git a/cmd/ioworkload/scenario_polling_test.go b/cmd/ioworkload/scenario_polling_test.go
new file mode 100644
index 0000000..83e6d7c
--- /dev/null
+++ b/cmd/ioworkload/scenario_polling_test.go
@@ -0,0 +1,33 @@
+package main
+
+import (
+	"fmt"
+	"syscall"
+	"testing"
+)
+
+func TestIsUnsupportedEpollPwait2Err(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name string
+		err  error
+		want bool
+	}{
+		{name: "nil", err: nil, want: false},
+		{name: "enosys", err: fmt.Errorf("epoll_pwait2: %w", syscall.ENOSYS), want: true},
+		{name: "enotsup", err: fmt.Errorf("epoll_pwait2: %w", syscall.ENOTSUP), want: true},
+		{name: "einval", err: fmt.Errorf("epoll_pwait2: %w", syscall.EINVAL), want: false},
+		{name: "opaque", err: fmt.Errorf("epoll_pwait2: not-supported"), want: false},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+			got := isUnsupportedEpollPwait2Err(tt.err)
+			if got != tt.want {
+				t.Fatalf("isUnsupportedEpollPwait2Err(%v) = %v, want %v", tt.err, got, tt.want)
+			}
+		})
+	}
+}