1 files changed, 43 insertions, 0 deletions
diff --git a/system/enablement_test.go b/system/enablement_test.go
new file mode 100644
index 00000000..f4f362d2
--- /dev/null
+++ b/system/enablement_test.go
@@ -0,0 +1,43 @@
+package system_test
+
+import (
+	"testing"
+
+	"git.gensokyo.uk/security/fortify/system"
+)
+
+func TestEnablementString(t *testing.T) {
+	testCases := []struct {
+		flags system.Enablement
+		want  string
+	}{
+		{0, "(no enablements)"},
+		{system.EWayland, "wayland"},
+		{system.EX11, "x11"},
+		{system.EDBus, "dbus"},
+		{system.EPulse, "pulseaudio"},
+		{system.EWayland | system.EX11, "wayland, x11"},
+		{system.EWayland | system.EDBus, "wayland, dbus"},
+		{system.EWayland | system.EPulse, "wayland, pulseaudio"},
+		{system.EX11 | system.EDBus, "x11, dbus"},
+		{system.EX11 | system.EPulse, "x11, pulseaudio"},
+		{system.EDBus | system.EPulse, "dbus, pulseaudio"},
+		{system.EWayland | system.EX11 | system.EDBus, "wayland, x11, dbus"},
+		{system.EWayland | system.EX11 | system.EPulse, "wayland, x11, pulseaudio"},
+		{system.EWayland | system.EDBus | system.EPulse, "wayland, dbus, pulseaudio"},
+		{system.EX11 | system.EDBus | system.EPulse, "x11, dbus, pulseaudio"},
+		{system.EWayland | system.EX11 | system.EDBus | system.EPulse, "wayland, x11, dbus, pulseaudio"},
+
+		{1 << 5, "e20"},
+		{1 << 6, "e40"},
+		{1 << 7, "e80"},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.want, func(t *testing.T) {
+			if got := tc.flags.String(); got != tc.want {
+				t.Errorf("String: %q, want %q", got, tc.want)
+			}
+		})
+	}
+}