This commit is contained in:
Petitminion 2023-06-23 14:40:36 +02:00
parent 4f7e289d0a
commit 281993c903
2 changed files with 1 additions and 4 deletions

View File

@ -184,9 +184,6 @@ class SessionRadio(SimpleRadio):
return sliced_queryset[:quantity]
def pick_v2(self, **kwargs):
return self.pick_many_v2(quantity=1, **kwargs)[0]
def pick_many_v2(self, quantity, **kwargs):
if self.session:
sliced_queryset = self.get_choices_v2(quantity=quantity, **kwargs)

View File

@ -525,7 +525,7 @@ def test_can_cache_radio_track(factories):
user = factories["users.User"]()
radio = radios.RandomRadio()
session = radio.start_session(user)
picked = session.radio.pick_v2(filter_playable=False)
picked = session.radio.pick_many_v2(quantity=1, filter_playable=False)
assert len(picked) == 1
for t in cache.get(f"radiosessiontracks{session.id}"):
assert t.track in uploads