From 991ff35249305ef1df8ac9151827025f9d12c4d4 Mon Sep 17 00:00:00 2001
From: Lily Foster <lily@lily.flowers>
Date: Mon, 22 Aug 2022 11:01:54 -0400
Subject: [PATCH] udiskie: add tests

---
 tests/default.nix                          |  1 +
 tests/modules/services/udiskie/basic.nix   | 15 +++++++++++++++
 tests/modules/services/udiskie/basic.yml   |  4 ++++
 tests/modules/services/udiskie/default.nix |  4 ++++
 tests/modules/services/udiskie/no-tray.nix | 18 ++++++++++++++++++
 tests/modules/services/udiskie/no-tray.yml |  4 ++++
 6 files changed, 46 insertions(+)
 create mode 100644 tests/modules/services/udiskie/basic.nix
 create mode 100644 tests/modules/services/udiskie/basic.yml
 create mode 100644 tests/modules/services/udiskie/default.nix
 create mode 100644 tests/modules/services/udiskie/no-tray.nix
 create mode 100644 tests/modules/services/udiskie/no-tray.yml

diff --git a/tests/default.nix b/tests/default.nix
index 6adee16a0..18ec9ab12 100644
--- a/tests/default.nix
+++ b/tests/default.nix
@@ -182,6 +182,7 @@ import nmt {
     ./modules/services/syncthing
     ./modules/services/trayer
     ./modules/services/twmn
+    ./modules/services/udiskie
     ./modules/services/window-managers/bspwm
     ./modules/services/window-managers/herbstluftwm
     ./modules/services/window-managers/i3
diff --git a/tests/modules/services/udiskie/basic.nix b/tests/modules/services/udiskie/basic.nix
new file mode 100644
index 000000000..5fa623c3e
--- /dev/null
+++ b/tests/modules/services/udiskie/basic.nix
@@ -0,0 +1,15 @@
+{
+  config = {
+    services.udiskie.enable = true;
+
+    test.stubs.udiskie = { };
+
+    nmt.script = ''
+      serviceFile="home-files/.config/systemd/user/udiskie.service"
+      assertFileRegex "$serviceFile" 'After=tray\.target'
+      assertFileRegex "$serviceFile" 'Requires=tray\.target'
+      assertFileContent "home-files/.config/udiskie/config.yml" \
+          ${./basic.yml}
+    '';
+  };
+}
diff --git a/tests/modules/services/udiskie/basic.yml b/tests/modules/services/udiskie/basic.yml
new file mode 100644
index 000000000..423f4ec82
--- /dev/null
+++ b/tests/modules/services/udiskie/basic.yml
@@ -0,0 +1,4 @@
+program_options:
+  automount: true
+  notify: true
+  tray: auto
diff --git a/tests/modules/services/udiskie/default.nix b/tests/modules/services/udiskie/default.nix
new file mode 100644
index 000000000..b79be5a9c
--- /dev/null
+++ b/tests/modules/services/udiskie/default.nix
@@ -0,0 +1,4 @@
+{
+  udiskie-basic = ./basic.nix;
+  udiskie-no-tray = ./no-tray.nix;
+}
diff --git a/tests/modules/services/udiskie/no-tray.nix b/tests/modules/services/udiskie/no-tray.nix
new file mode 100644
index 000000000..0c7ede0bd
--- /dev/null
+++ b/tests/modules/services/udiskie/no-tray.nix
@@ -0,0 +1,18 @@
+{
+  config = {
+    services.udiskie = {
+      enable = true;
+      tray = "never";
+    };
+
+    test.stubs.udiskie = { };
+
+    nmt.script = ''
+      serviceFile="home-files/.config/systemd/user/udiskie.service"
+      assertFileNotRegex "$serviceFile" 'After=tray\.target'
+      assertFileNotRegex "$serviceFile" 'Requires=tray\.target'
+      assertFileContent "home-files/.config/udiskie/config.yml" \
+          ${./no-tray.yml}
+    '';
+  };
+}
diff --git a/tests/modules/services/udiskie/no-tray.yml b/tests/modules/services/udiskie/no-tray.yml
new file mode 100644
index 000000000..75eca2ff8
--- /dev/null
+++ b/tests/modules/services/udiskie/no-tray.yml
@@ -0,0 +1,4 @@
+program_options:
+  automount: true
+  notify: true
+  tray: false