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)