def test_saving_and_retrieving_items(self): list_ = List() db_session.add(list_) db_session.commit() first_item = Item(text='첫 번째 아이템', list=list_.id) second_item = Item(text='두 번째 아이템', list=list_.id) db_session.add(first_item) db_session.add(second_item) db_session.commit() saved_items = Item.query.filter( or_(Item.text == '첫 번째 아이템', Item.text == '두 번째 아이템')).all() first_saved_item = saved_items[0] second_saved_item = saved_items[1] assert first_saved_item.text == '첫 번째 아이템' assert first_saved_item.list == list_.id assert second_saved_item.text == '두 번째 아이템' assert second_saved_item.list == list_.id Item.query.filter(Item.text == '첫 번째 아이템').delete() Item.query.filter(Item.text == '두 번째 아이템').delete() List.query.filter(List.id == list_.id).delete() db_session.commit()
def test_displays_only_items_for_that_list(self): correct_list = List() db_session.add(correct_list) db_session.commit() item_01 = Item(text='itemey 1', list=correct_list.id) item_02 = Item(text='itemey 2', list=correct_list.id) db_session.add(item_01) db_session.add(item_02) db_session.commit() other_list = List() db_session.add(other_list) db_session.commit() other_item_01 = Item(text='다른 목록 아이템 1', list=other_list.id) other_item_02 = Item(text='다른 목록 아이템 2', list=other_list.id) with app.test_request_context('/lists/%d' % correct_list.id): res = view_list(list_id=correct_list.id) assert item_01.text in res assert item_02.text in res assert other_item_01.text not in res assert other_item_02.text not in res Item.query.filter(Item.list == correct_list.id).delete() Item.query.filter(Item.list == other_list.id).delete() db_session.commit() List.query.filter(List.id == correct_list.id).delete() List.query.filter(List.id == other_list.id).delete() db_session.commit()
def new_list(): list_ = List() db_session.add(list_) db_session.commit() item = Item(text=request.form['item_text'], list=list_.id) db_session.add(item) db_session.commit() return redirect(url_for('base.view_list', list_id=list_.id))
def add_item(list_id): list_ = List.query.filter(List.id == list_id).first() item = Item(text=request.form['item_text'], list=list_.id) db_session.add(item) db_session.commit() return redirect(url_for('base.view_list', list_id=list_.id))