Dark-Alex-17
11524bb232
Check / stable / fmt (push) Failing after 25s
Check / beta / clippy (push) Failing after 1m15s
Check / stable / clippy (push) Failing after 1m12s
Check / nightly / doc (push) Successful in 1m2s
Check / 1.95.0 / check (push) Successful in 1m8s
Check / dockerfile / msrv (push) Successful in 5s
Test Suite / ubuntu / beta (push) Successful in 2m21s
Test Suite / ubuntu / stable (push) Successful in 2m20s
Test Suite / ubuntu / stable / coverage (push) Failing after 1m59s
Test Suite / macos-latest / stable (push) Canceled after 0s
Test Suite / windows-latest / stable (push) Canceled after 0s
313 lines
8.9 KiB
Rust
313 lines
8.9 KiB
Rust
use std::{
|
|
fs,
|
|
path::{Path, PathBuf},
|
|
process, str,
|
|
sync::atomic::{AtomicU64, Ordering},
|
|
};
|
|
|
|
use assert_cmd::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);
|
|
}
|
|
}
|
|
|
|
#[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 = cargo_bin_cmd!("managarr");
|
|
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 = cargo_bin_cmd!("managarr");
|
|
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 = cargo_bin_cmd!("managarr");
|
|
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 = cargo_bin_cmd!("managarr");
|
|
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 = cargo_bin_cmd!("managarr");
|
|
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 = cargo_bin_cmd!("managarr");
|
|
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();
|
|
}
|