def run(self, shelf=None):
"""Request a shelf audit."""
if not shelf:
raise base_action.MissingShelfError(
'Cannot send audit request. Task did not receive a shelf.')
send_email.send_shelf_audit_email(shelf)
shelf.audit_requested = True
shelf.put()
def run(self, **kwargs):
"""Request a shelf audit."""
shelf = kwargs.get('shelf')
if not shelf:
raise SendAuditError(
'Cannot send audit request. Task did not receive a shelf; only '
'kwargs: {}'.format(str(kwargs)))
send_email.send_shelf_audit_email(shelf)
shelf.audit_requested = True
shelf.put()
def tests_send_shelf_audit_email(self, mock_render, mock_sendemail,
mock_logging):
"""Test sending mail to users."""
shelf = shelf_model.Shelf.enroll(loanertest.USER_EMAIL, 'US-BLD', 24,
'Overlook Hotel', 40.6892534,
-74.0466891, 1.0,
loanertest.USER_EMAIL)
shelf.last_audit_time = datetime.datetime.utcnow()
shelf.put()
mock_render.return_value = ('Fake title', 'Fake body.')
send_email.config_model.Config.set(
'img_banner_primary', 'https://button.site/banners/primary.png')
send_email.send_shelf_audit_email(shelf)
assert mock_logging.info.called
assert mock_sendemail.called
assert mock_render.called