class BookmarkDatabase(PersistenceLayer): def __init__(self): self.table_name = 'bookmarks' self.db = DatabaseManager('bookmarks.db') self.db.create_table(self.table_name, { 'id': 'integer primary key autoincrement', 'title': 'text not null', 'url': 'text not null', 'notes': 'text', 'date_added': 'text not null' }) def create(self, bookmark_data): self.db.add(self.table_name, bookmark_data) def list(self, order_by=None): return self.db.select(self.table_name, order_by=order_by).fetchall() def edit(self, bookmark_id, bookmark_data): self.db.update(self.table_name, {'id': bookmark_id}, bookmark_data) def delete(self, bookmark_id): self.db.delete(self.table_name, {'id': bookmark_id})
class BookmarkDatabase(PersistenceLayer): def __init__(self): self.table_name = 'bookmarks' self.db = DatabaseManager('bookmarks.db') self.db.create_table( 'bookmarks', { 'id': 'INTEGER PRIMARY KEY AUTOINCREMENT', 'title': 'TEXT NOT NULL', 'url': 'TEXT NOT NULL', 'notes': 'TEXT', 'date_added': 'TEXT NOT NULL' }) def create(self, bookmark_data): self.db.add(self.table_name, bookmark_data) def list(self, order_by=None): return self.db.select(self.table_name, order_by=order_by).fetchall() def delete(self, bookmark_id): self.db.delete(self.table_name, {'id': bookmark_id}) def update(self, bookmark_id, bookmark_data): self.db.update(self.table_name, {'id': bookmark_id}, bookmark_data)
def testadd(self): tablename = "BookMarks" data = Mock expected = Mock self.assertEqual(expected, DatabaseManager.add(self, tablename, data))