def test_soft_time_limit(_):
session = UploadSessionFactory()
session.status = session.REQUEUED
session.save()
build_images(upload_session_pk=session.pk)
session.refresh_from_db()
assert session.status == session.FAILURE
assert session.error_message == "Time limit exceeded."
def test_failed_image_import_notification():
image = ["corrupt.png"]
session, _ = create_raw_upload_image_session(images=image)
build_images(upload_session_pk=session.pk)
session.refresh_from_db()
assert RawImageUploadSession.objects.count() == 1
assert is_following(
user=RawImageUploadSession.objects.get().creator,
obj=RawImageUploadSession.objects.get(),
)
assert Notification.objects.count() == 1
assert (Notification.objects.get().user ==
RawImageUploadSession.objects.get().creator)
def add_images_to_component_interface_value(*, component_interface_value_pk,
upload_pk):
session = RawImageUploadSession.objects.get(pk=upload_pk)
session.status = RawImageUploadSession.REQUEUED
session.save()
build_images(upload_session_pk=upload_pk)
if session.image_set.count() != 1:
error_message = "Image imports should result in a single image"
session.status = RawImageUploadSession.FAILURE
session.error_message = error_message
session.save()
raise ValueError(error_message)
civ = ComponentInterfaceValue.objects.get(pk=component_interface_value_pk)
civ.image = session.image_set.get()
civ.save()
civ.image.update_viewer_groups_permissions()