def test_get_one_or_else_multiple_results(self, mock): journo_1, _ = db_helper.init_journalist() journo_2, _ = db_helper.init_journalist() with mock.patch('logger') as mock_logger: get_one_or_else(Journalist.query, mock_logger, mock) mock_logger.error.assert_called() # Not specifying very long log line mock.assert_called_with(500)
def test_get_one_or_else_no_result_found(self, mock): query = Journalist.query.filter(Journalist.username == "alice") with mock.patch('logger') as mock_logger: get_one_or_else(query, mock_logger, mock) log_line = ('Found none when one was expected: ' 'No row was found for one()') mock_logger.error.assert_called_with(log_line) mock.assert_called_with(404)
def get_source(filesystem_id): """Return a Source object, representing the database row, for the source with the `filesystem_id`""" source = None query = Source.query.filter(Source.filesystem_id == filesystem_id) source = get_one_or_else(query, current_app.logger, abort) return source
def get_source(sid): """Return a Source object, representing the database row, for the source with id `sid`""" source = None query = Source.query.filter(Source.filesystem_id == sid) source = get_one_or_else(query, app.logger, abort) return source
def test_get_one_or_else_returns_one(self, mock): new_journo, _ = db_helper.init_journalist() query = Journalist.query.filter( Journalist.username == new_journo.username) with mock.patch('logger') as mock_logger: selected_journo = get_one_or_else(query, mock_logger, mock) self.assertEqual(new_journo, selected_journo)
def delete(): query = Reply.query.filter(Reply.filename == request.form['reply_filename']) reply = get_one_or_else(query, app.logger, abort) store.secure_unlink(store.path(g.sid, reply.filename)) db_session.delete(reply) db_session.commit() flash("Reply deleted", "notification") return redirect(url_for('lookup'))
def delete(): query = Reply.query.filter( Reply.filename == request.form['reply_filename']) reply = get_one_or_else(query, current_app.logger, abort) srm(store.path(g.filesystem_id, reply.filename)) db_session.delete(reply) db_session.commit() flash(gettext("Reply deleted"), "notification") return redirect(url_for('.lookup'))