X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbumsTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FAlbumsTest.kt;h=6821a24e3ef5806ed3f7dbb3ae999205b8e46c12;hp=1f938a3124a6d84443c12dca1c388488ff5ee8f5;hb=e13cd46faa0404418176f8747d52e8160b2a7068;hpb=5473139d890c40c85712f59ba534583961c2772e

diff --git a/src/test/kotlin/net/pterodactylus/sone/data/AlbumsTest.kt b/src/test/kotlin/net/pterodactylus/sone/data/AlbumsTest.kt
index 1f938a3..6821a24 100644
--- a/src/test/kotlin/net/pterodactylus/sone/data/AlbumsTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/data/AlbumsTest.kt
@@ -114,4 +114,23 @@ class AlbumsTest {
 		assertThat(notEmpty(album), equalTo(true))
 	}
 
+	@Test
+	fun `allImages returns images from album`() {
+		val sone = IdOnlySone("sone")
+		val album = AlbumImpl(sone)
+		val image1 = createImage(sone, "1").also(album::addImage)
+		val image2 = createImage(sone, "2").also(album::addImage)
+		assertThat(album.allImages, contains(image1, image2))
+	}
+
+	@Test
+	fun `allImages returns images from subalbum`() {
+		val sone = IdOnlySone("sone")
+		val album1 = AlbumImpl(sone)
+		val album2 = AlbumImpl(sone).also(album1::addAlbum)
+		val image1 = createImage(sone, "1").also(album1::addImage)
+		val image2 = createImage(sone, "2").also(album2::addImage)
+		assertThat(album1.allImages, contains(image1, image2))
+	}
+
 }