59 行
2.5 KiB
Python
59 行
2.5 KiB
Python
import pytest
|
|
|
|
from yandex_music import DownloadInfo
|
|
|
|
|
|
@pytest.fixture(scope='class')
|
|
def download_info():
|
|
return DownloadInfo(TestDownloadInfo.codec, TestDownloadInfo.bitrate_in_kbps, TestDownloadInfo.gain,
|
|
TestDownloadInfo.preview, TestDownloadInfo.download_info_url)
|
|
|
|
|
|
class TestDownloadInfo:
|
|
codec = 'mp3'
|
|
bitrate_in_kbps = 192
|
|
gain = False
|
|
preview = False
|
|
download_info_url = 'https://storage.mds.yandex.net/file-download-info/136146_d158926e.14534319.6.10994777/320' \
|
|
'?sign=8caf5ea72c946d4753f15298e4033b961c7acb1bb4db48eb5e6b59621e387d64&ts=5dc4a6f2 '
|
|
|
|
def test_expected_values(self, download_info):
|
|
assert download_info.codec == self.codec
|
|
assert download_info.bitrate_in_kbps == self.bitrate_in_kbps
|
|
assert download_info.gain == self.gain
|
|
assert download_info.preview == self.preview
|
|
assert download_info.download_info_url == self.download_info_url
|
|
|
|
def test_de_json_required(self, client):
|
|
json_dict = {'codec': self.codec, 'bitrate_in_kbps': self.bitrate_in_kbps, 'gain': self.gain,
|
|
'preview': self.preview, 'download_info_url': self.download_info_url}
|
|
download_info = DownloadInfo.de_json(json_dict, client)
|
|
|
|
assert download_info.codec == self.codec
|
|
assert download_info.bitrate_in_kbps == self.bitrate_in_kbps
|
|
assert download_info.gain == self.gain
|
|
assert download_info.preview == self.preview
|
|
assert download_info.download_info_url == self.download_info_url
|
|
|
|
def test_de_json_all(self, client):
|
|
json_dict = {'codec': self.codec, 'bitrate_in_kbps': self.bitrate_in_kbps, 'gain': self.gain,
|
|
'preview': self.preview, 'download_info_url': self.download_info_url}
|
|
download_info = DownloadInfo.de_json(json_dict, client)
|
|
|
|
assert download_info.codec == self.codec
|
|
assert download_info.bitrate_in_kbps == self.bitrate_in_kbps
|
|
assert download_info.gain == self.gain
|
|
assert download_info.preview == self.preview
|
|
assert download_info.download_info_url == self.download_info_url
|
|
|
|
def test_equality(self):
|
|
a = DownloadInfo(self.codec, self.bitrate_in_kbps, self.gain, self.preview, self.download_info_url)
|
|
b = DownloadInfo(self.codec, 128, self.gain, True, self.download_info_url)
|
|
c = DownloadInfo(self.codec, self.bitrate_in_kbps, self.gain, self.preview, self.download_info_url)
|
|
|
|
assert a != b
|
|
assert hash(a) != hash(b)
|
|
assert a is not b
|
|
|
|
assert a == c
|