X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FSoneTest.kt;h=8866b5128e3ce6a570f608045c8e7ffc99668137;hp=5cc0ee2ccc862031b55459f45861c705908b2d02;hb=e03a570dffcf1277952bc03b5e1fb54c50e4394a;hpb=9b0b5b760adf41d52603a4b65f3e5f3220da28eb

diff --git a/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt b/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
index 5cc0ee2..8866b51 100644
--- a/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/data/SoneTest.kt
@@ -29,6 +29,39 @@ import kotlin.test.*
 class SoneTest {
 
 	@Test
+	fun `nice name comparator correctly compares Sones by their nice name`() {
+		val sone1 = object : IdOnlySone("1") {
+			override fun getProfile() = Profile(this).apply { firstName = "Left" }
+		}
+		val sone2 = object : IdOnlySone("2") {
+			override fun getProfile() = Profile(this).apply { firstName = "Right" }
+		}
+		assertThat(niceNameComparator.compare(sone1, sone2), lessThan(0))
+	}
+
+	@Test
+	fun `nice name comparator correctly compares Sones by their ID if nice name is the same`() {
+		val sone1 = object : IdOnlySone("1") {
+			override fun getProfile() = Profile(this).apply { firstName = "Left" }
+		}
+		val sone2 = object : IdOnlySone("2") {
+			override fun getProfile() = Profile(this).apply { firstName = "Left" }
+		}
+		assertThat(niceNameComparator.compare(sone1, sone2), lessThan(0))
+	}
+
+	@Test
+	fun `nice name comparator treats Sones as equal if nice name and ID are the same`() {
+		val sone1 = object : IdOnlySone("1") {
+			override fun getProfile() = Profile(this).apply { firstName = "Left" }
+		}
+		val sone2 = object : IdOnlySone("1") {
+			override fun getProfile() = Profile(this).apply { firstName = "Left" }
+		}
+		assertThat(niceNameComparator.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())