test image selection

Author: Tschuppi81

tests/onegov/org/test_views_photoalbum.py

@@ -1,12 +1,15 @@
 from __future__ import annotations
 
+import os
 import re
 
+from onegov.org.models import ImageFileCollection
+
 from tests.shared.utils import create_image
 from webtest import Upload
 
-
 from typing import TYPE_CHECKING
+
 if TYPE_CHECKING:
     from .conftest import Client
 
@@ -54,3 +57,44 @@ def test_manage_album(client: Client) -> None:
 
     album = albums.click("Comicon 2016")
     assert "This is an alt text" in album
+
+
+def test_image_selection(client: Client) -> None:
+    client.login_admin()
+
+    number_of_images = 3
+
+    albums = client.get('/photoalbums')
+    new = albums.click('Fotoalbum')
+    new.form['title'] = "Vacation Destinations 2026"
+    new.form.submit()
+
+    album = client.get('/photoalbums').click("Vacation Destinations 2026")
+    for i in range(number_of_images):
+        images = albums.click("Bilder verwalten")
+        images.form['file'] = [Upload(f'image_{i}.jpg', create_image().read())]
+        images.form.submit()
+
+    # select all images
+    select = album.click("Bilder auswählen")
+    select.form[tuple(select.form.fields.keys())[1]] = True
+    select.form[tuple(select.form.fields.keys())[2]] = True
+    select.form[tuple(select.form.fields.keys())[3]] = True
+    select.form.submit()
+
+    images = ImageFileCollection(client.app.session()).query().all()
+    assert len(images) == number_of_images
+    images = {(i.id, i.name) for i in images}
+    print(images)
+    album = client.get('/photoalbums').click("Vacation Destinations 2026")
+    for i in images:
+        assert i[0] in album, '{} not found in album'.format(i)
+
+    # switch to grid mode
+    settings = album.click("Bearbeiten")
+    settings.form['view'] = 'grid'
+    settings.form.submit()
+
+    album = client.get('/photoalbums').click("Vacation Destinations 2026")
+    for i in images:
+        assert i[0] in album, '{} not found in album'.format(i)