secstate/vrobbler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[tests] Fix slow aggregation tests

Browse Source
This commit is contained in:
Colin Powell
2026-03-02 10:30:32 -05:00
parent d1844c01a0
commit 2d8f433314
1 changed files with 29 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
tests/scrobbles_tests/test_aggregators.py
View File

@ -1,4 +1,5 @@
from datetime import datetime, timedelta from datetime import datetime, timedelta
from unittest.mock import patch
import pytest import pytest
import time_machine import time_machine
@ -26,8 +27,21 @@ def build_scrobbles(client, request_json, num=7, spacing=2):
@pytest.mark.django_db @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)) @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) build_scrobbles(client, mopidy_track.request_json)
user = get_user_model().objects.first() user = get_user_model().objects.first()
count_dict = scrobble_counts(user) count_dict = scrobble_counts(user)
@ -41,8 +55,21 @@ def test_scrobble_counts_data(client, mopidy_track):
@pytest.mark.django_db @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)) @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) build_scrobbles(client, mopidy_track.request_json, 7, 1)
user = get_user_model().objects.first() user = get_user_model().objects.first()