emmental/db/test_state.py

83 lines
2.6 KiB
Python

# Copyright 2021 (c) Anna Schumaker.
import db
import sqlite3
import unittest
from gi.repository import GObject
class TestPlaylistState(unittest.TestCase):
def test_init(self):
state = db.state.Table.insert()
self.assertIsInstance(state, GObject.GObject)
self.assertIsNotNone(state._plstateid)
self.assertEqual(state.rowid, state._plstateid)
def test_random(self):
state = db.state.Table.insert()
self.assertFalse(state._random)
self.assertFalse(state.get_property("random"))
state.random = True
self.assertTrue(state._random)
def test_loop(self):
state = db.state.Table.insert()
self.assertFalse(state._loop)
self.assertFalse(state.get_property("loop"))
state.loop = True
self.assertTrue(state._loop)
def test_current(self):
state = db.state.Table.insert()
self.assertEqual(state._current, -1)
self.assertEqual(state.current, -1)
state.current = 3
self.assertEqual(state._current, 3)
def test_sort(self):
state = db.state.Table.insert()
self.assertEqual(state._sort, ",".join(db.state.DefaultSort))
self.assertEqual(state.sort, db.state.DefaultSort)
state.sort = [ "test", "sort" ]
self.assertEqual(state._sort, "test,sort,tracks.trackid ASC" )
state.sort = [ ]
self.assertEqual(state._sort, "tracks.trackid ASC" )
class TestPlaylistStateTable(unittest.TestCase):
def setUp(self):
db.reset()
def test_init(self):
table = db.state.PlaylistStateTable()
self.assertIsInstance(table, db.table.Table)
self.assertEqual(table.table, "playlist_states")
self.assertIsInstance(db.state.Table, db.state.PlaylistStateTable)
db.sql.execute("SELECT plstateid,random,loop,current,sort "
"FROM playlist_states")
def test_default_sort(self):
self.assertEqual(db.state.DefaultSort[0], "artists.sort ASC")
self.assertEqual(db.state.DefaultSort[1], "albums.release ASC")
self.assertEqual(db.state.DefaultSort[2], "albums.sort ASC")
self.assertEqual(db.state.DefaultSort[3], "discs.number ASC")
self.assertEqual(db.state.DefaultSort[4], "tracks.number ASC")
def test_insert(self):
table = db.state.PlaylistStateTable()
state = table.insert()
self.assertFalse(state.random)
self.assertFalse(state.loop)
self.assertEqual(state.current, -1)
self.assertEqual(state.sort, db.state.DefaultSort)
def test_lookup(self):
with self.assertRaises(NotImplementedError):
db.state.Table.lookup()