use std::{panic, time::Duration}; use futures::{stream::FuturesUnordered, SinkExt, StreamExt}; use mpvipc::{Error, ErrorCode, Mpv, MpvExt, Playlist, PlaylistEntry}; use serde_json::{json, Value}; use test_log::test; use tokio::{net::UnixStream, task::JoinHandle}; use tokio_util::codec::{Framed, LinesCodec, LinesCodecError}; fn test_socket(answers: Vec) -> (UnixStream, JoinHandle>) { let (socket, server) = UnixStream::pair().unwrap(); let join_handle = tokio::spawn(async move { let mut framed = Framed::new(socket, LinesCodec::new()); for answer in answers { framed.next().await; framed.send(answer).await?; } Ok(()) }); (server, join_handle) } #[test(tokio::test)] async fn test_set_property_successful() { let (server, join_handle) = test_socket(vec![ json!({ "data": null, "request_id": 0, "error": "success" }).to_string(), ]); let mpv = Mpv::connect_socket(server).await.unwrap(); let volume = mpv.set_property("volume", 64.0).await; assert!(volume.is_ok()); join_handle.await.unwrap().unwrap(); } #[test(tokio::test)] async fn test_set_property_broken_pipe() { let (server, join_handle) = test_socket(vec![]); let mpv = Mpv::connect_socket(server).await.unwrap(); let maybe_set_volume = mpv.set_property("volume", 64.0).await; assert_eq!( maybe_set_volume, Err(Error(ErrorCode::ConnectError( "Broken pipe (os error 32)".to_owned() ))) ); join_handle.await.unwrap().unwrap(); } #[test(tokio::test)] async fn test_set_property_wrong_type() { let (server, join_handle) = test_socket(vec![ json!({"request_id":0,"error":"unsupported format for accessing property"}).to_string(), ]); let mpv = Mpv::connect_socket(server).await.unwrap(); let maybe_volume = mpv.set_property::("volume", true).await; assert_eq!( maybe_volume, Err(Error(ErrorCode::MpvError( "unsupported format for accessing property".to_owned() ))) ); join_handle.await.unwrap().unwrap(); } #[test(tokio::test)] async fn test_get_property_error() { let (server, join_handle) = test_socket(vec![ json!({"request_id":0,"error":"property not found"}).to_string(), ]); let mpv = Mpv::connect_socket(server).await.unwrap(); let maybe_volume = mpv.set_property("nonexistent", true).await; assert_eq!( maybe_volume, Err(Error(ErrorCode::MpvError("property not found".to_owned()))) ); join_handle.await.unwrap().unwrap(); } #[test(tokio::test)] async fn test_set_property_simultaneous_requests() { let (socket, server) = UnixStream::pair().unwrap(); let mpv_handle: JoinHandle> = tokio::spawn(async move { let mut framed = Framed::new(socket, LinesCodec::new()); while let Some(request) = framed.next().await { if let Ok(json) = serde_json::from_str::(&request.unwrap()) { let property = json["command"][1].as_str().unwrap(); let value = &json["command"][2]; log::info!( "Received set property command: {:?} => {:?}", property, value ); match property { "volume" => { let response = json!({ "request_id": 0, "error": "success" }).to_string(); framed.send(response).await.unwrap(); } "pause" => { let response = json!({ "request_id": 0, "error": "success" }).to_string(); framed.send(response).await.unwrap(); } _ => { let response = json!({ "error":"property not found", "request_id": 0 }).to_string(); framed.send(response).await.unwrap(); } } } } Ok(()) }); let mpv = Mpv::connect_socket(server).await.unwrap(); let mpv_clone_1 = mpv.clone(); let mpv_poller_1 = tokio::spawn(async move { loop { let status = mpv_clone_1.set_property("volume", 100).await; assert_eq!(status, Ok(())); } }); let mpv_clone_2 = mpv.clone(); let mpv_poller_2 = tokio::spawn(async move { loop { tokio::time::sleep(Duration::from_millis(1)).await; let status = mpv_clone_2.set_property("pause", false).await; assert_eq!(status, Ok(())); } }); let mpv_clone_3 = mpv.clone(); let mpv_poller_3 = tokio::spawn(async move { loop { tokio::time::sleep(Duration::from_millis(2)).await; let maybe_volume = mpv_clone_3.set_property("nonexistent", "a").await; assert_eq!( maybe_volume, Err(Error(ErrorCode::MpvError("property not found".to_owned()))) ); } }); let mut tasks = FuturesUnordered::new(); tasks.push(mpv_handle); tasks.push(mpv_poller_1); tasks.push(mpv_poller_2); tasks.push(mpv_poller_3); if tokio::time::timeout(Duration::from_millis(200), tasks.next()) .await .is_ok() { panic!("One of the pollers quit unexpectedly"); }; } #[test(tokio::test)] async fn test_get_playlist() { let expected = Playlist(vec![ PlaylistEntry { id: 0, filename: "file1".to_string(), title: "title1".to_string(), current: false, }, PlaylistEntry { id: 1, filename: "file2".to_string(), title: "title2".to_string(), current: true, }, PlaylistEntry { id: 2, filename: "file3".to_string(), title: "title3".to_string(), current: false, }, ]); let (server, join_handle) = test_socket(vec![json!({ "data": expected.0.iter().map(|entry| { json!({ "filename": entry.filename, "title": entry.title, "current": entry.current }) }).collect::>(), "request_id": 0, "error": "success" }) .to_string()]); let mpv = Mpv::connect_socket(server).await.unwrap(); let playlist = mpv.get_playlist().await.unwrap(); assert_eq!(playlist, expected); join_handle.await.unwrap().unwrap(); } #[test(tokio::test)] async fn test_get_playlist_empty() { let (server, join_handle) = test_socket(vec![ json!({ "data": [], "request_id": 0, "error": "success" }).to_string(), ]); let mpv = Mpv::connect_socket(server).await.unwrap(); let playlist = mpv.get_playlist().await.unwrap(); assert_eq!(playlist, Playlist(vec![])); join_handle.await.unwrap().unwrap(); }