X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneRescuerTest.kt;h=1fd2553314cd1baebc4025899ebf15f99f4f1138;hb=b7f0567e2a3e88f2327dfdb045a1b0e4bca9dcac;hp=af8a40290f77af77ea0d68817115d75f2f14e422;hpb=22458ba912ec789646f387a7b25d12b5f45c907d;p=Sone.git

diff --git a/src/test/kotlin/net/pterodactylus/sone/core/SoneRescuerTest.kt b/src/test/kotlin/net/pterodactylus/sone/core/SoneRescuerTest.kt
index af8a402..1fd2553 100644
--- a/src/test/kotlin/net/pterodactylus/sone/core/SoneRescuerTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/core/SoneRescuerTest.kt
@@ -5,12 +5,12 @@ import net.pterodactylus.sone.data.*
 import net.pterodactylus.sone.test.*
 import org.hamcrest.MatcherAssert.*
 import org.hamcrest.Matchers.equalTo
-import org.junit.*
 import org.mockito.ArgumentMatchers.any
 import org.mockito.ArgumentMatchers.eq
 import org.mockito.Mockito.doAnswer
 import org.mockito.Mockito.never
 import org.mockito.Mockito.verify
+import kotlin.test.Test
 
 /**
  * Unit test for [SoneRescuer].
@@ -67,7 +67,7 @@ class SoneRescuerTest {
 	@Test
 	fun successfulInsert() {
 		val fetchedSone = mock<Sone>()
-		returnUriOnInsert(fetchedSone)
+		returnUriOnInsert(fetchedSone, currentEdition - 1)
 		soneRescuer.startNextFetch()
 		soneRescuer.serviceRun()
 		verify(core).lockSone(eq(sone))
@@ -78,8 +78,21 @@ class SoneRescuerTest {
 	}
 
 	@Test
+	fun `starting fetch with skipping one edition skips one edition`() {
+		val fetchedSone = mock<Sone>()
+		returnUriOnInsert(fetchedSone, currentEdition - 2)
+		soneRescuer.startNextFetchWithSkip()
+		soneRescuer.serviceRun()
+		verify(core).lockSone(eq(sone))
+		verify(core).updateSone(eq(fetchedSone), eq(true))
+		assertThat(soneRescuer.isLastFetchSuccessful, equalTo(true))
+		assertThat(soneRescuer.isFetching, equalTo(false))
+		assertThat(soneRescuer.currentEdition, equalTo(currentEdition - 2))
+	}
+
+	@Test
 	fun nonSuccessfulInsertIsRecognized() {
-		returnUriOnInsert(null)
+		returnUriOnInsert(null, (currentEdition - 1))
 		soneRescuer.startNextFetch()
 		soneRescuer.serviceRun()
 		verify(core).lockSone(eq(sone))
@@ -89,20 +102,20 @@ class SoneRescuerTest {
 		assertThat(soneRescuer.currentEdition, equalTo(currentEdition))
 	}
 
-	private fun returnUriOnInsert(fetchedSone: Sone?) {
-		val keyWithMetaStrings = setupFreenetUri()
+	private fun returnUriOnInsert(fetchedSone: Sone?, edition: Long) {
+		val keyWithMetaStrings = setupFreenetUri(edition)
 		doAnswer {
 			soneRescuer.stop()
 			fetchedSone
 		}.whenever(soneDownloader).fetchSone(eq(sone), eq(keyWithMetaStrings), eq(true))
 	}
 
-	private fun setupFreenetUri(): FreenetURI {
+	private fun setupFreenetUri(edition: Long): FreenetURI {
 		val sskKey = mock<FreenetURI>()
 		val keyWithDocName = mock<FreenetURI>()
 		val keyWithMetaStrings = mock<FreenetURI>()
 		whenever(keyWithDocName.setMetaString(eq(arrayOf("sone.xml")))).thenReturn(keyWithMetaStrings)
-		whenever(sskKey.setDocName(eq("Sone-" + (currentEdition - 1)))).thenReturn(keyWithDocName)
+		whenever(sskKey.setDocName(eq("Sone-" + edition))).thenReturn(keyWithDocName)
 		whenever(sone.requestUri.setKeyType(eq("SSK"))).thenReturn(sskKey)
 		return keyWithMetaStrings
 	}