|
25 | 25 | AlbumMatch, |
26 | 26 | TrackInfo, |
27 | 27 | TrackMatch, |
| 28 | + _is_valid_mb_id, |
28 | 29 | correct_list_fields, |
29 | 30 | ) |
30 | 31 | from beets.library import Item |
@@ -426,3 +427,146 @@ def test_correct_list_fields( |
426 | 427 | data = correct_list_fields(input_data) |
427 | 428 |
|
428 | 429 | assert (data[single_field], data[list_field]) == expected_values |
| 430 | + |
| 431 | + |
| 432 | +# Tests for issue #6519: non-MB sources must not pollute MusicBrainz ID tags |
| 433 | +# Real IDs from the issue report (Deezer numeric) and a typical Spotify ID. |
| 434 | +_SPOTIFY_ALBUM_ID = "4aawyAB9vmqN3uQ7FjRGTy" |
| 435 | +_SPOTIFY_TRACK_ID = "6rqhFgbbKwnb9MLmUQDhG6" |
| 436 | +_SPOTIFY_ARTIST_ID = "4Z8W4fKeB5YxbusRsdQVPb" |
| 437 | +_DEEZER_ALBUM_ID = "158760742" |
| 438 | +_DEEZER_TRACK_ID = "930705291" |
| 439 | +_DEEZER_ARTIST_ID = "3876259001" |
| 440 | + |
| 441 | +_VALID_MB_ALBUM_UUID = "7edb51cb-77d6-4416-a23c-3a8c2994a2c7" |
| 442 | +_VALID_MB_TRACK_UUID = "dfa939ec-118c-4d0f-84a0-60f3d1e6522c" |
| 443 | +_VALID_MB_ARTIST_UUID = "a6623d39-2d8e-4f70-8242-0a9553b91e50" |
| 444 | + |
| 445 | + |
| 446 | +@pytest.mark.parametrize( |
| 447 | + "value, expected", |
| 448 | + [ |
| 449 | + # Valid UUIDs |
| 450 | + (_VALID_MB_ALBUM_UUID, True), |
| 451 | + (_VALID_MB_TRACK_UUID, True), |
| 452 | + # Valid UUID list |
| 453 | + ([_VALID_MB_ALBUM_UUID, _VALID_MB_ARTIST_UUID], True), |
| 454 | + # Spotify base-62 IDs |
| 455 | + (_SPOTIFY_ALBUM_ID, False), |
| 456 | + (_SPOTIFY_TRACK_ID, False), |
| 457 | + # Deezer numeric IDs |
| 458 | + (_DEEZER_ALBUM_ID, False), |
| 459 | + (_DEEZER_TRACK_ID, False), |
| 460 | + # Edge cases |
| 461 | + (None, False), |
| 462 | + ("", False), |
| 463 | + ([], False), |
| 464 | + ], |
| 465 | +) |
| 466 | +def test_is_valid_mb_id(value, expected): |
| 467 | + assert _is_valid_mb_id(value) == expected |
| 468 | + |
| 469 | + |
| 470 | +# Regression tests for https://github.com/beetbox/beets/issues/6519: |
| 471 | +# non-MB sources must not write their IDs into MusicBrainz tags. |
| 472 | + |
| 473 | + |
| 474 | +@pytest.mark.parametrize( |
| 475 | + "album_id, artist_id, track_id", |
| 476 | + [ |
| 477 | + (_SPOTIFY_ALBUM_ID, _SPOTIFY_ARTIST_ID, _SPOTIFY_TRACK_ID), |
| 478 | + (_DEEZER_ALBUM_ID, _DEEZER_ARTIST_ID, _DEEZER_TRACK_ID), |
| 479 | + ], |
| 480 | + ids=["spotify", "deezer"], |
| 481 | +) |
| 482 | +def test_non_mb_album_ids_not_in_mb_fields( |
| 483 | + config, album_id, artist_id, track_id |
| 484 | +): |
| 485 | + """Non-MB album/artist IDs must not appear in MusicBrainz tag fields.""" |
| 486 | + info = AlbumInfo( |
| 487 | + tracks=[ |
| 488 | + TrackInfo( |
| 489 | + title="Track", |
| 490 | + track_id=track_id, |
| 491 | + artist_id=artist_id, |
| 492 | + medium=1, |
| 493 | + medium_index=1, |
| 494 | + medium_total=1, |
| 495 | + index=1, |
| 496 | + ) |
| 497 | + ], |
| 498 | + album="Album", |
| 499 | + album_id=album_id, |
| 500 | + artist="Artist", |
| 501 | + artist_id=artist_id, |
| 502 | + year=2024, |
| 503 | + ) |
| 504 | + data = info.item_data |
| 505 | + assert "mb_albumid" not in data |
| 506 | + assert "mb_albumartistid" not in data |
| 507 | + assert "mb_albumartistids" not in data |
| 508 | + |
| 509 | + |
| 510 | +@pytest.mark.parametrize( |
| 511 | + "track_id, artist_id", |
| 512 | + [ |
| 513 | + (_SPOTIFY_TRACK_ID, _SPOTIFY_ARTIST_ID), |
| 514 | + (_DEEZER_TRACK_ID, _DEEZER_ARTIST_ID), |
| 515 | + ], |
| 516 | + ids=["spotify", "deezer"], |
| 517 | +) |
| 518 | +def test_non_mb_track_ids_not_in_mb_fields(config, track_id, artist_id): |
| 519 | + """Non-MB track/artist IDs must not appear in MusicBrainz tag fields.""" |
| 520 | + track = TrackInfo( |
| 521 | + title="Track", |
| 522 | + track_id=track_id, |
| 523 | + artist_id=artist_id, |
| 524 | + medium=1, |
| 525 | + medium_index=1, |
| 526 | + index=1, |
| 527 | + ) |
| 528 | + data = track.item_data |
| 529 | + assert "mb_trackid" not in data |
| 530 | + assert "mb_artistid" not in data |
| 531 | + assert "mb_artistids" not in data |
| 532 | + |
| 533 | + |
| 534 | +def test_valid_mb_album_uuids_are_written(config): |
| 535 | + """Regression: valid MusicBrainz UUIDs must still be written to MB tags.""" |
| 536 | + info = AlbumInfo( |
| 537 | + tracks=[ |
| 538 | + TrackInfo( |
| 539 | + title="Track", |
| 540 | + track_id=_VALID_MB_TRACK_UUID, |
| 541 | + artist_id=_VALID_MB_ARTIST_UUID, |
| 542 | + medium=1, |
| 543 | + medium_index=1, |
| 544 | + medium_total=1, |
| 545 | + index=1, |
| 546 | + ) |
| 547 | + ], |
| 548 | + album="Album", |
| 549 | + album_id=_VALID_MB_ALBUM_UUID, |
| 550 | + artist="Artist", |
| 551 | + artist_id=_VALID_MB_ARTIST_UUID, |
| 552 | + data_source="MusicBrainz", |
| 553 | + year=2024, |
| 554 | + ) |
| 555 | + data = info.item_data |
| 556 | + assert data["mb_albumid"] == _VALID_MB_ALBUM_UUID |
| 557 | + assert data["mb_albumartistid"] == _VALID_MB_ARTIST_UUID |
| 558 | + |
| 559 | + |
| 560 | +def test_valid_mb_track_uuid_is_written(config): |
| 561 | + """Regression: valid MusicBrainz UUID in track_id must still be written.""" |
| 562 | + track = TrackInfo( |
| 563 | + title="Track", |
| 564 | + track_id=_VALID_MB_TRACK_UUID, |
| 565 | + artist_id=_VALID_MB_ARTIST_UUID, |
| 566 | + medium=1, |
| 567 | + medium_index=1, |
| 568 | + index=1, |
| 569 | + ) |
| 570 | + data = track.item_data |
| 571 | + assert data["mb_trackid"] == _VALID_MB_TRACK_UUID |
| 572 | + assert data["mb_artistid"] == _VALID_MB_ARTIST_UUID |
0 commit comments