fix(radarr): Pass the movie ID alongside all UpdateAndScan events published to the networking channel

This commit is contained in:
2024-12-17 21:34:14 -07:00
parent cb8035a2ce
commit 43410fac60
10 changed files with 26 additions and 192 deletions
+5 -23
View File
@@ -75,7 +75,7 @@ pub enum RadarrEvent {
TestAllIndexers,
TriggerAutomaticSearch(i64),
UpdateAllMovies,
UpdateAndScan(Option<i64>),
UpdateAndScan(i64),
UpdateCollections,
UpdateDownloads,
}
@@ -1778,13 +1778,12 @@ impl<'a, 'b> Network<'a, 'b> {
.await
}
async fn update_and_scan_movie(&mut self, movie_id: Option<i64>) -> Result<Value> {
let (id, _) = self.extract_movie_id(movie_id).await;
let event = RadarrEvent::UpdateAndScan(None);
info!("Updating and scanning movie with ID: {id}");
async fn update_and_scan_movie(&mut self, movie_id: i64) -> Result<Value> {
let event = RadarrEvent::UpdateAndScan(movie_id);
info!("Updating and scanning movie with ID: {movie_id}");
let body = MovieCommandBody {
name: "RefreshMovie".to_owned(),
movie_ids: vec![id],
movie_ids: vec![movie_id],
};
let request_props = self
@@ -1857,23 +1856,6 @@ impl<'a, 'b> Network<'a, 'b> {
})
.collect()
}
async fn extract_movie_id(&mut self, movie_id: Option<i64>) -> (i64, String) {
let movie_id = if let Some(id) = movie_id {
id
} else {
self
.app
.lock()
.await
.data
.radarr_data
.movies
.current_selection()
.id
};
(movie_id, format!("movieId={movie_id}"))
}
}
fn get_movie_status(has_file: bool, downloads_vec: &[DownloadRecord], movie_id: i64) -> String {
+3 -94
View File
@@ -216,7 +216,7 @@ mod test {
RadarrEvent::StartTask(RadarrTaskName::default()),
RadarrEvent::GetQueuedEvents,
RadarrEvent::TriggerAutomaticSearch(0),
RadarrEvent::UpdateAndScan(None),
RadarrEvent::UpdateAndScan(0),
RadarrEvent::UpdateAllMovies,
RadarrEvent::UpdateDownloads,
RadarrEvent::UpdateCollections
@@ -988,39 +988,7 @@ mod test {
})),
Some(json!({})),
None,
RadarrEvent::UpdateAndScan(None),
None,
None,
)
.await;
app_arc
.lock()
.await
.data
.radarr_data
.movies
.set_items(vec![movie()]);
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_radarr_event(RadarrEvent::UpdateAndScan(None))
.await
.is_ok());
async_server.assert_async().await;
}
#[tokio::test]
async fn test_handle_update_and_scan_movie_event_uses_provied_movie_id() {
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Post,
Some(json!({
"name": "RefreshMovie",
"movieIds": [ 1 ]
})),
Some(json!({})),
None,
RadarrEvent::UpdateAndScan(None),
RadarrEvent::UpdateAndScan(1),
None,
None,
)
@@ -1028,7 +996,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_radarr_event(RadarrEvent::UpdateAndScan(Some(1)))
.handle_radarr_event(RadarrEvent::UpdateAndScan(1))
.await
.is_ok());
@@ -3954,65 +3922,6 @@ mod test {
);
}
#[tokio::test]
async fn test_extract_movie_id() {
let app_arc = Arc::new(Mutex::new(App::default()));
app_arc
.lock()
.await
.data
.radarr_data
.movies
.set_items(vec![Movie {
id: 1,
..Movie::default()
}]);
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
let (id, movie_id_param) = network.extract_movie_id(None).await;
assert_eq!(id, 1);
assert_str_eq!(movie_id_param, "movieId=1");
}
#[tokio::test]
async fn test_extract_movie_id_uses_provided_id() {
let app_arc = Arc::new(Mutex::new(App::default()));
app_arc
.lock()
.await
.data
.radarr_data
.movies
.set_items(vec![Movie {
id: 1,
..Movie::default()
}]);
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
let (id, movie_id_param) = network.extract_movie_id(Some(2)).await;
assert_eq!(id, 2);
assert_str_eq!(movie_id_param, "movieId=2");
}
#[tokio::test]
async fn test_extract_movie_id_filtered_movies() {
let app_arc = Arc::new(Mutex::new(App::default()));
let mut filtered_movies = StatefulTable::default();
filtered_movies.set_filtered_items(vec![Movie {
id: 1,
..Movie::default()
}]);
app_arc.lock().await.data.radarr_data.movies = filtered_movies;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
let (id, movie_id_param) = network.extract_movie_id(None).await;
assert_eq!(id, 1);
assert_str_eq!(movie_id_param, "movieId=1");
}
#[test]
fn test_get_movie_status_downloaded() {
assert_str_eq!(get_movie_status(true, &[], 0), "Downloaded");