def test_notification_es_mapping(dummy_notification, loan_validated_martigny):
"""Test notification elasticsearch mapping."""
search = NotificationsSearch()
mapping = get_mapping(search.Meta.index)
assert mapping
notif = deepcopy(dummy_notification)
validated_pid = loan_validated_martigny.get('pid')
loan_ref = f'https://bib.rero.ch/api/loans/{validated_pid}'
notif['context']['loan']['$ref'] = loan_ref
Notification.create(notif, dbcommit=True, delete_pid=True, reindex=True)
assert mapping == get_mapping(search.Meta.index)
def test_notification_create(es_clear, dummy_notification,
loan_validated_martigny, mailbox):
"""Test notification creation."""
notif = deepcopy(dummy_notification)
notif_data = {
'loan_url': 'https://ils.rero.ch/api/loans/',
'pid': loan_validated_martigny.get('pid')
}
loan_ref = '{loan_url}{pid}'.format(**notif_data)
notif['loan'] = {"$ref": loan_ref}
notification = Notification.create(notif,
dbcommit=True,
delete_pid=True,
reindex=True)
assert notification == notif
pid = notification.get('pid')
notification = Notification.get_record_by_pid(pid)
assert notification == notif
fetched_pid = fetcher(notification.id, notification)
assert fetched_pid.pid_value == pid
assert fetched_pid.pid_type == 'notif'
notification.dispatch(enqueue=False, verbose=True)
assert len(mailbox) == 1
def test_notification_es_mapping(dummy_notification, loan_validated_martigny):
"""Test notification elasticsearch mapping."""
search = NotificationsSearch()
mapping = get_mapping(search.Meta.index)
assert mapping
notif = deepcopy(dummy_notification)
notif_data = {
'loan_url': 'https://ils.rero.ch/api/loans/',
'pid': loan_validated_martigny.get('pid')
}
loan_ref = '{loan_url}{pid}'.format(**notif_data)
notif['loan'] = {"$ref": loan_ref}
Notification.create(notif, dbcommit=True, delete_pid=True, reindex=True)
assert mapping == get_mapping(search.Meta.index)
def test_notifications_post_put_delete(
client, dummy_notification, loan_validated_martigny, json_header):
"""Test record delete and update."""
record = deepcopy(dummy_notification)
del record['pid']
notif_data = {
'loan_url': 'https://ils.rero.ch/api/loans/',
'pid': loan_validated_martigny.get('pid')
}
loan_ref = '{loan_url}{pid}'.format(**notif_data)
record['loan'] = {"$ref": loan_ref}
notif = Notification.create(
record,
dbcommit=True,
reindex=True,
delete_pid=True
)
assert notif == record
flush_index(NotificationsSearch.Meta.index)
pid = notif.get('pid')
item_url = url_for('invenio_records_rest.notif_item', pid_value=pid)
list_url = url_for('invenio_records_rest.notif_list', q='pid:pid')
new_record = deepcopy(record)
# Create record / POST
new_record['pid'] = 'x'
res, data = postdata(
client,
'invenio_records_rest.notif_list',
new_record
)
assert res.status_code == 201
flush_index(NotificationsSearch.Meta.index)
# Check that the returned record matches the given data
assert data['metadata'] == new_record
res = client.get(item_url)
assert res.status_code == 200
data = get_json(res)
assert notif == data['metadata']
# Update record/PUT
data['notification_type'] = 'due_soon'
res = client.put(
item_url,
data=json.dumps(data),
headers=json_header
)
assert res.status_code == 200
# assert res.headers['ETag'] != '"{}"'.format(librarie.revision_id)
# Check that the returned record matches the given data
data = get_json(res)
assert data['metadata']['notification_type'] == 'due_soon'
res = client.get(item_url)
assert res.status_code == 200
data = get_json(res)
assert data['metadata']['notification_type'] == 'due_soon'
res = client.get(list_url)
assert res.status_code == 200
# Delete record/DELETE
res = client.delete(item_url)
assert res.status_code == 204
res = client.get(item_url)
assert res.status_code == 410
links = notif.get_links_to_me()
assert links == {}
assert notif.can_delete
reasons = notif.reasons_not_to_delete()
assert reasons == {}
notif.delete(dbcommit=True, delindex=True)
def test_cancel_notifications(
client, patron_martigny, lib_martigny, item_lib_martigny,
librarian_martigny, loc_public_martigny, circulation_policies, mailbox
):
"""Test cancel notifications."""
login_user_via_session(client, librarian_martigny.user)
# CREATE and VALIDATE a request ...
res, data = postdata(
client,
'api_item.librarian_request',
dict(
item_pid=item_lib_martigny.pid,
pickup_location_pid=loc_public_martigny.pid,
patron_pid=patron_martigny.pid,
transaction_library_pid=lib_martigny.pid,
transaction_user_pid=librarian_martigny.pid
)
)
assert res.status_code == 200
request_loan_pid = data.get(
'action_applied')[LoanAction.REQUEST].get('pid')
flush_index(NotificationsSearch.Meta.index)
res, data = postdata(
client,
'api_item.validate_request',
dict(
pid=request_loan_pid,
transaction_location_pid=loc_public_martigny.pid,
transaction_user_pid=librarian_martigny.pid
)
)
assert res.status_code == 200
# At this time, an AVAILABILITY notification should be create but not yet
# dispatched
loan = Loan.get_record_by_pid(request_loan_pid)
notification = get_notification(loan, NotificationType.AVAILABILITY)
assert notification \
and notification['status'] == NotificationStatus.CREATED
# BORROW the requested item
res, data = postdata(
client,
'api_item.checkout',
dict(
item_pid=item_lib_martigny.pid,
patron_pid=patron_martigny.pid,
transaction_location_pid=loc_public_martigny.pid,
transaction_user_pid=librarian_martigny.pid,
)
)
assert res.status_code == 200
loan_pid = data.get(
'action_applied')[LoanAction.CHECKOUT].get('pid')
loan = Loan.get_record_by_pid(loan_pid)
# Try to dispatch pending availability notifications.
# As the item is now checkout, then the availability notification is not
# yet relevant.
mailbox.clear()
process_notifications(NotificationType.AVAILABILITY)
notification = get_notification(loan, NotificationType.AVAILABILITY)
assert notification and \
notification['status'] == NotificationStatus.CANCELLED
assert len(mailbox) == 0
# restore to initial state
res, data = postdata(
client,
'api_item.checkin',
dict(
item_pid=item_lib_martigny.pid,
# patron_pid=patron_martigny.pid,
transaction_location_pid=loc_public_martigny.pid,
transaction_user_pid=librarian_martigny.pid,
)
)
assert res.status_code == 200
mailbox.clear()
# Test REMINDERS notifications.
# reminders notification check about the end_date. As the loan end_date
# is not yet over, the notification could be cancelled.
notification = loan.create_notification(_type=NotificationType.DUE_SOON)[0]
can_cancel, _ = notification.can_be_cancelled()
assert not can_cancel
process_notifications(NotificationType.DUE_SOON)
notification = Notification.get_record_by_pid(notification.pid)
assert notification['status'] == NotificationStatus.DONE
flush_index(NotificationsSearch.Meta.index)
# try to create a new DUE_SOON notification for the same loan
record = {
'creation_date': datetime.now(timezone.utc).isoformat(),
'notification_type': NotificationType.DUE_SOON,
'context': {
'loan': {'$ref': get_ref_for_pid('loans', loan.pid)},
'reminder_counter': 0
}
}
notification = Notification.create(record)
can_cancel, _ = notification.can_be_cancelled()
assert can_cancel
Dispatcher.dispatch_notifications([notification.pid])
notification = Notification.get_record_by_pid(notification.pid)
assert notification['status'] == NotificationStatus.CANCELLED
def test_notifications_post_put_delete(
client, dummy_notification, loan_validated_martigny, json_header):
"""Test record delete and update."""
record = deepcopy(dummy_notification)
del record['pid']
loan_ref = get_ref_for_pid('loans', loan_validated_martigny.get('pid'))
record['context']['loan'] = {'$ref': loan_ref}
notif = Notification.create(
record,
dbcommit=True,
reindex=True,
delete_pid=True
)
assert notif == record
flush_index(NotificationsSearch.Meta.index)
pid = notif.get('pid')
item_url = url_for('invenio_records_rest.notif_item', pid_value=pid)
list_url = url_for('invenio_records_rest.notif_list', q='pid:pid')
new_record = deepcopy(record)
# Create record / POST
new_record['pid'] = 'x'
res, data = postdata(
client,
'invenio_records_rest.notif_list',
new_record
)
assert res.status_code == 201
flush_index(NotificationsSearch.Meta.index)
# Check that the returned record matches the given data
assert data['metadata'] == new_record
res = client.get(item_url)
assert res.status_code == 200
data = get_json(res)
assert notif == data['metadata']
# Update record/PUT
data = data['metadata']
data['notification_type'] = NotificationType.DUE_SOON
res = client.put(
item_url,
data=json.dumps(data),
headers=json_header
)
assert res.status_code == 200
# Check that the returned record matches the given data
data = get_json(res)
assert data['metadata']['notification_type'] == NotificationType.DUE_SOON
res = client.get(item_url)
assert res.status_code == 200
data = get_json(res)
assert data['metadata']['notification_type'] == NotificationType.DUE_SOON
res = client.get(list_url)
assert res.status_code == 200
# Delete record/DELETE
res = client.delete(item_url)
assert res.status_code == 204
res = client.get(item_url)
assert res.status_code == 410
can, reasons = notif.can_delete
assert can
assert reasons == {}
notif.delete(dbcommit=True, delindex=True)