🎨 Replace image count comparator with Kotlin version

[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 8866b51..1429e54 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
@@ -62,6 +62,28 @@ class SoneTest {
        }
 
        @Test
+       fun `last activity comparator correctly compares Sones by last activity`() {
+               val sone1 = object : IdOnlySone("1") {
+                       override fun getTime() = 1000L
+               }
+               val sone2 = object : IdOnlySone("2") {
+                       override fun getTime() = 2000L
+               }
+               assertThat(lastActivityComparator.compare(sone1, sone2), greaterThan(0))
+       }
+
+       @Test
+       fun `last activity comparator treats Sones as equal if last activity is the same`() {
+               val sone1 = object : IdOnlySone("1") {
+                       override fun getTime() = 1000L
+               }
+               val sone2 = object : IdOnlySone("2") {
+                       override fun getTime() = 1000L
+               }
+               assertThat(lastActivityComparator.compare(sone1, sone2), equalTo(0))
+       }
+
+       @Test
        fun `post count comparator sorts sones with different number of posts correctly`() {
                val sone1 = object : IdOnlySone("1") {
                        override fun getPosts() = listOf(createPost(), createPost())
@@ -98,4 +120,26 @@ class SoneTest {
                assertThat(postCountComparator.compare(sone1, sone2), equalTo(0))
        }
 
+       @Test
+       fun `image count comparator sorts Sones correctly if number of images is different`() {
+               val sone1 = object : IdOnlySone("1") {
+                       override fun getRootAlbum() = AlbumImpl(this).also { it.addImage(createImage(this)) }
+               }
+               val sone2 = object : IdOnlySone("2") {
+                       override fun getRootAlbum() = AlbumImpl(this).also { it.addImage(createImage(this)); it.addImage(createImage(this)) }
+               }
+               assertThat(imageCountComparator.compare(sone1, sone2), greaterThan(0))
+       }
+
+       @Test
+       fun `image count comparator treats Sones as equal if number of images is the same`() {
+               val sone1 = object : IdOnlySone("1") {
+                       override fun getRootAlbum() = AlbumImpl(this).also { it.addImage(createImage(this)) }
+               }
+               val sone2 = object : IdOnlySone("2") {
+                       override fun getRootAlbum() = AlbumImpl(this).also { it.addImage(createImage(this)) }
+               }
+               assertThat(imageCountComparator.compare(sone1, sone2), equalTo(0))
+       }
+
 }