use std::{ collections::HashSet, fs, path::PathBuf, sync::{Arc, Mutex}, }; use once_cell::sync::Lazy; use musichoard::library::{ beets::{executor::BeetsLibraryProcessExecutor, BeetsLibrary}, Field, ILibrary, Item, Query, }; use crate::library::testmod::LIBRARY_ITEMS; pub static BEETS_TEST_CONFIG_PATH: Lazy = Lazy::new(|| fs::canonicalize("./tests/files/library/config.yml").unwrap()); pub static BEETS_EMPTY_CONFIG: Lazy>>> = Lazy::new(|| { Arc::new(Mutex::new(BeetsLibrary::new( BeetsLibraryProcessExecutor::default(), ))) }); pub static BEETS_TEST_CONFIG: Lazy>>> = Lazy::new(|| { Arc::new(Mutex::new(BeetsLibrary::new( BeetsLibraryProcessExecutor::default().config(Some(&*BEETS_TEST_CONFIG_PATH)), ))) }); #[test] fn test_no_config_list() { let beets_arc = BEETS_EMPTY_CONFIG.clone(); let beets = &mut beets_arc.lock().unwrap(); let output = beets.list(&Query::new()).unwrap(); let expected: Vec = vec![]; assert_eq!(output, expected); } #[test] fn test_invalid_config() { let mut beets = BeetsLibrary::new(BeetsLibraryProcessExecutor::default().config(Some( &PathBuf::from("./tests/files/library/config-does-not-exist.yml"), ))); let result = beets.list(&Query::new()); assert!(result.is_err()); assert!(!result.unwrap_err().to_string().is_empty()); } #[test] fn test_full_list() { let beets_arc = BEETS_TEST_CONFIG.clone(); let beets = &mut beets_arc.lock().unwrap(); let output = beets.list(&Query::new()).unwrap(); let expected: Vec = LIBRARY_ITEMS.to_owned(); let output: HashSet<_> = output.iter().collect(); let expected: HashSet<_> = expected.iter().collect(); assert_eq!(output, expected); } #[test] fn test_album_artist_query() { let beets_arc = BEETS_TEST_CONFIG.clone(); let beets = &mut beets_arc.lock().unwrap(); let output = beets .list(Query::new().include(Field::AlbumArtist(String::from("Аркона")))) .unwrap(); let expected: Vec = LIBRARY_ITEMS .iter() .filter(|it| it.album_artist == "Аркона") .cloned() .collect(); assert_eq!(output, expected); } #[test] fn test_album_title_query() { let beets_arc = BEETS_TEST_CONFIG.clone(); let beets = &mut beets_arc.lock().unwrap(); let output = beets .list(Query::new().include(Field::AlbumTitle(String::from("Slovo")))) .unwrap(); let expected: Vec = LIBRARY_ITEMS .iter() .filter(|it| it.album_title == "Slovo") .cloned() .collect(); assert_eq!(output, expected); } #[test] fn test_exclude_query() { let beets_arc = BEETS_TEST_CONFIG.clone(); let beets = &mut beets_arc.lock().unwrap(); let output = beets .list(Query::new().exclude(Field::AlbumArtist(String::from("Аркона")))) .unwrap(); let expected: Vec = LIBRARY_ITEMS .iter() .filter(|it| it.album_artist != "Аркона") .cloned() .collect(); let output: HashSet<_> = output.iter().collect(); let expected: HashSet<_> = expected.iter().collect(); assert_eq!(output, expected); }