X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FDefaultElementLoaderTest.kt;h=424aaff7b15324e619a2f16fefbcca066552304e;hb=7bb75fc81cd59d728fbc076a0d7356e43720c92a;hp=adc5bc96c28421a0acb4cd629172d21740165159;hpb=675710fc669a9f5ccfab42296a3aa0b822539e14;p=Sone.git diff --git a/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt b/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt index adc5bc9..424aaff 100644 --- a/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt +++ b/src/test/kotlin/net/pterodactylus/sone/core/DefaultElementLoaderTest.kt @@ -1,23 +1,21 @@ package net.pterodactylus.sone.core +import com.google.common.base.Ticker import com.google.common.io.ByteStreams -import com.google.common.io.Files import freenet.keys.FreenetURI import net.pterodactylus.sone.core.FreenetInterface.BackgroundFetchCallback import net.pterodactylus.sone.test.capture import net.pterodactylus.sone.test.mock -import org.hamcrest.MatcherAssert import org.hamcrest.MatcherAssert.assertThat -import org.hamcrest.Matchers import org.hamcrest.Matchers.`is` -import org.hamcrest.Matchers.instanceOf -import org.hamcrest.Matchers.nullValue import org.junit.Test import org.mockito.ArgumentMatchers.any import org.mockito.ArgumentMatchers.eq +import org.mockito.Mockito.`when` import org.mockito.Mockito.times import org.mockito.Mockito.verify import java.io.ByteArrayOutputStream +import java.util.concurrent.TimeUnit /** * Unit test for [DefaultElementLoaderTest]. @@ -26,23 +24,29 @@ class DefaultElementLoaderTest { companion object { private const val IMAGE_ID = "KSK@gpl.png" + private val freenetURI = FreenetURI(IMAGE_ID) + private const val decomposedKey = "CHK@DCiVgTWW9nnWHJc9EVwtFJ6jAfBSVyy~rgiPvhUKbS4,mNY85V0x7dYcv7SnEYo1PCC6y2wNWMDNt-y9UWQx9fI,AAMC--8/fru%CC%88hstu%CC%88ck.jpg" + private const val normalizedKey = "CHK@DCiVgTWW9nnWHJc9EVwtFJ6jAfBSVyy~rgiPvhUKbS4,mNY85V0x7dYcv7SnEYo1PCC6y2wNWMDNt-y9UWQx9fI,AAMC--8/frühstück.jpg" + private val sizeOkay = 2097152L + private val sizeNotOkay = sizeOkay + 1 } private val freenetInterface = mock() - private val elementLoader = DefaultElementLoader(freenetInterface) + private val ticker = mock() + private val elementLoader = DefaultElementLoader(freenetInterface, ticker) private val callback = capture() @Test fun `image loader starts request for link that is not known`() { elementLoader.loadElement(IMAGE_ID) - verify(freenetInterface).startFetch(eq(FreenetURI(IMAGE_ID)), any()) + verify(freenetInterface).startFetch(eq(freenetURI), any()) } @Test fun `element loader only starts request once`() { elementLoader.loadElement(IMAGE_ID) elementLoader.loadElement(IMAGE_ID) - verify(freenetInterface).startFetch(eq(FreenetURI(IMAGE_ID)), any()) + verify(freenetInterface).startFetch(eq(freenetURI), any()) } @Test @@ -51,23 +55,68 @@ class DefaultElementLoaderTest { } @Test - fun `image loader can load image`() { + fun `element loader does not cancel on image mime type with 2 mib size`() { elementLoader.loadElement(IMAGE_ID) - verify(freenetInterface).startFetch(eq(FreenetURI(IMAGE_ID)), callback.capture()) - callback.value.loaded(FreenetURI(IMAGE_ID), "image/png", read("/static/images/unknown-image-0.png")) - val linkedElement = elementLoader.loadElement(IMAGE_ID) - assertThat(linkedElement.link, `is`(IMAGE_ID)) - assertThat(linkedElement.loading, `is`(false)) - assertThat(linkedElement, instanceOf(LinkedImage::class.java)) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + assertThat(callback.value.shouldCancel(freenetURI, "image/png", sizeOkay), `is`(false)) + } + + @Test + fun `element loader does cancel on image mime type with more than 2 mib size`() { + elementLoader.loadElement(IMAGE_ID) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + assertThat(callback.value.shouldCancel(freenetURI, "image/png", sizeNotOkay), `is`(true)) + } + + @Test + fun `element loader does cancel on audio mime type`() { + elementLoader.loadElement(IMAGE_ID) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + assertThat(callback.value.shouldCancel(freenetURI, "audio/mpeg", sizeOkay), `is`(true)) + } + + @Test + fun `element loader does not cancel on video mime type`() { + elementLoader.loadElement(IMAGE_ID) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + assertThat(callback.value.shouldCancel(freenetURI, "video/mkv", sizeOkay), `is`(true)) } @Test - fun `image can be loaded again after it failed`() { + fun `element loader does not cancel on text mime type`() { elementLoader.loadElement(IMAGE_ID) - verify(freenetInterface).startFetch(eq(FreenetURI(IMAGE_ID)), callback.capture()) - callback.value.failed(FreenetURI(IMAGE_ID)) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + assertThat(callback.value.shouldCancel(freenetURI, "text/plain", sizeOkay), `is`(true)) + } + + @Test + fun `image loader can load image`() { + elementLoader.loadElement(decomposedKey) + verify(freenetInterface).startFetch(eq(FreenetURI(decomposedKey)), callback.capture()) + callback.value.loaded(FreenetURI(normalizedKey), "image/png", read("/static/images/unknown-image-0.png")) + val linkedElement = elementLoader.loadElement(decomposedKey) + assertThat(linkedElement, `is`(LinkedElement(normalizedKey))) + } + + @Test + fun `image is not loaded again after it failed`() { + elementLoader.loadElement(IMAGE_ID) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + callback.value.failed(freenetURI) + assertThat(elementLoader.loadElement(IMAGE_ID).failed, `is`(true)) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + } + + @Test + fun `image is loaded again after failure cache is expired`() { elementLoader.loadElement(IMAGE_ID) - verify(freenetInterface, times(2)).startFetch(eq(FreenetURI(IMAGE_ID)), callback.capture()) + verify(freenetInterface).startFetch(eq(freenetURI), callback.capture()) + callback.value.failed(freenetURI) + `when`(ticker.read()).thenReturn(TimeUnit.MINUTES.toNanos(31)) + val linkedElement = elementLoader.loadElement(IMAGE_ID) + assertThat(linkedElement.failed, `is`(false)) + assertThat(linkedElement.loading, `is`(true)) + verify(freenetInterface, times(2)).startFetch(eq(freenetURI), callback.capture()) } private fun read(resource: String): ByteArray =