arr-init/tests/bazarr.nix

{
  pkgs,
  lib,
  self,
}:

pkgs.testers.runNixOSTest {
  name = "arr-init-bazarr";

  nodes = {
    # Test 1: Bazarr with Sonarr only
    sonarr_only =
      { pkgs, lib, ... }:
      {
        imports = [ self.nixosModules.default ];

        system.stateVersion = "24.11";

        virtualisation.memorySize = 4096;

        environment.systemPackages = with pkgs; [
          curl
          jq
          gnugrep
          gawk
        ];

        services.sonarr = {
          enable = true;
          dataDir = "/var/lib/sonarr/.config/NzbDrone";
          settings.server.port = lib.mkDefault 8989;
        };

        services.bazarr = {
          enable = true;
          listenPort = 6767;
        };

        services.bazarrInit = {
          enable = true;
          dataDir = "/var/lib/bazarr";
          port = 6767;
          sonarr = {
            enable = true;
            dataDir = "/var/lib/sonarr/.config/NzbDrone";
            port = 8989;
            serviceName = "sonarr";
          };
          radarr = {
            enable = false;
          };
        };
      };

    # Test 2: Bazarr with Radarr only
    radarr_only =
      { pkgs, lib, ... }:
      {
        imports = [ self.nixosModules.default ];

        system.stateVersion = "24.11";

        virtualisation.memorySize = 4096;

        environment.systemPackages = with pkgs; [
          curl
          jq
          gnugrep
          gawk
        ];

        services.radarr = {
          enable = true;
          dataDir = "/var/lib/radarr/.config/Radarr";
          settings.server.port = lib.mkDefault 7878;
        };

        services.bazarr = {
          enable = true;
          listenPort = 6767;
        };

        services.bazarrInit = {
          enable = true;
          dataDir = "/var/lib/bazarr";
          port = 6767;
          sonarr = {
            enable = false;
          };
          radarr = {
            enable = true;
            dataDir = "/var/lib/radarr/.config/Radarr";
            port = 7878;
            serviceName = "radarr";
          };
        };
      };
  };

  testScript = ''
    # Test 1: Sonarr only
    start_all()
    sonarr_only.wait_for_unit("sonarr.service")
    sonarr_only.wait_for_unit("bazarr.service")

    # Wait for Sonarr API
    sonarr_only.wait_until_succeeds(
        "API_KEY=$(grep -oP '(?<=<ApiKey>)[^<]+' /var/lib/sonarr/.config/NzbDrone/config.xml) && "
        "curl -sf http://localhost:8989/api/v3/system/status -H \"X-Api-Key: $API_KEY\"",
        timeout=120,
    )

    # Wait for Bazarr config.yaml
    sonarr_only.wait_until_succeeds(
        "test -f /var/lib/bazarr/config/config.yaml",
        timeout=120,
    )

    # Wait for Bazarr API
    sonarr_only.wait_until_succeeds(
        "API_KEY=$(awk '/^auth:/{f=1} f && /apikey:/{gsub(/.*apikey: /, \"\"); print; exit}' /var/lib/bazarr/config/config.yaml) && "
        "curl -sf http://localhost:6767/api/system/status -H \"X-API-KEY: $API_KEY\"",
        timeout=120,
    )

    # Trigger bazarr-init
    sonarr_only.succeed("systemctl restart bazarr-init.service")
    sonarr_only.wait_for_unit("bazarr-init.service")

    # Verify Sonarr provider is configured
    sonarr_only.succeed(
        "API_KEY=$(awk '/^auth:/{f=1} f && /apikey:/{gsub(/.*apikey: /, \"\"); print; exit}' /var/lib/bazarr/config/config.yaml) && "
        "curl -sf http://localhost:6767/api/system/settings -H \"X-API-KEY: $API_KEY\" | "
        "jq -e '.general.use_sonarr == true and (.sonarr.apikey // \"\") != \"\"'"
    )

    # Verify Radarr provider is NOT configured
    result = sonarr_only.succeed(
        "API_KEY=$(awk '/^auth:/{f=1} f && /apikey:/{gsub(/.*apikey: /, \"\"); print; exit}' /var/lib/bazarr/config/config.yaml) && "
        "curl -sf http://localhost:6767/api/system/settings -H \"X-API-KEY: $API_KEY\" | "
        "jq '.general.use_radarr'"
    ).strip()
    assert result == "false", f"Expected use_radarr to be false, got {result}"

    # Test 2: Radarr only
    start_all()
    radarr_only.wait_for_unit("radarr.service")
    radarr_only.wait_for_unit("bazarr.service")

    # Wait for Radarr API
    radarr_only.wait_until_succeeds(
        "API_KEY=$(grep -oP '(?<=<ApiKey>)[^<]+' /var/lib/radarr/.config/Radarr/config.xml) && "
        "curl -sf http://localhost:7878/api/v3/system/status -H \"X-Api-Key: $API_KEY\"",
        timeout=120,
    )

    # Wait for Bazarr config.yaml
    radarr_only.wait_until_succeeds(
        "test -f /var/lib/bazarr/config/config.yaml",
        timeout=120,
    )

    # Wait for Bazarr API
    radarr_only.wait_until_succeeds(
        "API_KEY=$(awk '/^auth:/{f=1} f && /apikey:/{gsub(/.*apikey: /, \"\"); print; exit}' /var/lib/bazarr/config/config.yaml) && "
        "curl -sf http://localhost:6767/api/system/status -H \"X-API-KEY: $API_KEY\"",
        timeout=120,
    )

    # Trigger bazarr-init
    radarr_only.succeed("systemctl restart bazarr-init.service")
    radarr_only.wait_for_unit("bazarr-init.service")

    # Verify Radarr provider is configured
    radarr_only.succeed(
        "API_KEY=$(awk '/^auth:/{f=1} f && /apikey:/{gsub(/.*apikey: /, \"\"); print; exit}' /var/lib/bazarr/config/config.yaml) && "
        "curl -sf http://localhost:6767/api/system/settings -H \"X-API-KEY: $API_KEY\" | "
        "jq -e '.general.use_radarr == true and (.radarr.apikey // \"\") != \"\"'"
    )

    # Verify Sonarr provider is NOT configured
    result = radarr_only.succeed(
        "API_KEY=$(awk '/^auth:/{f=1} f && /apikey:/{gsub(/.*apikey: /, \"\"); print; exit}' /var/lib/bazarr/config/config.yaml) && "
        "curl -sf http://localhost:6767/api/system/settings -H \"X-API-KEY: $API_KEY\" | "
        "jq '.general.use_sonarr'"
    ).strip()
    assert result == "false", f"Expected use_sonarr to be false, got {result}"
  '';
}