def testArtistRatingPickle(self):
Factories.clear()
trackA= TrackFactory.get(self.artist, self.titleA)
trackB = TrackFactory.get(self.artist, self.titleB)
artist = trackA.artist
trackA.played()
trackB.skipped()
rating = artist.rating
i = 0
for r in RatingFactory.ratings():
i+=1
# track init rating: 1
# artist init rating: 2
# track play rating: 3
# artist play rating: 4
# track skip rating: 5
# artist skip rating: 6
self.assertEqual(i, 6)
state = pickle.dumps(Factories.getstate())
Factories.clear()
pickle.loads(state)
i = 0
for r in RatingFactory.ratings():
i+=1
self.assertEqual(i, 6)
track = TrackFactory.get(self.artist, self.titleA)
artist = track.artist
self.assertEqual(artist.rating, rating)
def testArtistPickle(self):
Factories.clear()
artist = ArtistFactory.get(self.artist)
artist.test = self.artist_test
state = pickle.dumps(Factories.getstate())
Factories.clear()
pickle.loads(state)
artist = ArtistFactory.get(self.artist)
self.assertEqual(artist.test, self.artist_test)
def testTrackPickle(self):
Factories.clear()
track = TrackFactory.get(self.artist, self.title)
track.test = self.track_test
track.artist.test = self.artist_test
state = pickle.dumps(Factories.getstate())
Factories.clear()
pickle.loads(state)
self.assertEqual(track.title, self.title)
self.assertEqual(track.test, self.track_test)
self.assertEqual(track.artist.name, self.artist)
self.assertEqual(track.artist.test, self.artist_test)
def testArtistRelationPickle(self):
Factories.clear()
artistA = ArtistFactory.get(self.artistA)
artistB = ArtistFactory.get(self.artistB)
artist_relation = ArtistRelationFactory.get(artistA, artistB)
artist_relation.test = self.artist_relation_test
state = pickle.dumps(Factories.getstate())
Factories.clear()
pickle.loads(state)
artistA = ArtistFactory.get(self.artistA)
artistB = ArtistFactory.get(self.artistB)
artist_relation = ArtistRelationFactory.get(artistA, artistB)
artist_relation.test = self.artist_relation_test
self.assertEqual(artist_relation.test, self.artist_relation_test)
def testRatingPickle(self):
Factories.clear()
track = TrackFactory.get(self.artist, self.title)
track.played()
track.skipped()
track.played(0.5)
track.skipped(0.5)
track.ratingref.test = self.rating_test
state = pickle.dumps(Factories.getstate())
Factories.clear()
pickle.loads(state)
track = TrackFactory.get(self.artist, self.title)
self.assertEqual(track.ratingref.playcount, 1)
self.assertEqual(track.ratingref.skipcount, 1)
self.assertEqual(track.ratingref.playfactor, 1.5)
self.assertEqual(track.ratingref.skipfactor, 1.5)
self.assertEqual(track.ratingref.test, self.rating_test)
def testFilePickle(self):
Factories.clear()
file = FileFactory.get(self.path, self.artist, self.title)
file.test = self.file_test
file.track.test = self.track_test
file.track.artist.test = self.artist_test
state = pickle.dumps(Factories.getstate())
Factories.clear()
file = FileFactory.get(self.path, self.artist, self.title)
Factories.clear()
pickle.loads(state)
file = FileFactory.get(self.path, self.artist, self.title)
self.assertEqual(file.path, self.path)
self.assertEqual(file.track.artist.name, self.artist)
self.assertEqual(file.track.title, self.title)
self.assertEqual(file.test, self.file_test)
self.assertEqual(file.track.test, self.track_test)
self.assertEqual(file.track.artist.test, self.artist_test)
