X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FSoneTest.kt;h=9e087814523d4150a020ec6eb0b6feaf9626658e;hb=5076d8379e6413189273defd4f01f241b4a6a57e;hp=42980eeb12a7a242f75a0ec459a625aa73bc0c95;hpb=5473139d890c40c85712f59ba534583961c2772e;p=Sone.git

diff --git a/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt b/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
index 42980ee..9e08781 100644
--- a/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
@@ -98,11 +98,11 @@ class SoneTest {
 	fun `post count comparator compares replies if posts are not different`() {
 		val sone1 = object : IdOnlySone("1") {
 			override fun getPosts() = listOf(createPost(), createPost())
-			override fun getReplies() = setOf(emptyPostReply(), emptyPostReply())
+			override fun getReplies() = setOf(createPostReply(), createPostReply())
 		}
 		val sone2 = object : IdOnlySone("2") {
 			override fun getPosts() = listOf(createPost(), createPost())
-			override fun getReplies() = setOf(emptyPostReply(), emptyPostReply(), emptyPostReply())
+			override fun getReplies() = setOf(createPostReply(), createPostReply(), createPostReply())
 		}
 		assertThat(postCountComparator.compare(sone1, sone2), greaterThan(0))
 	}
@@ -111,11 +111,11 @@ class SoneTest {
 	fun `post count comparator sorts sone with same amount of posts and replies as equal`() {
 		val sone1 = object : IdOnlySone("1") {
 			override fun getPosts() = listOf(createPost(), createPost())
-			override fun getReplies() = setOf(emptyPostReply(), emptyPostReply())
+			override fun getReplies() = setOf(createPostReply(), createPostReply())
 		}
 		val sone2 = object : IdOnlySone("2") {
 			override fun getPosts() = listOf(createPost(), createPost())
-			override fun getReplies() = setOf(emptyPostReply(), emptyPostReply())
+			override fun getReplies() = setOf(createPostReply(), createPostReply())
 		}
 		assertThat(postCountComparator.compare(sone1, sone2), equalTo(0))
 	}
@@ -154,4 +154,19 @@ class SoneTest {
 		assertThat(sone.allAlbums, contains<Album>(album1, album11, album2))
 	}
 
+	@Test
+	fun `allImages returns all images of a Sone`() {
+		val sone = object : IdOnlySone("1") {
+			private val rootAlbum = AlbumImpl(this)
+			override fun getRootAlbum() = rootAlbum
+		}
+		val album1 = AlbumImpl(sone).also(sone.rootAlbum::addAlbum)
+		val album11 = AlbumImpl(sone).also(album1::addAlbum)
+		val album2 = AlbumImpl(sone).also(sone.rootAlbum::addAlbum)
+		val image1 = createImage(sone).also(album1::addImage)
+		val image11 = createImage(sone).also(album11::addImage)
+		val image2 = createImage(sone).also(album2::addImage)
+		assertThat(sone.allImages, containsInAnyOrder(image1, image11, image2))
+	}
+
 }