emmental/trackdb/test_track.py

# Copyright 2020 (c) Anna Schumaker.
from . import track
import datetime
import pathlib
import unittest

test_tracks  = pathlib.Path("./trier/Test Album")

class TestTrack(unittest.TestCase):
    def test_track_init(self):
        trak = track.Track(1, test_tracks / "01 - Test Track.ogg")
        self.assertEqual(trak.trackid, 1)
        self.assertEqual(trak.filepath, test_tracks / "01 - Test Track.ogg")
        self.assertEqual(trak.playcount, 0)
        self.assertIsNone(trak.lastplayed)
        self.assertIsInstance(trak.tags, dict)
        self.assertEqual(trak.length, 10.0)

    def test_track_played(self):
        trak = track.Track(1, test_tracks / "01 - Test Track.ogg")
        trak.played()
        self.assertEqual(trak.playcount, 1)
        self.assertEqual(trak.lastplayed, datetime.date.today())
        trak.played()
        self.assertEqual(trak.playcount, 2)