From 343236d38dee56198d5dc60c99f5b7be7e8333cb Mon Sep 17 00:00:00 2001
From: Il`ya Semyonov <Ilya@marshal.by>
Date: Wed, 11 Mar 2020 01:13:14 +0300
Subject: [PATCH] =?UTF-8?q?=D0=92=20=D0=BA=D0=BB=D0=B0=D1=81=D1=81=20Track?=
 =?UTF-8?q?=20=D0=B4=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5=D0=BD=D1=8B=20?=
 =?UTF-8?q?=D1=80=D0=B0=D0=BD=D0=B5=D0=B5=20=D0=BD=D0=B5=D0=B8=D0=B7=D0=B2?=
 =?UTF-8?q?=D0=B5=D1=81=D1=82=D0=BD=D1=8B=D0=B5=20=D0=BF=D0=BE=D0=BB=D1=8F?=
 =?UTF-8?q?:=20version,=20remember=5Fposition.=20#238?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/conftest.py           |  3 ++-
 tests/test_track.py         | 11 +++++++++--
 yandex_music/track/track.py |  4 ++++
 3 files changed, 15 insertions(+), 3 deletions(-)

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