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_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 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 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'))
