Function test_remove_songs

src-tauri/database/src/test.rs:210–291

()

// Test removing songs
#[test]
fn test_remove_songs() {
    let db_path = get_test_db_path();
    let db = Database::new(db_path.clone());

    // Insert test songs
    let songs = db
        .insert_songs(vec![
            create_test_song("Song to Keep", "/path/to/keep.mp3"),
            create_test_song("Song to Remove", "/path/to/remove.mp3"),
        ])
        .unwrap();

    // Get IDs
    let keep_id = songs[0].song._id.clone().unwrap();
    let remove_id = songs[1].song._id.clone().unwrap();

    // Make sure we have 2 songs before removing
    let initial_songs = db
        .get_songs_by_options(GetSongOptions {
            song: Some(SearchableSong {
                type_: Some(SongType::LOCAL), // Filter by song type to ensure we get all test songs
                ..Default::default()
            }),
            inclusive: Some(true),
            ..Default::default()
        })
        .unwrap();
    assert_eq!(initial_songs.len(), 2);

    // Add analytics data to both songs
    db.increment_play_count(keep_id.clone()).unwrap();
    db.increment_play_count(remove_id.clone()).unwrap();
    db.increment_play_time(keep_id.clone(), 60.0).unwrap();
    db.increment_play_time(remove_id.clone(), 120.0).unwrap();

    // Remove one song
    db.remove_songs(vec![remove_id.clone()]).unwrap();

    // Verify only one song remains
    let all_songs = db
        .get_songs_by_options(GetSongOptions {
            song: Some(SearchableSong {
                type_: Some(SongType::LOCAL), // Filter by song type to ensure we get all remaining songs
                ..Default::default()
            }),
            inclusive: Some(true),
            ..Default::default()
        })
        .unwrap();
    assert_eq!(all_songs.len(), 1);
    assert_eq!(all_songs[0].song._id.as_ref().unwrap(), &keep_id);

    // Verify analytics data for the removed song is also gone
    let analytics = db.get_top_listened_songs().unwrap();
    let removed_song_analytics = analytics.songs.iter().find(|(id, _)| id == &remove_id);
    assert!(
        removed_song_analytics.is_none(),
        "Analytics for removed song should be deleted"