use std::sync::Arc;
use crate::options::GazelleClientOptions;
use crate::GazelleClient;
use log::info;
use rogue_config::{OptionsProvider, YamlOptionsProvider};
use rogue_logging::{Error, Logger, LoggerBuilder};
use rogue_logging::Verbosity::Trace;
use serde::{Deserialize, Serialize};
#[derive(Deserialize, Serialize)]
struct ExampleValues {
pub torrent: u32,
pub group: u32,
pub user: u32,
}
fn get_options() -> Result<Vec<(GazelleClientOptions, ExampleValues)>, Error> {
let clients: Vec<GazelleClientOptions> = YamlOptionsProvider::get()?;
let examples: Vec<ExampleValues> = YamlOptionsProvider::get()?;
let vec = clients.into_iter().zip(examples).collect();
Ok(vec)
}
fn init_logger() -> Arc<Logger> {
LoggerBuilder::new()
.with_exclude_filter("reqwest".to_owned())
.with_exclude_filter("cookie".to_owned())
.with_verbosity(Trace)
.create()
}
#[tokio::test]
async fn get_torrent() -> Result<(), Error> {
init_logger();
for (options, example) in get_options()? {
println!("Indexer: {}", options.name);
let mut client = GazelleClient::from_options(options);
let response = client.get_torrent(example.torrent).await?;
assert_eq!(response.torrent.id, example.torrent);
}
Ok(())
}
#[tokio::test]
#[allow(clippy::panic)]
async fn get_torrent_invalid() -> Result<(), Error> {
init_logger();
let id = u32::MAX;
let options: Vec<GazelleClientOptions> = YamlOptionsProvider::get()?;
for options in options {
info!("Indexer: {}", options.name);
let mut client = GazelleClient::from_options(options.clone());
let response = client.get_torrent(id).await;
match response {
Ok(_) => panic!("should be an error"),
Err(e) => {
assert_eq!(e.action, "get torrent");
if options.name == *"red" {
assert_eq!(e.status_code, Some(400));
assert_eq!(e.message, "bad id parameter".to_owned());
} else {
assert_eq!(e.status_code, Some(200));
assert_eq!(e.message, "bad parameters".to_owned());
}
}
}
}
Ok(())
}
#[tokio::test]
async fn get_torrent_group() -> Result<(), Error> {
init_logger();
for (options, example) in get_options()? {
info!("Indexer: {}", options.name);
let mut client = GazelleClient::from_options(options);
let response = client.get_torrent_group(example.group).await?;
assert_eq!(response.group.id, example.group);
}
Ok(())
}
#[tokio::test]
#[allow(clippy::panic)]
async fn get_torrent_group_invalid() -> Result<(), Error> {
init_logger();
let id = u32::MAX;
for (options, _example) in get_options()? {
info!("Indexer: {}", options.name);
let mut client = GazelleClient::from_options(options.clone());
let response = client.get_torrent_group(id).await;
match response {
Ok(_) => panic!("should be an error"),
Err(e) => {
assert_eq!(e.action, "get torrent group");
if options.name == *"red" {
assert_eq!(e.status_code, Some(400));
assert_eq!(e.message, "bad id parameter".to_owned());
} else {
assert_eq!(e.status_code, Some(200));
assert_eq!(e.message, "bad parameters".to_owned());
}
}
}
}
Ok(())
}
#[tokio::test]
async fn get_user() -> Result<(), Error> {
init_logger();
for (options, example) in get_options()? {
println!("Indexer: {}", options.name);
let mut client = GazelleClient::from_options(options);
let user = client.get_user(example.user).await?;
assert!(!user.username.is_empty());
}
Ok(())
}