src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt

   1 package net.pterodactylus.sone.template
   2
   3 import net.pterodactylus.sone.core.ElementLoader
   4 import net.pterodactylus.sone.core.LinkedElement
   5 import net.pterodactylus.sone.core.LinkedImage
   6 import net.pterodactylus.sone.test.mock
   7 import net.pterodactylus.sone.text.FreenetLinkPart
   8 import net.pterodactylus.sone.text.LinkPart
   9 import net.pterodactylus.sone.text.PlainTextPart
  10 import org.hamcrest.MatcherAssert.assertThat
  11 import org.hamcrest.Matchers.contains
  12 import org.junit.Test
  13 import org.mockito.Mockito.`when`
  14
  15 /**
  16  * Unit test for [LinkedElementsFilter].
  17  */
  18 class LinkedElementsFilterTest {
  19
  20         private val imageLoader = mock<ElementLoader>()
  21         private val filter = LinkedElementsFilter(imageLoader)
  22
  23         @Test
  24         fun `filter finds all loaded freenet images`() {
  25                 val parts = listOf(
  26                                 PlainTextPart("text"),
  27                                 LinkPart("http://link", "link"),
  28                                 FreenetLinkPart("KSK@link", "link", false),
  29                                 FreenetLinkPart("KSK@link.png", "link", false)
  30                 )
  31                 `when`(imageLoader.loadElement("KSK@link")).thenReturn(LinkedImage("KSK@link", true))
  32                 `when`(imageLoader.loadElement("KSK@link.png")).thenReturn(LinkedImage("KSK@link.png"))
  33                 val loadedImages = filter.format(null, parts, null)
  34                 assertThat(loadedImages, contains<LinkedElement>(
  35                                 LinkedImage("KSK@link", true),
  36                                 LinkedImage("KSK@link.png")
  37                 ))
  38         }
  39
  40 }