X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fcore%2FSoneParserTest.kt;h=38419a2fbc891405f727511b2abb1e878d20903e;hp=3a7010a19e0abc7b110213c538a8f4a9a423dfe6;hb=HEAD;hpb=388401b79c3eacd94eb070b5c8ebab6055740246

diff --git a/src/test/kotlin/net/pterodactylus/sone/core/SoneParserTest.kt b/src/test/kotlin/net/pterodactylus/sone/core/SoneParserTest.kt
index 3a7010a..38419a2 100644
--- a/src/test/kotlin/net/pterodactylus/sone/core/SoneParserTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/core/SoneParserTest.kt
@@ -1,9 +1,11 @@
 package net.pterodactylus.sone.core
 
+import com.codahale.metrics.*
 import com.google.common.base.Optional.*
 import freenet.crypt.*
 import freenet.keys.InsertableClientSSK.*
 import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.data.impl.AlbumImpl
 import net.pterodactylus.sone.database.memory.*
 import net.pterodactylus.sone.freenet.wot.*
 import net.pterodactylus.sone.test.*
@@ -21,7 +23,8 @@ import kotlin.test.*
 class SoneParserTest {
 
 	private val database = MemoryDatabase(Configuration(MapConfigurationBackend()))
-	private val soneParser = SoneParser(database)
+	private val metricRegistry = MetricRegistry()
+	private val soneParser = SoneParser(database, metricRegistry)
 	private val sone = mock<Sone>()
 
 	@BeforeTest
@@ -39,6 +42,7 @@ class SoneParserTest {
 		whenever(sone.identity).thenReturn(identity)
 		whenever(sone.requestUri).thenAnswer { clientSSK.uri.setKeyType("USK").setDocName("Sone") }
 		whenever(sone.time).thenReturn(currentTimeMillis() - DAYS.toMillis(1))
+		whenever(sone.rootAlbum).thenReturn(AlbumImpl(sone))
 	}
 
 	@Test
@@ -395,4 +399,20 @@ class SoneParserTest {
 		assertThat(sone.profile.avatar, equalTo("image-id"))
 	}
 
+	@Test
+	fun `unsuccessful parsing does not add a histogram entry`() {
+		val inputStream = javaClass.getResourceAsStream("sone-parser-with-invalid-image-height.xml")
+		assertThat(soneParser.parseSone(sone, inputStream), nullValue())
+		val histogram = metricRegistry.histogram("sone.parse.duration")
+		assertThat(histogram.count, equalTo(0L))
+	}
+
+	@Test
+	fun `successful parsing adds histogram entry`() {
+		val inputStream = javaClass.getResourceAsStream("sone-parser-without-images.xml")
+		assertThat(soneParser.parseSone(sone, inputStream), notNullValue())
+		val histogram = metricRegistry.histogram("sone.parse.duration")
+		assertThat(histogram.count, equalTo(1L))
+	}
+
 }