59 lines
2.7 KiB
Python
59 lines
2.7 KiB
Python
# Copyright 2024 (c) Anna Schumaker.
|
|
"""Test creating a liblistenbrainz.Listen from a Track."""
|
|
import datetime
|
|
import dateutil.tz
|
|
import emmental.listenbrainz.listen
|
|
import liblistenbrainz
|
|
import pathlib
|
|
import tests.util
|
|
|
|
|
|
class TestListen(tests.util.TestCase):
|
|
"""ListenBrainz Listen test case."""
|
|
|
|
def setUp(self):
|
|
"""Set up common variables."""
|
|
super().setUp()
|
|
self.library = self.sql.libraries.create(pathlib.Path("/a/b"))
|
|
self.artists = [self.sql.artists.create("Artist 1", mbid="mbid-ar1"),
|
|
self.sql.artists.create("Artist 2"),
|
|
self.sql.artists.create("Artist 3", mbid="mbid-ar3")]
|
|
self.album = self.sql.albums.create("Test Album", "Test Artist",
|
|
release="1988-06",
|
|
mbid="mbid-release")
|
|
self.medium = self.sql.media.create(self.album, "", number=1)
|
|
self.year = self.sql.years.create(1988)
|
|
self.track = self.sql.tracks.create(self.library,
|
|
pathlib.Path("/a/b/c.ogg"),
|
|
self.medium, self.year,
|
|
title="Track 1", number=1,
|
|
artist="Track Artist")
|
|
|
|
for artist in self.artists:
|
|
artist.add_track(self.track)
|
|
|
|
self.listen = emmental.listenbrainz.listen.Listen(self.track)
|
|
|
|
def test_init(self):
|
|
"""Test initializing our Listen instance."""
|
|
self.assertIsInstance(self.listen, liblistenbrainz.Listen)
|
|
self.assertEqual(self.listen.track_name, "Track 1")
|
|
self.assertEqual(self.listen.artist_name, "Track Artist")
|
|
self.assertEqual(self.listen.release_name, "Test Album")
|
|
self.assertEqual(self.listen.release_group_mbid, "mbid-release")
|
|
self.assertEqual(self.listen.tracknumber, 1)
|
|
self.assertDictEqual(self.listen.additional_info,
|
|
{"media_player": "emmental-debug"})
|
|
self.assertListEqual(self.listen.artist_mbids,
|
|
["mbid-ar1", "mbid-ar3"])
|
|
self.assertIsNone(self.listen.listened_at)
|
|
self.assertIsNone(self.listen.listenid)
|
|
|
|
utc_now = datetime.datetime.utcnow()
|
|
local_now = utc_now.replace(tzinfo=dateutil.tz.tzutc()).astimezone()
|
|
listen = emmental.listenbrainz.listen.Listen(self.track,
|
|
listenid=1234,
|
|
listened_at=utc_now)
|
|
self.assertEqual(listen.listenid, 1234)
|
|
self.assertEqual(listen.listened_at, local_now.timestamp())
|