models: Fix TlTrack deserialization

This commit is contained in:
Stein Magnus Jodal 2012-11-18 17:28:37 +01:00
parent 68e4b207cb
commit 3bc4126b45
2 changed files with 6 additions and 4 deletions

View File

@ -131,6 +131,8 @@ def model_json_decoder(dct):
return Artist(**dct)
if obj_type == 'Playlist':
return Playlist(**dct)
if obj_type == 'TlTrack':
return TlTrack(**dct)
if obj_type == 'Track':
return Track(**dct)
return dct

View File

@ -622,10 +622,10 @@ class TlTrackTest(unittest.TestCase):
TlTrack(tlid=123, track=track).serialize())
def test_to_json_and_back(self):
track1 = Track(uri='uri', name='name')
serialized = json.dumps(track1, cls=ModelJSONEncoder)
track2 = json.loads(serialized, object_hook=model_json_decoder)
self.assertEqual(track1, track2)
tl_track1 = TlTrack(tlid=123, track=Track(uri='uri', name='name'))
serialized = json.dumps(tl_track1, cls=ModelJSONEncoder)
tl_track2 = json.loads(serialized, object_hook=model_json_decoder)
self.assertEqual(tl_track1, tl_track2)
def test_eq(self):
tlid = 123