From 5b40cf37079412e8a24c44e38bd54298a0533a29 Mon Sep 17 00:00:00 2001
From: ibizaman <ibizaman@tiserbox.com>
Date: Mon, 29 Sep 2025 20:57:13 +0200
Subject: [PATCH] tests: wait automatically on authelia and lldap to start

I had issues with race conditions happening where authelia didn't start yet
when the test script was already testing the integration.
---
 test/common.nix | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/test/common.nix b/test/common.nix
index ec4c606..e62f2d5 100644
--- a/test/common.nix
+++ b/test/common.nix
@@ -30,6 +30,9 @@ let
         node.config = nodes.server;
         inherit fqdn proto_fqdn;
       };
+
+      autheliaEnabled = (hasAttr "authelia" nodes.server.shb) && nodes.server.shb.authelia.enable;
+      lldapEnabled = (hasAttr "lldap" nodes.server.shb) && nodes.server.shb.lldap.enable;
     in
     ''
     import json
@@ -63,12 +66,13 @@ let
     ''
     + lib.strings.concatMapStrings (s: ''server.wait_for_unit("${s}")'' + "\n") (
       waitForServices args
-      ++ (lib.optionals redirectSSO [ "authelia-auth.${cfg.domain}.service" ])
+      ++ (lib.optionals autheliaEnabled [ "authelia-auth.${cfg.domain}.service" ])
+      ++ (lib.optionals lldapEnabled [ "lldap.service" ])
     )
     + lib.strings.concatMapStrings (p: ''server.wait_for_open_port(${toString p})'' + "\n") (
       waitForPorts args
       # TODO: when the SSO block exists, replace this hardcoded port.
-      ++ (lib.optionals redirectSSO [ 9091 /* nodes.server.services.authelia.instances."auth.${domain}".settings.server.port */ ] )
+      ++ (lib.optionals autheliaEnabled [ 9091 /* nodes.server.services.authelia.instances."auth.${domain}".settings.server.port */ ])
     )
     + lib.strings.concatMapStrings (u: ''server.wait_for_open_unix_socket("${u}")'' + "\n") (waitForUnixSocket args)
     + ''