managarr/tests/servarr_name_cli_integration_tests.rs

use std::{
  fs,
  path::{Path, PathBuf},
  process, str,
  sync::atomic::{AtomicU64, Ordering},
};

use assert_cmd::{Command, cargo::cargo_bin_cmd};
use mockito::Server;

static NEXT_CONFIG_ID: AtomicU64 = AtomicU64::new(0);

struct TemporaryConfig {
  path: PathBuf,
}

impl TemporaryConfig {
  fn new(contents: &str) -> Self {
    let path = std::env::temp_dir().join(format!(
      "managarr-servarr-name-{}-{}.yaml",
      process::id(),
      NEXT_CONFIG_ID.fetch_add(1, Ordering::Relaxed)
    ));
    fs::write(&path, contents).unwrap();
    Self { path }
  }

  fn path(&self) -> &Path {
    &self.path
  }
}

impl Drop for TemporaryConfig {
  fn drop(&mut self) {
    let _ = fs::remove_file(&self.path);
  }
}

fn managarr_command() -> Command {
  let home = std::env::temp_dir().join(format!("managarr-servarr-name-home-{}", process::id()));
  fs::create_dir_all(&home).unwrap();
  let mut command = cargo_bin_cmd!("managarr");
  command.env("HOME", &home).env("XDG_CACHE_HOME", &home);
  command
}

#[test]
fn named_instance_of_different_servarr_type_rejects_command() {
  let mut radarr_server = Server::new();
  let radarr_mock = radarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"radarr","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(0)
    .create();
  let mut sonarr_server = Server::new();
  let sonarr_mock = sonarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"sonarr","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(0)
    .create();
  let config = TemporaryConfig::new(&format!(
    "radarr:\n  - name: Movies\n    uri: {}\n    api_token: test-token\nsonarr:\n  - name: Shows\n    uri: {}\n    api_token: test-token\n",
    radarr_server.url(),
    sonarr_server.url()
  ));

  let mut command = managarr_command();
  let assertion = command
    .arg("--config-file")
    .arg(config.path())
    .args([
      "--disable-spinner",
      "--servarr-name",
      "Shows",
      "radarr",
      "get",
      "system-status",
    ])
    .assert()
    .failure();
  let stderr = str::from_utf8(&assertion.get_output().stderr).unwrap();

  assert!(
    stderr.contains("Shows"),
    "CLI error did not identify the rejected instance: {stderr}"
  );
  radarr_mock.assert();
  sonarr_mock.assert();
}

