Show loading animation while loading elements

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / template / LinkedElementsFilterTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt b/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt
new file mode 100644 (file)
index 0000000..d519e41
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt
@@ -0,0 +1,40 @@
+package net.pterodactylus.sone.template
+
+import net.pterodactylus.sone.core.ElementLoader
+import net.pterodactylus.sone.core.LinkedElement
+import net.pterodactylus.sone.core.LinkedImage
+import net.pterodactylus.sone.test.mock
+import net.pterodactylus.sone.text.FreenetLinkPart
+import net.pterodactylus.sone.text.LinkPart
+import net.pterodactylus.sone.text.PlainTextPart
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.contains
+import org.junit.Test
+import org.mockito.Mockito.`when`
+
+/**
+ * Unit test for [LinkedElementsFilter].
+ */
+class LinkedElementsFilterTest {
+
+       private val imageLoader = mock<ElementLoader>()
+       private val filter = LinkedElementsFilter(imageLoader)
+
+       @Test
+       fun `filter finds all loaded freenet images`() {
+               val parts = listOf(
+                               PlainTextPart("text"),
+                               LinkPart("http://link", "link"),
+                               FreenetLinkPart("KSK@link", "link", false),
+                               FreenetLinkPart("KSK@link.png", "link", false)
+               )
+               `when`(imageLoader.loadElement("KSK@link")).thenReturn(LinkedImage("KSK@link", true))
+               `when`(imageLoader.loadElement("KSK@link.png")).thenReturn(LinkedImage("KSK@link.png"))
+               val loadedImages = filter.format(null, parts, null)
+               assertThat(loadedImages, contains<LinkedElement>(
+                               LinkedImage("KSK@link", true),
+                               LinkedImage("KSK@link.png")
+               ))
+       }
+
+}