def test_delete_document(self): new_doc_id = dm.create_new_document(self.template.id, {'name': 'foobar'}) documents = dm.get_all_documents() self.assertEquals(len(documents), 2) dm.delete_document(self.document) documents = dm.get_all_documents() self.assertEquals(len(documents), 1) self.assertEquals(documents[0].id, new_doc_id)
def edit_in_progress_document(document_id): ''' Allows updating and deleting of individual documents. ''' document_base = dm.get_single_document(document_id) if request.args.get('method') == 'DELETE': if dm.delete_document(document_base): return redirect(url_for('generator.in_progress_documents')) return abort(403)