#[test]
fn named_readarr_instance_selected_for_readarr_command() {
  let mut radarr_server = Server::new();
  let radarr_mock = radarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"wrong-instance","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(0)
    .create();
  let mut readarr_server = Server::new();
  let readarr_mock = readarr_server
    .mock("GET", "/api/v1/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"expected-instance","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(1)
    .create();
  let config = TemporaryConfig::new(&format!(
    "radarr:\n  - name: Servarr\n    uri: {}\n    api_token: test-token\nreadarr:\n  - name: Servarr\n    uri: {}\n    api_token: test-token\n",
    radarr_server.url(),
    readarr_server.url()
  ));

  let mut command = managarr_command();
  let assertion = command
    .arg("--config-file")
    .arg(config.path())
    .args([
      "--disable-spinner",
      "--servarr-name",
      "Servarr",
      "readarr",
      "get",
      "system-status",
    ])
    .assert()
    .success();
  let stdout = str::from_utf8(&assertion.get_output().stdout).unwrap();

  assert!(
    stdout.contains("\"version\": \"expected-instance\""),
    "unexpected CLI output: {stdout}"
  );
  assert!(!stdout.contains("wrong-instance"));
  readarr_mock.assert();
  radarr_mock.assert();
}

#[test]
fn same_named_servarrs_select_command_matching_instance() {
  let mut radarr_server = Server::new();
  let radarr_mock = radarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"wrong-instance","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(0)
    .create();
  let mut sonarr_server = Server::new();
  let sonarr_mock = sonarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"expected-instance","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(1)
    .create();
  let config = TemporaryConfig::new(&format!(
    "radarr:\n  - name: Servarr\n    uri: {}\n    api_token: test-token\nsonarr:\n  - name: Servarr\n    uri: {}\n    api_token: test-token\n",
    radarr_server.url(),
    sonarr_server.url()
  ));

  let mut command = managarr_command();
  let assertion = command
    .arg("--config-file")
    .arg(config.path())
    .args([
      "--disable-spinner",
      "--servarr-name",
      "Servarr",
      "sonarr",
      "get",
      "system-status",
    ])
    .assert()
    .success();
  let stdout = str::from_utf8(&assertion.get_output().stdout).unwrap();

  assert!(
    stdout.contains("\"version\": \"expected-instance\""),
    "unexpected CLI output: {stdout}"
  );
  assert!(!stdout.contains("wrong-instance"));
  sonarr_mock.assert();
  radarr_mock.assert();
}

#[test]
fn unknown_servarr_name_fails() {
  let mut radarr_server = Server::new();
  let radarr_mock = radarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"radarr","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(0)
    .create();

  let config = TemporaryConfig::new(&format!(
    "radarr:\n  - name: Movies\n    uri: {}\n    api_token: test-token\n",
    radarr_server.url()
  ));
  let mut command = managarr_command();
  let assertion = command
    .arg("--config-file")
    .arg(config.path())
    .args([
      "--servarr-name",
      "Unknown",
      "radarr",
      "get",
      "system-status",
    ])
    .assert()
    .failure();

  let stderr = str::from_utf8(&assertion.get_output().stderr).unwrap();

  assert!(
    stderr.contains("A Servarr titled 'Unknown'"),
    "CLI error did not identify the unknown instance: {stderr}"
  );
  radarr_mock.assert();
}

#[test]
fn whitespace_servarr_name_trimmed() {
  let mut radarr_server = Server::new();
  let radarr_mock = radarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"radarr-whitespace","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(1)
    .create();

  let config = TemporaryConfig::new(&format!(
    "radarr:\n  - name: Movies\n    uri: {}\n    api_token: test-token\n",
    radarr_server.url()
  ));
  let mut command = managarr_command();
  let assertion = command
    .arg("--config-file")
    .arg(config.path())
    .args([
      "--disable-spinner",
      "--servarr-name",
      "  Movies  ",
      "radarr",
      "get",
      "system-status",
    ])
    .assert()
    .success();

  let stdout = str::from_utf8(&assertion.get_output().stdout).unwrap();

  assert!(
    stdout.contains("\"version\": \"radarr-whitespace\""),
    "unexpected CLI output: {stdout}"
  );
  radarr_mock.assert();
}

#[test]
fn no_servarr_name_uses_first_configured_instance() {
  let mut radarr_server = Server::new();
  let radarr_mock = radarr_server
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"expected-instance","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(1)
    .create();
  let mut radarr_server2 = Server::new();
  let radarr_mock2 = radarr_server2
    .mock("GET", "/api/v3/system/status")
    .match_header("X-Api-Key", "test-token")
    .with_status(200)
    .with_header("content-type", "application/json")
    .with_body(r#"{"version":"wrong-instance","startTime":"2024-01-01T00:00:00Z"}"#)
    .expect(0)
    .create();
  let config = TemporaryConfig::new(&format!(
    "radarr:\n  - name: Movies\n    uri: {}\n    api_token: test-token\n    weight: 2\n  - name: 4K-Movies\n    uri: {}\n    api_token: test-token\n    weight: 1\n",
    radarr_server.url(),
    radarr_server2.url()
  ));

  let mut command = managarr_command();
  let assertion = command
    .arg("--config-file")
    .arg(config.path())
    .args(["--disable-spinner", "radarr", "get", "system-status"])
    .assert()
    .success();
  let stdout = str::from_utf8(&assertion.get_output().stdout).unwrap();

  assert!(
    stdout.contains("\"version\": \"expected-instance\""),
    "unexpected CLI output: {stdout}"
  );
  assert!(!stdout.contains("wrong-instance"));
  radarr_mock.assert();
  radarr_mock2.assert();
}