90 行
4.8 KiB
Python
90 行
4.8 KiB
Python
import pytest
|
|
|
|
from yandex_music import BriefInfo
|
|
|
|
|
|
@pytest.fixture(scope='class')
|
|
def brief_info(artist, track, album, cover, playlist_id, video, chart, vinyl):
|
|
return BriefInfo(artist, [album], [album], TestBriefInfo.last_release_ids, [track], [artist], [cover],
|
|
TestBriefInfo.concerts, [video], [vinyl], TestBriefInfo.has_promotions, [playlist_id], [chart])
|
|
|
|
|
|
class TestBriefInfo:
|
|
last_release_ids = [8501194, 8302547, 8302836, 8302450]
|
|
concerts = None
|
|
has_promotions = False
|
|
|
|
def test_expected_values(self, brief_info, artist, track, album, cover, playlist_id, video, chart, vinyl):
|
|
assert brief_info.artist == artist
|
|
assert brief_info.albums == [album]
|
|
assert brief_info.also_albums == [album]
|
|
assert brief_info.last_release_ids == self.last_release_ids
|
|
assert brief_info.popular_tracks == [track]
|
|
assert brief_info.similar_artists == [artist]
|
|
assert brief_info.all_covers == [cover]
|
|
assert brief_info.concerts == self.concerts
|
|
assert brief_info.videos == [video]
|
|
assert brief_info.vinyls == [vinyl]
|
|
assert brief_info.has_promotions == self.has_promotions
|
|
assert brief_info.playlist_ids == [playlist_id]
|
|
assert brief_info.tracks_in_chart == [chart]
|
|
|
|
def test_de_json_required(self, client, artist, track, album, cover, playlist_id, video, vinyl):
|
|
json_dict = {'artist': artist.to_dict(), 'albums': [album.to_dict()], 'also_albums': [album.to_dict()],
|
|
'last_release_ids': self.last_release_ids, 'popular_tracks': [track.to_dict()],
|
|
'similar_artists': [artist.to_dict()], 'all_covers': [cover.to_dict()], 'concerts': self.concerts,
|
|
'videos': [video.to_dict()], 'vinyls': [vinyl.to_dict()], 'has_promotions': self.has_promotions,
|
|
'playlist_ids': [playlist_id.to_dict()]}
|
|
brief_info = BriefInfo.de_json(json_dict, client)
|
|
|
|
assert brief_info.artist == artist
|
|
assert brief_info.albums == [album]
|
|
assert brief_info.also_albums == [album]
|
|
assert brief_info.last_release_ids == self.last_release_ids
|
|
assert brief_info.popular_tracks == [track]
|
|
assert brief_info.similar_artists == [artist]
|
|
assert brief_info.all_covers == [cover]
|
|
assert brief_info.concerts == self.concerts
|
|
assert brief_info.videos == [video]
|
|
assert brief_info.vinyls == [vinyl]
|
|
assert brief_info.has_promotions == self.has_promotions
|
|
assert brief_info.playlist_ids == [playlist_id]
|
|
|
|
def test_de_json_all(self, client, artist, track, album, cover, playlist_id, video, chart, vinyl):
|
|
json_dict = {'artist': artist.to_dict(), 'albums': [album.to_dict()], 'also_albums': [album.to_dict()],
|
|
'last_release_ids': self.last_release_ids, 'popular_tracks': [track.to_dict()],
|
|
'similar_artists': [artist.to_dict()], 'all_covers': [cover.to_dict()], 'concerts': self.concerts,
|
|
'videos': [video.to_dict()], 'vinyls': [vinyl.to_dict()], 'has_promotions': self.has_promotions,
|
|
'playlist_ids': [playlist_id.to_dict()], 'tracks_in_chart': [chart.to_dict()]}
|
|
brief_info = BriefInfo.de_json(json_dict, client)
|
|
|
|
assert brief_info.artist == artist
|
|
assert brief_info.albums == [album]
|
|
assert brief_info.also_albums == [album]
|
|
assert brief_info.last_release_ids == self.last_release_ids
|
|
assert brief_info.popular_tracks == [track]
|
|
assert brief_info.similar_artists == [artist]
|
|
assert brief_info.all_covers == [cover]
|
|
assert brief_info.concerts == self.concerts
|
|
assert brief_info.videos == [video]
|
|
assert brief_info.vinyls == [vinyl]
|
|
assert brief_info.has_promotions == self.has_promotions
|
|
assert brief_info.playlist_ids == [playlist_id]
|
|
assert brief_info.tracks_in_chart == [chart]
|
|
|
|
def test_equality(self, artist, track, album, cover, playlist_id, video, vinyl):
|
|
a = BriefInfo(artist, [album], [album], self.last_release_ids, [track], [artist], [cover], self.concerts,
|
|
[video], [vinyl], self.has_promotions, [playlist_id])
|
|
b = BriefInfo(artist, [album], [album], self.last_release_ids, [], [artist], [cover], self.concerts,
|
|
[video], [vinyl], True, [playlist_id])
|
|
c = BriefInfo(artist, [album], [album], [1, 2, 3], [track], [artist], [], self.concerts,
|
|
[video], [vinyl], self.has_promotions, [playlist_id])
|
|
d = BriefInfo(artist, [album], [album], self.last_release_ids, [track], [artist], [cover], self.concerts,
|
|
[video], [vinyl], self.has_promotions, [playlist_id])
|
|
|
|
assert a != b != c
|
|
assert hash(a) != hash(b) != hash(c)
|
|
assert a is not b is not c
|
|
|
|
assert a == d
|