From cbb7691e827419d220a6c43e259d1323a74fbbf7 Mon Sep 17 00:00:00 2001 From: Chris Fulljames Date: Sat, 5 Apr 2025 14:48:32 -0400 Subject: [PATCH] Fix random songs page --- src/littlesongplace/songs.py | 1 + test/test_songlists.py | 10 ++++++++++ 2 files changed, 11 insertions(+) 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"] -- 2.39.5