Add auto-closeable bucket wrapper and use try-with-resources

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / utils / AutoCloseableBucketTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/utils/AutoCloseableBucketTest.kt b/src/test/kotlin/net/pterodactylus/sone/utils/AutoCloseableBucketTest.kt
new file mode 100644 (file)
index 0000000..a705844
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/utils/AutoCloseableBucketTest.kt
@@ -0,0 +1,26 @@
+package net.pterodactylus.sone.utils
+
+import freenet.support.api.Bucket
+import net.pterodactylus.sone.test.mock
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.equalTo
+import org.junit.Test
+import org.mockito.Mockito.verify
+
+class AutoCloseableBucketTest {
+
+       private val bucket = mock<Bucket>()
+       private val autoCloseableBucket = AutoCloseableBucket(bucket)
+
+       @Test
+       fun `bucket can be retrieved`() {
+               assertThat(autoCloseableBucket.bucket, equalTo(bucket))
+       }
+
+       @Test
+       fun `bucket will be free’d when close is called`() {
+               autoCloseableBucket.close()
+               verify(bucket).free()
+       }
+
+}