214 lines
9.1 KiB
Python
214 lines
9.1 KiB
Python
# Copyright 2023 (c) Anna Schumaker.
|
|
"""Tests our PlaylistModel."""
|
|
import pathlib
|
|
import unittest.mock
|
|
import tests.util
|
|
import emmental.playlist.playlist
|
|
from gi.repository import GObject
|
|
|
|
|
|
class TestPlaylist(tests.util.TestCase):
|
|
"""Tests the Playlist Gio.ListModel."""
|
|
|
|
def setUp(self):
|
|
"""Set up common variables."""
|
|
super().setUp()
|
|
self.db_plist = self.sql.playlists.create("Test Playlist")
|
|
self.playlist = emmental.playlist.playlist.Playlist(self.sql)
|
|
|
|
self.library = self.sql.libraries.create(pathlib.Path("/a/b"))
|
|
self.album = self.sql.albums.create("Test Album", "Artist", "2023")
|
|
self.medium = self.sql.media.create(self.album, "", number=1)
|
|
self.year = self.sql.years.create(2023)
|
|
|
|
self.track1 = self.sql.tracks.create(self.library,
|
|
pathlib.Path("/a/b/1.ogg"),
|
|
self.medium, self.year,
|
|
number=1, length=10)
|
|
self.track2 = self.sql.tracks.create(self.library,
|
|
pathlib.Path("/a/b/2.ogg"),
|
|
self.medium, self.year,
|
|
number=2, length=20)
|
|
self.track3 = self.sql.tracks.create(self.library,
|
|
pathlib.Path("/a/b/3.ogg"),
|
|
self.medium, self.year,
|
|
number=3, length=30)
|
|
|
|
def test_property_flags(self):
|
|
"""Test the FLAGS constant."""
|
|
self.assertEqual(emmental.playlist.playlist.FLAGS,
|
|
GObject.ParamFlags.READWRITE |
|
|
GObject.ParamFlags.EXPLICIT_NOTIFY)
|
|
|
|
def test_init(self):
|
|
"""Test that the Playlist was set up correctly."""
|
|
self.assertIsInstance(self.playlist,
|
|
emmental.playlist.model.TrackidModel)
|
|
self.assertEqual(self.playlist.sql, self.sql)
|
|
self.assertDictEqual(self.playlist._Playlist__sort_keys, {})
|
|
|
|
self.assertIsNone(self.playlist._Playlist__playlist)
|
|
self.assertIsNone(self.playlist._Playlist__sort_order)
|
|
|
|
plist2 = emmental.playlist.playlist.Playlist(self.sql, self.db_plist)
|
|
self.assertEqual(plist2.playlist, self.db_plist)
|
|
|
|
def test_add_track(self):
|
|
"""Test the playlist add_track() function."""
|
|
self.playlist.add_track(self.track1)
|
|
|
|
self.playlist.playlist = self.db_plist
|
|
self.playlist.add_track(self.track1)
|
|
self.assertListEqual(self.playlist.trackids, [self.track1.trackid])
|
|
|
|
def test_index(self):
|
|
"""Test the playlist index() function."""
|
|
self.assertIsNone(self.playlist.index(self.track1))
|
|
|
|
self.playlist.playlist = self.db_plist
|
|
self.playlist.add_track(self.track1)
|
|
|
|
self.assertEqual(self.playlist.index(self.track1), 0)
|
|
self.assertIsNone(self.playlist.index(self.track2))
|
|
self.assertIsNone(super(type(self.playlist), self.playlist).index(0))
|
|
self.assertIsNone(self.playlist.index(None))
|
|
|
|
def test_move_track_down(self):
|
|
"""Test the playlist move_track_down() function."""
|
|
self.playlist.move_track_down(self.track1)
|
|
|
|
self.db_plist.add_track(self.track1)
|
|
self.db_plist.add_track(self.track2)
|
|
self.db_plist.add_track(self.track3)
|
|
self.playlist.playlist = self.db_plist
|
|
|
|
items_changed = unittest.mock.Mock()
|
|
self.playlist.connect("items-changed", items_changed)
|
|
|
|
self.playlist.move_track_down(self.track2)
|
|
self.assertEqual(self.db_plist.sort_order, "user")
|
|
self.assertDictEqual(self.playlist._Playlist__sort_keys,
|
|
{self.track1.trackid: 0,
|
|
self.track3.trackid: 1,
|
|
self.track2.trackid: 2})
|
|
self.assertListEqual(self.playlist.trackids, [self.track1.trackid,
|
|
self.track3.trackid,
|
|
self.track2.trackid])
|
|
items_changed.assert_called_once_with(self.playlist, 0, 3, 3)
|
|
|
|
items_changed.reset_mock()
|
|
self.playlist.move_track_down(self.track1)
|
|
self.assertDictEqual(self.playlist._Playlist__sort_keys,
|
|
{self.track3.trackid: 0,
|
|
self.track1.trackid: 1,
|
|
self.track2.trackid: 2})
|
|
self.assertListEqual(self.playlist.trackids, [self.track3.trackid,
|
|
self.track1.trackid,
|
|
self.track2.trackid])
|
|
items_changed.assert_called_once_with(self.playlist, 0, 2, 2)
|
|
|
|
def test_move_track_up(self):
|
|
"""Test the playlist move_track_up() function."""
|
|
self.playlist.move_track_up(self.track1)
|
|
|
|
self.db_plist.add_track(self.track1)
|
|
self.db_plist.add_track(self.track2)
|
|
self.db_plist.add_track(self.track3)
|
|
self.playlist.playlist = self.db_plist
|
|
|
|
items_changed = unittest.mock.Mock()
|
|
self.playlist.connect("items-changed", items_changed)
|
|
|
|
self.playlist.move_track_up(self.track2)
|
|
self.assertEqual(self.db_plist.sort_order, "user")
|
|
self.assertDictEqual(self.playlist._Playlist__sort_keys,
|
|
{self.track2.trackid: 0,
|
|
self.track1.trackid: 1,
|
|
self.track3.trackid: 2})
|
|
self.assertListEqual(self.playlist.trackids, [self.track2.trackid,
|
|
self.track1.trackid,
|
|
self.track3.trackid])
|
|
items_changed.assert_called_once_with(self.playlist, 0, 3, 3)
|
|
|
|
items_changed.reset_mock()
|
|
self.playlist.move_track_up(self.track3)
|
|
self.assertDictEqual(self.playlist._Playlist__sort_keys,
|
|
{self.track2.trackid: 0,
|
|
self.track3.trackid: 1,
|
|
self.track1.trackid: 2})
|
|
self.assertListEqual(self.playlist.trackids, [self.track2.trackid,
|
|
self.track3.trackid,
|
|
self.track1.trackid])
|
|
items_changed.assert_called_once_with(self.playlist, 1, 2, 2)
|
|
|
|
def test_remove_track(self):
|
|
"""Test the playlist remove_track() function."""
|
|
self.playlist.remove_track(self.track1)
|
|
|
|
self.db_plist.add_track(self.track1)
|
|
self.playlist.playlist = self.db_plist
|
|
self.playlist.remove_track(self.track1)
|
|
self.assertListEqual(self.playlist.trackids, [])
|
|
|
|
def test_playlist(self):
|
|
"""Test the playlist property."""
|
|
self.assertIsNone(self.playlist.playlist)
|
|
|
|
self.playlist.playlist = self.db_plist
|
|
self.assertEqual(self.playlist._Playlist__playlist, self.db_plist)
|
|
self.assertEqual(self.playlist.playlist, self.db_plist)
|
|
self.assertEqual(self.playlist.trackid_set, self.db_plist.tracks)
|
|
self.assertEqual(self.playlist.sort_order, self.db_plist.sort_order)
|
|
|
|
self.playlist.playlist = None
|
|
self.assertIsNone(self.playlist.playlist)
|
|
self.assertIsNone(self.playlist.trackid_set)
|
|
|
|
def test_sort_order(self):
|
|
"""Test the sort-order property."""
|
|
notify = unittest.mock.Mock()
|
|
self.playlist.connect("notify::sort-order", notify)
|
|
|
|
self.assertIsNone(self.playlist.sort_order)
|
|
self.db_plist.add_track(self.track1)
|
|
self.db_plist.add_track(self.track2)
|
|
self.db_plist.add_track(self.track3)
|
|
|
|
self.playlist.playlist = self.db_plist
|
|
self.assertEqual(self.playlist.sort_order, self.db_plist.sort_order)
|
|
notify.assert_called()
|
|
|
|
self.playlist.sort_order = "length DESC"
|
|
self.assertEqual(self.db_plist.sort_order, "length DESC")
|
|
self.assertDictEqual(self.playlist._Playlist__sort_keys,
|
|
{self.track3.trackid: 0,
|
|
self.track2.trackid: 1,
|
|
self.track1.trackid: 2})
|
|
self.assertListEqual(self.playlist.trackids,
|
|
[self.track3.trackid,
|
|
self.track2.trackid,
|
|
self.track1.trackid])
|
|
|
|
self.db_plist.remove_track(self.track2)
|
|
self.assertDictEqual(self.playlist._Playlist__sort_keys,
|
|
{self.track3.trackid: 0,
|
|
self.track1.trackid: 2})
|
|
|
|
notify.reset_mock()
|
|
self.db_plist.sort_order = "user"
|
|
self.assertEqual(self.playlist.sort_order, "user")
|
|
notify.assert_called()
|
|
|
|
notify.reset_mock()
|
|
self.playlist.playlist = None
|
|
notify.assert_called()
|
|
|
|
notify.reset_mock()
|
|
self.db_plist.sort_order = "length"
|
|
self.assertIsNone(self.playlist.sort_order)
|
|
notify.assert_not_called()
|
|
|
|
self.playlist.sort_order = "length"
|
|
self.assertIsNone(self.playlist.sort_order)
|
|
notify.assert_not_called()
|