diff --git a/tests/conftest.py b/tests/conftest.py index 0fdc102..fb2db5c 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -193,8 +193,8 @@ def icon(): @pytest.fixture(scope='session') def cover(): - return Cover(TestCover.type, TestCover.uri, TestCover.items_uri, TestCover.dir, - TestCover.version, TestCover.custom, TestCover.prefix, TestCover.error) + return Cover(TestCover.type, TestCover.uri, TestCover.items_uri, TestCover.dir, TestCover.version, + TestCover.custom, TestCover.is_custom, TestCover.prefix, TestCover.error) @pytest.fixture(scope='session') diff --git a/tests/test_cover.py b/tests/test_cover.py index 10201fc..d80e308 100644 --- a/tests/test_cover.py +++ b/tests/test_cover.py @@ -8,6 +8,7 @@ class TestCover: dir = '/get-music-user-playlist/34120/pvg900XixWaHcr/' version = '1572609906461' custom = True + is_custom = True prefix = None error = None @@ -18,6 +19,7 @@ class TestCover: assert cover.dir == self.dir assert cover.version == self.version assert cover.custom == self.custom + assert cover.is_custom == self.is_custom assert cover.prefix == self.prefix assert cover.error == self.error @@ -33,7 +35,8 @@ class TestCover: def test_de_json_all(self, client): json_dict = {'type_': self.type, 'uri': self.uri, 'items_uri': self.items_uri, 'dir_': self.dir, - 'version': self.version, 'custom': self.custom, 'prefix': self.prefix, 'error': self.error} + 'version': self.version, 'custom': self.custom, 'is_custom': self.is_custom, 'prefix': self.prefix, + 'error': self.error} cover = Cover.de_json(json_dict, client) assert cover.type == self.type @@ -42,6 +45,7 @@ class TestCover: assert cover.dir == self.dir assert cover.version == self.version assert cover.custom == self.custom + assert cover.is_custom == self.is_custom assert cover.prefix == self.prefix assert cover.error == self.error diff --git a/yandex_music/cover.py b/yandex_music/cover.py index 71b607f..ac5f63c 100644 --- a/yandex_music/cover.py +++ b/yandex_music/cover.py @@ -15,6 +15,7 @@ class Cover(YandexMusicObject): items_uri (:obj:`str`): Список ссылок на изображения. dir (:obj:`str`): Директория хранения изображения на сервере. version (:obj:`str`): Версия. + is_custom (:obj:`bool`): Является ли обложка пользовательской. custom (:obj:`bool`): Является ли обложка пользовательской. prefix (:obj:`str`): Уникальный идентификатор. error (:obj:`str`): Сообщение об ошибке. @@ -26,6 +27,7 @@ class Cover(YandexMusicObject): items_uri (:obj:`str`, optional): Список ссылок на изображения. dir_ (:obj:`str`, optional): Директория хранения изображения на сервере. version (:obj:`str`, optional): Версия. + is_custom (:obj:`bool`, optional): Является ли обложка пользовательской. custom (:obj:`bool`, optional): Является ли обложка пользовательской. prefix (:obj:`str`, optional): Уникальный идентификатор. error (:obj:`str`, optional): Сообщение об ошибке. @@ -40,6 +42,7 @@ class Cover(YandexMusicObject): dir_: Optional[str] = None, version: Optional[str] = None, custom: Optional[bool] = None, + is_custom: Optional[bool] = None, prefix: Optional[str] = None, error: Optional[str] = None, client: Optional['Client'] = None, @@ -53,6 +56,7 @@ class Cover(YandexMusicObject): self.dir = dir_ self.version = version self.custom = custom + self.is_custom = is_custom self.error = error self.client = client