🎨 Rename method to create replies for tests

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / data / SoneTest.kt

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 (file)
--- 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))
+       }
+
 }