def test__delete_media_states(self):
root, containers, media = create_test_library()
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.create_media_table()
db.update_media(media)
db.update_containers(containers)
med = media[0]
db._delete_media_states([(med.id(), med.played())])
sql = f'select count() from media_states'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, len(media) - 1)
db._delete_media_states([(m.id(), m.played()) for m in media])
sql = f'select count() from media_states'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, 0)
db.close()
def test_delete_media(self):
root, containers, media = create_test_library()
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.create_media_table()
db.update_media(media)
db.update_containers(containers)
med = media[0]
db.delete_media([med])
sql = f'select count() from media'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, len(media) - 1)
db.delete_media(media)
sql = f'select count() from media'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, 0)
db.close()
def test__delete_identifiables(self):
root, containers, media = create_test_library()
identifiables = [
i for i in (containers + media) if isinstance(i, Identifiable)
]
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.update_containers(containers)
db.create_media_table()
db.update_media(media)
ide = identifiables[0]
db._delete_identifiables([ide])
sql = f'select count() from identifiables'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, len(identifiables) - 1)
db._delete_identifiables(identifiables)
sql = f'select count() from identifiables'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, 0)
db.close()
def test_create_media_table(self):
db = Sqlite()
db.connect(database=TMP_DB)
table_name = "media"
identifiables_table_name = "identifiables"
media_states_table_name = "media_states"
table_schema = remove_double_spaces("""(
id TEXT,
type TEXT,
parent_id TEXT,
file_path TEXT,
subtitles TEXT,
episode_number INTEGER,
title TEXT,
flags TEXT
)""")
db.create_media_table()
count = table_count(db, table_name, table_schema)
identifiables_count = table_count(db, identifiables_table_name)
media_states_count = table_count(db, media_states_table_name)
db.close()
self.assertEqual(count, 1)
self.assertEqual(identifiables_count, 1)
self.assertEqual(media_states_count, 1)
def test__update_media_states(self):
root, containers, media = create_test_library()
meta = create_test_meta()
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.create_media_table()
db.update_containers(containers)
db.update_media(media)
sql = f'select count() from media_states'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, len(media))
for med in media:
db_media = db.get_media(med.id())
self.assertTrue(testutils.compare_media_states(med, db_media))
# dublicates
a_media = media[0]
db.update_media([a_media])
sql = f'select count() from media_states where media_id=?'
cur = db.conn.cursor()
cur.execute(sql, (a_media.id(), ))
count = cur.fetchone()[0]
self.assertEqual(count, 1)
a_media = media[0]
if a_media:
a_media.set_played(not a_media.played())
db.update_media([a_media])
db_media = db.get_media(a_media.id())
self.assertTrue(testutils.compare_media_states(a_media, db_media))
a_media.set_played(not a_media.played())
db.update_media([a_media], overwrite_media_states=False)
db_media = db.get_media(a_media.id())
self.assertFalse(testutils.compare_media_states(a_media, db_media))
db.close()
def test_watchlist(self):
root, containers, media = create_test_library()
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.create_media_table()
db.update_media(media)
db.update_containers(containers)
db.create_watchlist_table()
container = containers[0]
CONTAINER_ID = container.id()
self.assertFalse(db.is_in_watchlists(CONTAINER_ID))
db.add_to_watchlist([CONTAINER_ID])
self.assertTrue(db.is_in_watchlists(CONTAINER_ID))
list = db.get_watchlist()
self.assertEqual(len(list), 1)
self.assertTrue(testutils.compare_containers(list[0], container))
db.remove_from_watchlist([CONTAINER_ID])
self.assertFalse(db.is_in_watchlists(CONTAINER_ID))
db.add_to_watchlist([containers[0].id(), containers[1].id()])
self.assertEqual(len(db.get_watchlist()), 2)
db.remove_all_from_watchlist()
self.assertEqual(len(db.get_watchlist()), 0)
db.add_to_watchlist([CONTAINER_ID])
self.assertTrue(db.is_in_watchlists(CONTAINER_ID))
db.delete_containers([container])
self.assertIsNone(db.get_container(CONTAINER_ID))
list = db.get_watchlist()
self.assertEqual(len(list), 0)
self.assertFalse(db.is_in_watchlists(CONTAINER_ID))
db.close()
def test_set_played(self):
root, containers, media = create_test_library()
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.create_media_table()
db.update_media(media)
db.update_containers(containers)
self.assertEqual(db.get_unplayed_count(root.id()), len(media))
db.set_played(root.id(), True)
self.assertEqual(db.get_unplayed_count(root.id()), 0)
db.set_played(root.id(), False)
self.assertEqual(db.get_unplayed_count(root.id()), len(media))
db.close()
db.connect()
def test__update_identifiables(self):
root, containers, media = create_test_library()
identifiables = [
i for i in (containers + media) if isinstance(i, Identifiable)
]
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.update_containers(containers)
db.create_media_table()
db.update_media(media)
db.update_meta(create_test_meta())
sql = f'select count() from identifiables'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, len(identifiables))
for ide in identifiables:
if isinstance(ide, Container):
db_identifiable = db.get_container(ide.id())
else:
db_identifiable = db.get_media(ide.id())
self.assertTrue(
testutils.compare_identifiables(ide, db_identifiable))
# dublicates
ide = identifiables[0]
db._update_identifiables([ide])
sql = f'select count() from identifiables where id=?'
cur = db.conn.cursor()
cur.execute(sql, (ide.id(), ))
count = cur.fetchone()[0]
self.assertEqual(count, 1)
an_identifiable = identifiables[0]
if an_identifiable:
year = an_identifiable.year() + 1 if an_identifiable.year() else 1
an_identifiable.set_year(year)
an_identifiable.ext_ids()["foo"] = "bar"
if isinstance(an_identifiable, Container):
db.update_containers([an_identifiable])
db_identifiable = db.get_container(an_identifiable.id())
else:
db.update_media([an_identifiable])
db_identifiable = db.get_media(an_identifiable.id())
self.assertTrue(
testutils.compare_identifiables(an_identifiable,
db_identifiable))
db.close()
def test_update_and_get_media(self):
root, containers, media = create_test_library()
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.create_media_table()
db.update_containers(containers)
db.update_meta(create_test_meta())
db.update_media(media)
sql = f'select count() from media'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, len(media))
for med in media:
db_media = db.get_media(med.id())
self.assertTrue(testutils.compare_media(med, db_media))
# dublicates
med = media[0]
db.update_media([med])
sql = f'select count() from media where id=?'
cur = db.conn.cursor()
cur.execute(sql, (med.id(), ))
count = cur.fetchone()[0]
self.assertEqual(count, 1)
a_movie = None
for med in media:
if isinstance(med, Movie):
a_movie = med
break
if a_movie:
a_movie._file_path = a_movie.file_path() + "foo"
a_movie._title = a_movie.title() + "foo"
a_movie._played = not a_movie._played
a_movie._parent = None
db.update_media([a_movie])
db_movie = db.get_media(a_movie.id())
self.assertTrue(testutils.compare_media(a_movie, db_movie))
an_episode = None
for med in media:
if isinstance(med, Episode):
an_episode = med
break
if an_episode:
an_episode._file_path = an_episode.file_path() + "foo"
an_episode._episode_number = an_episode.episode_number() + 1
an_episode._played = not an_episode._played
an_episode._parent = None
an_episode._is_oad = not an_episode.is_oad()
an_episode._is_ncop = not an_episode.is_ncop()
an_episode._is_nced = not an_episode.is_nced()
an_episode._is_ova = not an_episode.is_ova()
db.update_media([an_episode])
db_episode = db.get_media(an_episode.id())
self.assertTrue(testutils.compare_media(an_episode, db_episode))
db.close()
def test_update_and_get_containers(self):
root, containers, media = create_test_library()
db = Sqlite()
db.connect(database=TMP_DB)
db.create_containers_table()
db.create_media_table()
db.update_media(media)
db.update_meta(create_test_meta())
db.update_containers(containers)
sql = f'select count() from containers'
cur = db.conn.cursor()
cur.execute(sql)
count = cur.fetchone()[0]
self.assertEqual(count, len(containers))
for con in containers:
db_container = db.get_container(con.id())
self.assertTrue(testutils.compare_containers(con, db_container))
# dublicates
con = containers[0]
db.update_containers([con])
sql = f'select count() from containers where id=?'
cur = db.conn.cursor()
cur.execute(sql, (con.id(), ))
count = cur.fetchone()[0]
self.assertEqual(count, 1)
a_season = None
for con in containers:
if isinstance(con, Season):
a_season = con
break
if a_season:
a_season._path = a_season.path() + "foo"
a_season._show_name = a_season.show_name() + "foo"
a_season._season_number = a_season.season_number() + 1
a_season._parent = None
if (len(a_season.containers)):
a_season.containers.clear()
else:
for c in containers:
if isinstance(c, Extra):
a_season.containers.append(c)
c.set_parent(a_season)
break
if (len(a_season.media)):
a_season.media.clear()
else:
for m in media:
if isinstance(m, Episode):
a_season.media.append(m)
m.set_parent(a_season)
break
db.update_containers([a_season])
db_container = db.get_container(a_season.id())
self.assertTrue(
testutils.compare_containers(a_season, db_container))
db.close()