From 1a8b119e60158711ac53962f7a6ad478796f449a Mon Sep 17 00:00:00 2001
From: Austin Horstman <khaneliman12@gmail.com>
Date: Mon, 14 Jul 2025 22:00:11 -0500
Subject: [PATCH] tests/caffeine: add test coverage

- Add basic-service.nix test for caffeine service configuration
- Test systemd service generation with proper graphical-session.target
integration

Signed-off-by: Austin Horstman <khaneliman12@gmail.com>
---
 .../caffeine/basic-service-expected.service       | 13 +++++++++++++
 tests/modules/services/caffeine/basic-service.nix | 15 +++++++++++++++
 tests/modules/services/caffeine/default.nix       |  5 +++++
 3 files changed, 33 insertions(+)
 create mode 100644 tests/modules/services/caffeine/basic-service-expected.service
 create mode 100644 tests/modules/services/caffeine/basic-service.nix
 create mode 100644 tests/modules/services/caffeine/default.nix

diff --git a/tests/modules/services/caffeine/basic-service-expected.service b/tests/modules/services/caffeine/basic-service-expected.service
new file mode 100644
index 000000000..324f7e597
--- /dev/null
+++ b/tests/modules/services/caffeine/basic-service-expected.service
@@ -0,0 +1,13 @@
+[Install]
+WantedBy=graphical-session.target
+
+[Service]
+ExecStart=@caffeine-ng@/bin/caffeine
+PrivateTmp=true
+ProtectSystem=full
+Restart=on-failure
+Slice=session.slice
+Type=exec
+
+[Unit]
+Description=caffeine
diff --git a/tests/modules/services/caffeine/basic-service.nix b/tests/modules/services/caffeine/basic-service.nix
new file mode 100644
index 000000000..06ae0a4b0
--- /dev/null
+++ b/tests/modules/services/caffeine/basic-service.nix
@@ -0,0 +1,15 @@
+{ config, ... }:
+
+{
+  config = {
+    services.caffeine = {
+      enable = true;
+    };
+
+    nmt.script = ''
+      assertFileContent \
+        home-files/.config/systemd/user/caffeine.service \
+        ${./basic-service-expected.service}
+    '';
+  };
+}
diff --git a/tests/modules/services/caffeine/default.nix b/tests/modules/services/caffeine/default.nix
new file mode 100644
index 000000000..6dcd20420
--- /dev/null
+++ b/tests/modules/services/caffeine/default.nix
@@ -0,0 +1,5 @@
+{ lib, pkgs, ... }:
+
+lib.optionalAttrs pkgs.stdenv.hostPlatform.isLinux {
+  caffeine-basic-service = ./basic-service.nix;
+}