add tests

Author: Bas Buijsen

firebase-storage/src/androidInstrumentedTest/kotlin/dev/gitlive/firebase/storage/storage.android.kt

@@ -0,0 +1,7 @@
+package dev.gitlive.firebase.storage
+
+import android.net.Uri
+
+actual fun createTestData(): Data {
+    return Data("test".toByteArray())
+}

firebase-storage/src/androidMain/kotlin/dev/gitlive/firebase/storage/storage.kt

@@ -20,7 +20,10 @@ import kotlinx.coroutines.channels.awaitClose
 import kotlinx.coroutines.channels.trySendBlocking
 import kotlinx.coroutines.flow.FlowCollector
 import kotlinx.coroutines.flow.callbackFlow
+import kotlinx.coroutines.flow.channelFlow
 import kotlinx.coroutines.flow.emitAll
+import kotlinx.coroutines.flow.first
+import kotlinx.coroutines.launch
 import kotlinx.coroutines.tasks.await
 
 actual val Firebase.storage get() =
@@ -77,6 +80,14 @@ actual class StorageReference(val android: com.google.firebase.storage.StorageRe
         }
     }
 
+    actual suspend fun putData(data: Data, metadata: FirebaseStorageMetadata?) {
+        if (metadata != null) {
+            android.putBytes(data.data, metadata.toStorageMetadata()).await().run {}
+        } else {
+            android.putBytes(data.data).await().run {}
+        }
+    }
+
     actual fun putFileResumable(file: File, metadata: FirebaseStorageMetadata?): ProgressFlow {
         val android = if (metadata != null) {
             android.putFile(file.uri, metadata.toStorageMetadata())
@@ -118,6 +129,8 @@ actual class ListResult(android: com.google.firebase.storage.ListResult) {
 
 actual class File(val uri: Uri)
 
+actual class Data(val data: ByteArray)
+
 actual typealias FirebaseStorageException = com.google.firebase.storage.StorageException
 
 fun FirebaseStorageMetadata.toStorageMetadata(): StorageMetadata {

firebase-storage/src/androidUnitTest/kotlin/dev/gitlive/firebase/storage/storage.android.kt

@@ -0,0 +1,7 @@
+package dev.gitlive.firebase.storage
+
+import android.net.Uri
+
+actual fun createTestData(): Data {
+    return Data("test".toByteArray())
+}

firebase-storage/src/commonMain/kotlin/dev/gitlive/firebase/storage/storage.kt

@@ -4,6 +4,9 @@ import dev.gitlive.firebase.Firebase
 import dev.gitlive.firebase.FirebaseApp
 import dev.gitlive.firebase.FirebaseException
 import kotlinx.coroutines.flow.Flow
+import kotlinx.coroutines.flow.channelFlow
+import kotlinx.coroutines.flow.first
+import kotlinx.coroutines.launch
 
 /** Returns the [FirebaseStorage] instance of the default [FirebaseApp]. */
 expect val Firebase.storage: FirebaseStorage
@@ -44,6 +47,8 @@ expect class StorageReference {
 
     suspend fun putFile(file: File, metadata: FirebaseStorageMetadata? = null)
 
+    suspend fun putData(data: Data, metadata: FirebaseStorageMetadata? = null)
+
     fun putFileResumable(file: File, metadata: FirebaseStorageMetadata? = null): ProgressFlow
 }
 
@@ -55,6 +60,8 @@ expect class ListResult {
 
 expect class File
 
+expect class Data
+
 sealed class Progress(val bytesTransferred: Number, val totalByteCount: Number) {
     class Running internal constructor(bytesTransferred: Number, totalByteCount: Number): Progress(bytesTransferred, totalByteCount)
     class Paused internal constructor(bytesTransferred: Number, totalByteCount: Number): Progress(bytesTransferred, totalByteCount)

firebase-storage/src/commonTest/kotlin/dev/gitlive/firebase/storage/storage.kt

@@ -8,7 +8,16 @@ import dev.gitlive.firebase.Firebase
 import dev.gitlive.firebase.FirebaseOptions
 import dev.gitlive.firebase.apps
 import dev.gitlive.firebase.initialize
+import dev.gitlive.firebase.runBlockingTest
+import kotlinx.coroutines.flow.channelFlow
+import kotlinx.coroutines.flow.first
+import kotlinx.coroutines.launch
+import kotlin.test.AfterTest
 import kotlin.test.BeforeTest
+import kotlin.test.Test
+import kotlin.test.assertContentEquals
+import kotlin.test.assertEquals
+import kotlin.test.assertNotNull
 
 expect val emulatorHost: String
 expect val context: Any
@@ -35,6 +44,61 @@ class FirebaseStorageTest {
 
         storage = Firebase.storage(app).apply {
             useEmulator(emulatorHost, 9199)
+            setMaxOperationRetryTimeMillis(10000)
+            setMaxUploadRetryTimeMillis(10000)
         }
     }
-}
+
+    @AfterTest
+    fun deinitializeFirebase() = runBlockingTest {
+        Firebase.apps(context).forEach {
+            it.delete()
+        }
+    }
+
+    @Test
+    fun testStorageNotNull() {
+        assertNotNull(storage)
+    }
+
+    @Test
+    fun testUploadShouldNotCrash() = runBlockingTest {
+        val data = createTestData()
+        val ref = storage.reference("test").child("testFile.txt")
+        ref.putData(data)
+    }
+
+    @Test
+    fun testUploadMetadata() = runBlockingTest {
+        val data = createTestData()
+        val ref = storage.reference("test").child("testFile.txt")
+        val metadata = storageMetadata {
+            contentType = "text/plain"
+        }
+        ref.putData(data, metadata)
+
+        val metadataResult = ref.getMetadata()
+
+        assertNotNull(metadataResult)
+        assertNotNull(metadataResult.contentType)
+        assertEquals(metadataResult.contentType, metadata.contentType)
+    }
+
+    @Test
+    fun testUploadCustomMetadata() = runBlockingTest {
+        val data = createTestData()
+        val ref = storage.reference("test").child("testFile.txt")
+        val metadata = storageMetadata {
+            contentType = "text/plain"
+            setCustomMetadata("key", "value")
+        }
+        ref.putData(data, metadata)
+
+        val metadataResult = ref.getMetadata()
+
+        assertNotNull(metadataResult)
+        assertEquals(metadataResult.customMetadata["key"], metadata.customMetadata["key"])
+    }
+}
+
+expect fun createTestData(): Data

firebase-storage/src/iosMain/kotlin/dev/gitlive/firebase/storage/storage.kt

@@ -23,6 +23,7 @@ import kotlinx.coroutines.channels.trySendBlocking
 import kotlinx.coroutines.flow.FlowCollector
 import kotlinx.coroutines.flow.callbackFlow
 import kotlinx.coroutines.flow.emitAll
+import platform.Foundation.NSData
 import platform.Foundation.NSError
 import platform.Foundation.NSURL
 
@@ -87,7 +88,13 @@ actual class StorageReference(val ios: FIRStorageReference) {
         }
     }
 
-    actual suspend fun putFile(file: File, metadata: FirebaseStorageMetadata?) = ios.awaitResult { putFile(file.url, metadata?.toFIRMetadata(), completion = it) }.run {}
+    actual suspend fun putFile(file: File, metadata: FirebaseStorageMetadata?) = ios.awaitResult { callback ->
+        putFile(file.url, metadata?.toFIRMetadata(), callback)
+    }.run {}
+
+    actual suspend fun putData(data: Data, metadata: FirebaseStorageMetadata?) = ios.awaitResult { callback ->
+        putData(data.data, metadata?.toFIRMetadata(), callback)
+    }.run {}
 
     actual fun putFileResumable(file: File, metadata: FirebaseStorageMetadata?): ProgressFlow {
         val ios = ios.putFile(file.url, metadata?.toFIRMetadata())
@@ -133,6 +140,8 @@ actual class ListResult(ios: FIRStorageListResult) {
 
 actual class File(val url: NSURL)
 
+actual class Data(val data: NSData)
+
 actual class FirebaseStorageException(message: String): FirebaseException(message)
 
 suspend inline fun <T> T.await(function: T.(callback: (NSError?) -> Unit) -> Unit) {

firebase-storage/src/iosTest/kotlin/dev/gitlive/firebase/storage/storage.ios.kt

@@ -0,0 +1,20 @@
+package dev.gitlive.firebase.storage
+
+import kotlinx.cinterop.BetaInteropApi
+import kotlinx.cinterop.utf8
+import platform.Foundation.NSCoder
+import platform.Foundation.NSData
+import platform.Foundation.NSSearchPathDirectory
+import platform.Foundation.NSSearchPathDomainMask
+import platform.Foundation.NSSearchPathForDirectoriesInDomains
+import platform.Foundation.NSString
+import platform.Foundation.NSURL
+import platform.Foundation.NSUTF8StringEncoding
+import platform.Foundation.create
+import platform.Foundation.dataUsingEncoding
+
+@OptIn(BetaInteropApi::class)
+actual fun createTestData(): Data {
+    val value = NSString.create(string = "test")
+    return Data(value.dataUsingEncoding(NSUTF8StringEncoding, false)!!)
+}

firebase-storage/src/jsMain/kotlin/dev/gitlive/firebase/storage/storage.kt

@@ -63,6 +63,8 @@ actual class StorageReference(val js: dev.gitlive.firebase.storage.externals.Sto
 
     actual suspend fun putFile(file: File, metadata: FirebaseStorageMetadata?): Unit = rethrow { uploadBytes(js, file, metadata?.toStorageMetadata()).await() }
 
+    actual suspend fun putData(data: Data, metadata: FirebaseStorageMetadata?): Unit = rethrow { uploadBytes(js, data.data, metadata?.toStorageMetadata()).await() }
+
     actual fun putFileResumable(file: File, metadata: FirebaseStorageMetadata?): ProgressFlow = rethrow {
         val uploadTask = uploadBytesResumable(js, file, metadata?.toStorageMetadata())
 
@@ -101,6 +103,7 @@ actual class ListResult(js: dev.gitlive.firebase.storage.externals.ListResult) {
 }
 
 actual typealias File = org.w3c.files.File
+actual class Data(val data: org.khronos.webgl.Uint8Array)
 
 actual open class FirebaseStorageException(code: String, cause: Throwable) :
     FirebaseException(code, cause)

firebase-storage/src/jsTest/kotlin/dev/gitlive/firebase/storage/storage.js.kt

@@ -0,0 +1,5 @@
+package dev.gitlive.firebase.storage
+
+actual fun createTestData(): Data {
+    TODO("Not yet implemented")
+}

firebase-storage/src/jvmMain/kotlin/dev/gitlive/firebase/storage/storage.jvm.kt

@@ -77,6 +77,8 @@ actual class StorageReference {
     actual suspend fun putFile(file: File, metadata: FirebaseStorageMetadata?) {
     }
 
+    actual suspend fun putData(data: Data, metadata: FirebaseStorageMetadata?) {
+    }
 }
 
 actual class ListResult {
@@ -89,4 +91,5 @@ actual class ListResult {
 }
 
 actual class File
-actual class FirebaseStorageException internal constructor(message: String) : FirebaseException(message)
+actual class FirebaseStorageException internal constructor(message: String) : FirebaseException(message)
+actual class Data