From: Chris Fulljames Date: Sat, 5 Apr 2025 18:48:32 +0000 (-0400) Subject: Fix random songs page X-Git-Url: https://littlesong.place/gitweb/?a=commitdiff_plain;h=cbb7691e827419d220a6c43e259d1323a74fbbf7;p=littlesongplace.git Fix random songs page --- diff --git a/src/littlesongplace/songs.py b/src/littlesongplace/songs.py index ec7b14e..471da51 100644 --- a/src/littlesongplace/songs.py +++ b/src/littlesongplace/songs.py @@ -1,4 +1,5 @@ import json +import random from datetime import datetime from dataclasses import dataclass diff --git a/test/test_songlists.py b/test/test_songlists.py index 8c864b0..1bfd84a 100644 --- a/test/test_songlists.py +++ b/test/test_songlists.py @@ -105,3 +105,13 @@ def test_single_song(client): assert songs[0]["title"] == "song1" assert songs[0]["username"] == "user1" +def test_random_songs(client): + create_user(client, "user1", "password", login=True) + upload_song(client, b"Success", user="user1", title="song1", tags="tag") + upload_song(client, b"Success", user="user1", title="song2", tags="") + + songs = get_song_list_from_page(client, "/songs") + + assert len(songs) == 2 + assert songs[0]["title"] in ["song1", "song2"] + assert songs[1]["title"] in ["song1", "song2"]