diff --git a/tests/scrobbles_tests/test_aggregators.py b/tests/scrobbles_tests/test_aggregators.py index 433e02f..20e4588 100644 --- a/tests/scrobbles_tests/test_aggregators.py +++ b/tests/scrobbles_tests/test_aggregators.py @@ -1,4 +1,5 @@ from datetime import datetime, timedelta +from unittest.mock import patch import pytest import time_machine @@ -26,8 +27,21 @@ def build_scrobbles(client, request_json, num=7, spacing=2): @pytest.mark.django_db +@patch("music.models.get_album_metadata_with_artist", return_value={}) +@patch("music.models.get_track_metadata_with_artist", return_value={}) +@patch("music.models.get_recording_mbid_exact", return_value=(None, None)) +@patch("music.models.lookup_artist_from_tadb", return_value={}) +@patch("music.models.lookup_album_from_tadb", return_value={}) @time_machine.travel(datetime(2022, 3, 4, 1, 24)) -def test_scrobble_counts_data(client, mopidy_track): +def test_scrobble_counts_data( + mock_lookup_album_tadb, + mock_lookup_artist_tadb, + mock_get_recording, + mock_get_track, + mock_get_album, + client, + mopidy_track +): build_scrobbles(client, mopidy_track.request_json) user = get_user_model().objects.first() count_dict = scrobble_counts(user) @@ -41,8 +55,21 @@ def test_scrobble_counts_data(client, mopidy_track): @pytest.mark.django_db +@patch("music.models.get_album_metadata_with_artist", return_value={}) +@patch("music.models.get_track_metadata_with_artist", return_value={}) +@patch("music.models.get_recording_mbid_exact", return_value=(None, None)) +@patch("music.models.lookup_artist_from_tadb", return_value={}) +@patch("music.models.lookup_album_from_tadb", return_value={}) @time_machine.travel(datetime(2022, 3, 4, 1, 24)) -def test_live_charts(client, mopidy_track): +def test_live_charts( + mock_lookup_album_tadb, + mock_lookup_artist_tadb, + mock_get_recording, + mock_get_track, + mock_get_album, + client, + mopidy_track +): build_scrobbles(client, mopidy_track.request_json, 7, 1) user = get_user_model().objects.first()