Add failure cache to element loader

[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
index d519e41..1f6f681 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/template/LinkedElementsFilterTest.kt
@@ -2,7 +2,6 @@ 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
@@ -26,14 +25,16 @@ class LinkedElementsFilterTest {
                                PlainTextPart("text"),
                                LinkPart("http://link", "link"),
                                FreenetLinkPart("KSK@link", "link", false),
+                               FreenetLinkPart("KSK@loading.png", "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"))
+               `when`(imageLoader.loadElement("KSK@link")).thenReturn(LinkedElement("KSK@link", failed = true))
+               `when`(imageLoader.loadElement("KSK@loading.png")).thenReturn(LinkedElement("KSK@loading.png", loading = true))
+               `when`(imageLoader.loadElement("KSK@link.png")).thenReturn(LinkedElement("KSK@link.png"))
                val loadedImages = filter.format(null, parts, null)
                assertThat(loadedImages, contains<LinkedElement>(
-                               LinkedImage("KSK@link", true),
-                               LinkedImage("KSK@link.png")
+                               LinkedElement("KSK@loading.png", failed = false, loading = true),
+                               LinkedElement("KSK@link.png", failed = false, loading = false)
                ))
        }