feat(lidarr): highlight tracks whose album is downloading or awaiting import

Lidarr queues whole albums rather than individual tracks, so a track row
is styled from the queue entry whose albumId matches the track's albumId,
mirroring the Sonarr episode row logic. Tracks that already have a file
keep the downloaded style regardless of the queue.

src/ui/lidarr_ui/library/album_details_ui_tests.rs

@@ -2,14 +2,16 @@
 mod tests {
   use pretty_assertions::assert_eq;
   use ratatui::widgets::{Cell, Row};
+  use serde_json::Number;
   use strum::IntoEnumIterator;
 
   use crate::app::App;
-  use crate::models::lidarr_models::Track;
+  use crate::models::lidarr_models::{DownloadRecord, DownloadStatus, Track};
   use crate::models::servarr_data::lidarr::lidarr_data::{
     ALBUM_DETAILS_BLOCKS, ActiveLidarrBlock, TRACK_DETAILS_BLOCKS,
   };
   use crate::models::stateful_table::StatefulTable;
+  use crate::network::lidarr_network::lidarr_network_test_utils::test_utils::download_record;
   use crate::ui::DrawUi;
   use crate::ui::lidarr_ui::library::album_details_ui::{
     AlbumDetailsUi, decorate_track_row_with_style,
@@ -39,7 +41,7 @@ mod tests {
     };
     let row = Row::new(vec![Cell::from("test".to_owned())]);
 
-    let style = decorate_track_row_with_style(&track, row.clone());
+    let style = decorate_track_row_with_style(&[], &track, row.clone());
 
     assert_eq!(style, row.downloaded());
   }
@@ -52,11 +54,89 @@ mod tests {
     };
     let row = Row::new(vec![Cell::from("test".to_owned())]);
 
-    let style = decorate_track_row_with_style(&track, row.clone());
+    let style = decorate_track_row_with_style(&[], &track, row.clone());
 
     assert_eq!(style, row.missing());
   }
 
+  #[test]
+  fn test_decorate_track_row_with_style_downloading_when_track_has_no_file_and_album_is_downloading()
+   {
+    let downloads_vec = vec![album_download(7, DownloadStatus::Downloading)];
+    let track = Track {
+      has_file: false,
+      album_id: 7,
+      ..Track::default()
+    };
+    let row = Row::new(vec![Cell::from("test".to_owned())]);
+
+    let style = decorate_track_row_with_style(&downloads_vec, &track, row.clone());
+
+    assert_eq!(style, row.downloading());
+  }
+
+  #[test]
+  fn test_decorate_track_row_with_style_awaiting_import_when_track_has_no_file_and_album_download_is_completed()
+   {
+    let downloads_vec = vec![album_download(7, DownloadStatus::Completed)];
+    let track = Track {
+      has_file: false,
+      album_id: 7,
+      ..Track::default()
+    };
+    let row = Row::new(vec![Cell::from("test".to_owned())]);
+
+    let style = decorate_track_row_with_style(&downloads_vec, &track, row.clone());
+
+    assert_eq!(style, row.awaiting_import());
+  }
+
+  #[test]
+  fn test_decorate_track_row_with_style_missing_when_album_download_is_neither_downloading_nor_completed()
+   {
+    let downloads_vec = vec![album_download(7, DownloadStatus::Queued)];
+    let track = Track {
+      has_file: false,
+      album_id: 7,
+      ..Track::default()
+    };
+    let row = Row::new(vec![Cell::from("test".to_owned())]);
+
+    let style = decorate_track_row_with_style(&downloads_vec, &track, row.clone());
+
+    assert_eq!(style, row.missing());
+  }
+
+  #[test]
+  fn test_decorate_track_row_with_style_missing_when_only_a_different_album_is_downloading() {
+    let downloads_vec = vec![album_download(8, DownloadStatus::Downloading)];
+    let track = Track {
+      has_file: false,
+      album_id: 7,
+      ..Track::default()
+    };
+    let row = Row::new(vec![Cell::from("test".to_owned())]);
+
+    let style = decorate_track_row_with_style(&downloads_vec, &track, row.clone());
+
+    assert_eq!(style, row.missing());
+  }
+
+  #[test]
+  fn test_decorate_track_row_with_style_downloaded_when_track_has_file_and_album_is_downloading() {
+    let downloads_vec = vec![album_download(7, DownloadStatus::Downloading)];
+    let track = Track {
+      has_file: true,
+      album_id: 7,
+      ..Track::default()
+    };
+    let row = Row::new(vec![Cell::from("test".to_owned())]);
+
+    let style = decorate_track_row_with_style(&downloads_vec, &track, row.clone());
+
+    assert_eq!(style, row.downloaded());
+  }
+
   mod snapshot_tests {
     use crate::ui::ui_test_utils::test_utils::TerminalSize;
     use rstest::rstest;
@@ -177,4 +257,13 @@ mod tests {
       insta::assert_snapshot!(output);
     }
   }
+
+  fn album_download(album_id: i64, status: DownloadStatus) -> DownloadRecord {
+    DownloadRecord {
+      status,
+      album_id: Some(Number::from(album_id)),
+      artist_id: Some(Number::from(3i64)),
+      ..download_record()
+    }
+  }
 }