def test_database_exception_init(self):
with self.assertRaises(PlexCleanerException) as e:
db = Database(database_override='./tests/database/bad.db')
db.get_rows()
self.assertTrue('Could not open' in e.exception.message)
def test_database_exception_rows(self):
with self.assertRaises(PlexCleanerException) as e:
db = Database(database_override='./tests/database/empty.db')
db.get_rows()
self.assertTrue('Unabled to fetch' in e.exception.message)
def test_get_rows(self):
db = Database(database_override='./tests/database/com.plexapp.plugins.library.db')
rows = db.get_rows().fetchall()
self.assertEqual(len(rows), 98)
