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))
+ }
+
+ @Test
+ fun `allAlbums returns all albums of a Sone but the root album`() {
+ 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)
+ 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))
+ }
+
}