Skip to content

Commit 25acd75

Browse files
committed
Fix tests
1 parent 19eb9c0 commit 25acd75

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

tests/test_filters.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,22 @@
22

33
import requests_mock
44

5+
from troi.patch import Patch
56
from troi import Artist, ArtistCredit, Recording, Playlist
67
import troi.filters
78
from troi.musicbrainz.recording import RecordingListElement
89
from troi.listenbrainz.feedback import ListensFeedbackLookup
910

1011

12+
class DummyPatch:
13+
14+
def __init__(self):
15+
self.local_storage = {}
16+
17+
def set_patch_object(self, obj):
18+
pass
19+
20+
1121
class TestArtistCreditFilterElement(unittest.TestCase):
1222

1323
def test_artist_credit_filter_include(self):
@@ -56,6 +66,7 @@ def test_artist_credit_limiter_higher_ranked(self):
5666
Recording(mbid='73a9d0db-0ec7-490e-9a85-0525a5ccef8e', artist_credit=ArtistCredit(artist_credit_id=197), ranking=.1)
5767
]
5868

69+
5970
e = troi.filters.ArtistCreditLimiterElement(1)
6071
flist = e.read([rlist])
6172
assert len(flist) == 2
@@ -235,14 +246,17 @@ def test_never_listened_element(self):
235246
Recording(mbid='c88b3490-35a0-460d-b3bc-bc50c8855d00'),
236247
Recording(mbid='139654ae-2c02-4e0f-aee0-c47da6e59ff1', listenbrainz={"latest_listened_at": 345345345})
237248
]
249+
p = DummyPatch()
238250
e = troi.filters.NeverListenedFilterElement()
251+
e.set_patch_object(p)
239252
flist = e.read([rlist])
240253
assert len(flist) == 2
241254

242255
assert flist[0].mbid == '8756f690-18ca-488d-a456-680fdaf234bd'
243256
assert flist[1].mbid == '139654ae-2c02-4e0f-aee0-c47da6e59ff1'
244257

245258
e = troi.filters.NeverListenedFilterElement(remove_unlistened=False)
259+
e.set_patch_object(p)
246260
flist = e.read([rlist])
247261
assert len(flist) == 2
248262

0 commit comments

Comments
 (0)