diff --git a/tests/conftest.py b/tests/conftest.py index 4280158..6ba0c63 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -52,7 +52,8 @@ def track_factory(major, normalization): TestTrack.og_image, TestTrack.type, TestTrack.cover_uri, major, TestTrack.duration_ms, TestTrack.storage_dir, TestTrack.file_size, normalization, TestTrack.error, TestTrack.regions, TestTrack.available_as_rbt, TestTrack.content_warning, TestTrack.explicit, - TestTrack.preview_duration_ms, TestTrack.available_full_without_permission) + TestTrack.preview_duration_ms, TestTrack.available_full_without_permission, TestTrack.version, + TestTrack.remember_position) return TrackFactory() diff --git a/tests/test_track.py b/tests/test_track.py index 5bae80f..c407371 100644 --- a/tests/test_track.py +++ b/tests/test_track.py @@ -21,6 +21,8 @@ class TestTrack: explicit = None preview_duration_ms = 30000 available_full_without_permission = False + version = 'Radio Edit' + remember_position = False def test_expected_values(self, track, artist, album, major, normalization): assert track.id == self.id @@ -46,6 +48,8 @@ class TestTrack: assert track.explicit == self.explicit assert track.preview_duration_ms == self.preview_duration_ms assert track.available_full_without_permission == self.available_full_without_permission + assert track.version == self.version + assert track.remember_position == self.remember_position def test_de_json_none(self, client): assert Track.de_json({}, client) is None @@ -74,8 +78,9 @@ class TestTrack: 'file_size': self.file_size, 'normalization': normalization.to_dict(), 'error': self.error, 'regions': self.regions, 'available_as_rbt': self.available_as_rbt, 'content_warning': self.content_warning, 'explicit': self.explicit, - 'preview_duration_ms': self.preview_duration_ms, - 'available_full_without_permission': self.available_full_without_permission} + 'preview_duration_ms': self.preview_duration_ms, 'version': self.version, + 'available_full_without_permission': self.available_full_without_permission, + 'remember_position': self.remember_position} track = Track.de_json(json_dict, client) assert track.id == self.id @@ -101,6 +106,8 @@ class TestTrack: assert track.explicit == self.explicit assert track.preview_duration_ms == self.preview_duration_ms assert track.available_full_without_permission == self.available_full_without_permission + assert track.version == self.version + assert track.remember_position == self.remember_position def test_equality(self, artist, album): a = Track(self.id, self.title, self.available, [artist], [album]) diff --git a/yandex_music/track/track.py b/yandex_music/track/track.py index 0408d22..738248e 100644 --- a/yandex_music/track/track.py +++ b/yandex_music/track/track.py @@ -43,6 +43,8 @@ class Track(YandexMusicObject): explicit=None, preview_duration_ms: Optional[int] = None, available_full_without_permission: Optional[bool] = None, + version: Optional[str] = None, + remember_position: Optional[bool] = None, client: Optional['Client'] = None, **kwargs) -> None: self.id = id_ @@ -69,6 +71,8 @@ class Track(YandexMusicObject): self.explicit = explicit self.preview_duration_ms = preview_duration_ms self.available_full_without_permission = available_full_without_permission + self.version = version + self.remember_position = remember_position self.download_info = None