src/test/kotlin/net/pterodactylus/sone/core/DefaultImageLoaderTest.kt

   1 package net.pterodactylus.sone.core
   2
   3 import com.google.common.io.ByteStreams
   4 import com.google.common.io.Files
   5 import freenet.keys.FreenetURI
   6 import net.pterodactylus.sone.core.FreenetInterface.BackgroundFetchCallback
   7 import net.pterodactylus.sone.test.capture
   8 import net.pterodactylus.sone.test.mock
   9 import org.hamcrest.MatcherAssert
  10 import org.hamcrest.MatcherAssert.assertThat
  11 import org.hamcrest.Matchers
  12 import org.hamcrest.Matchers.`is`
  13 import org.hamcrest.Matchers.nullValue
  14 import org.junit.Test
  15 import org.mockito.ArgumentMatchers.any
  16 import org.mockito.ArgumentMatchers.eq
  17 import org.mockito.Mockito.verify
  18 import java.io.ByteArrayOutputStream
  19
  20 /**
  21  * Unit test for [DefaultImageLoaderTest].
  22  */
  23 class DefaultImageLoaderTest {
  24
  25         companion object {
  26                 private const val IMAGE_ID = "KSK@gpl.png"
  27         }
  28
  29         private val freenetInterface = mock<FreenetInterface>()
  30         private val imageLoader = DefaultImageLoader(freenetInterface)
  31         private val callback = capture<BackgroundFetchCallback>()
  32
  33         @Test
  34         fun `image loader starts request for link that is not known`() {
  35                 assertThat(imageLoader.toLoadedImage(IMAGE_ID), nullValue())
  36                 verify(freenetInterface).startFetch(eq(FreenetURI(IMAGE_ID)), any<BackgroundFetchCallback>())
  37         }
  38
  39         @Test
  40         fun `image loader can load image`() {
  41                 assertThat(imageLoader.toLoadedImage(IMAGE_ID), nullValue())
  42                 verify(freenetInterface).startFetch(eq(FreenetURI(IMAGE_ID)), callback.capture())
  43             callback.value.loaded(FreenetURI(IMAGE_ID), "image/png", read("/static/images/unknown-image-0.png"))
  44                 val loadedImage = imageLoader.toLoadedImage(IMAGE_ID)!!
  45                 assertThat(loadedImage.link, `is`(IMAGE_ID))
  46                 assertThat(loadedImage.mimeType, `is`("image/png"))
  47                 assertThat(loadedImage.width, `is`(200))
  48                 assertThat(loadedImage.height, `is`(150))
  49         }
  50
  51         private fun read(resource: String): ByteArray =
  52                         javaClass.getResourceAsStream(resource)?.use { input ->
  53                                 ByteArrayOutputStream().use {
  54                                         ByteStreams.copy(input, it)
  55                                         it
  56                                 }.toByteArray()
  57                         } ?: ByteArray(0)
  58
  59 }