Add database tests

Author: michaelprichardson

firebase-database/src/androidAndroidTest/kotlin/dev/gitlive/firebase/database/database.kt

@@ -0,0 +1,15 @@
+/*
+ * Copyright (c) 2020 GitLive Ltd.  Use of this source code is governed by the Apache 2.0 license.
+ */
+
+@file:JvmName("tests")
+package dev.gitlive.firebase.database
+
+import androidx.test.platform.app.InstrumentationRegistry
+import kotlinx.coroutines.runBlocking
+
+actual val emulatorHost: String = "10.0.2.2"
+
+actual val context: Any = InstrumentationRegistry.getInstrumentation().targetContext
+
+actual fun runTest(test: suspend () -> Unit) = runBlocking { test() }

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

@@ -13,14 +13,12 @@ import dev.gitlive.firebase.database.ChildEvent.Type
 import dev.gitlive.firebase.decode
 import dev.gitlive.firebase.safeOffer
 import kotlinx.coroutines.CompletableDeferred
-import kotlinx.coroutines.FlowPreview
 import kotlinx.coroutines.channels.awaitClose
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.flow.callbackFlow
 import kotlinx.coroutines.flow.filter
 import kotlinx.coroutines.flow.produceIn
-import kotlinx.coroutines.runBlocking
 import kotlinx.coroutines.selects.select
 import kotlinx.coroutines.tasks.asDeferred
 import kotlinx.coroutines.tasks.await

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

@@ -0,0 +1,92 @@
+package dev.gitlive.firebase.database
+
+import dev.gitlive.firebase.*
+import kotlinx.coroutines.flow.first
+import kotlinx.serialization.*
+import kotlin.test.*
+
+expect val emulatorHost: String
+expect val context: Any
+expect fun runTest(test: suspend () -> Unit)
+
+class FirebaseDatabaseTest {
+
+    @Serializable
+    data class DatabaseTest(val prop: String, val likes: Int = 0)
+
+    @BeforeTest
+    fun initializeFirebase() {
+        Firebase
+            .takeIf { Firebase.apps(context).isEmpty() }
+            ?.apply {
+                initialize(
+                    context,
+                    FirebaseOptions(
+                        applicationId = "1:846484016111:ios:dd1f6688bad7af768c841a",
+                        apiKey = "AIzaSyCK87dcMFhzCz_kJVs2cT2AVlqOTLuyWV0",
+                        databaseUrl = "http://fir-kotlin-sdk.firebaseio.com",
+                        storageBucket = "fir-kotlin-sdk.appspot.com",
+                        projectId = "fir-kotlin-sdk",
+                        gcmSenderId = "846484016111"
+                    )
+                )
+                Firebase.database.useEmulator(emulatorHost, 9000)
+            }
+    }
+
+    @AfterTest
+    fun tearDown() {
+        Firebase
+            .takeIf { Firebase.apps(context).isNotEmpty() }
+            ?.apply { app.delete() }
+    }
+
+    @Test
+    fun testBasicIncrementTransaction() = runTest {
+        val data = DatabaseTest("post1", 2)
+        val userRef = Firebase.database.reference("users/user_1/post_id_1")
+        setupDatabase(userRef, data, DatabaseTest.serializer())
+
+        // Check database before transaction
+        val userDocBefore = userRef.valueEvents.first().value(DatabaseTest.serializer())
+        assertEquals(data.prop, userDocBefore.prop)
+        assertEquals(data.likes, userDocBefore.likes)
+
+        // Run transaction
+        userRef.runTransaction(DatabaseTest.serializer()) { DatabaseTest(data.prop, it.likes + 1) }
+
+        // Check the database after transaction
+        val userDocAfter = userRef.valueEvents.first().value(DatabaseTest.serializer())
+        assertEquals(data.prop, userDocAfter.prop)
+        assertEquals(data.likes, userDocAfter.likes + 1)
+    }
+
+    @Test
+    fun testBasicDecrementTransaction() = runTest {
+        val data = DatabaseTest("post2", 2)
+        val userRef = Firebase.database.reference("users/user_1/post_id_2")
+        setupDatabase(userRef, data, DatabaseTest.serializer())
+
+        // Check database before transaction
+        val userDocBefore = userRef.valueEvents.first().value(DatabaseTest.serializer())
+        assertEquals(data.prop, userDocBefore.prop)
+        assertEquals(data.likes, userDocBefore.likes)
+
+        // Run transaction
+        userRef.runTransaction(DatabaseTest.serializer()) { DatabaseTest(data.prop, it.likes - 1) }
+
+        // Check the database after transaction
+        val userDocAfter = userRef.valueEvents.first().value(DatabaseTest.serializer())
+        assertEquals(data.prop, userDocAfter.prop)
+        assertEquals(data.likes, userDocAfter.likes - 1)
+    }
+
+    private suspend fun <T> setupDatabase(ref: DatabaseReference, data: T, strategy: SerializationStrategy<T>) {
+        try {
+            ref.setValue(strategy, data)
+        } catch (err: DatabaseException) {
+            println(err)
+        }
+    }
+
+}

firebase-database/src/iosTest/kotlin/dev/gitlive/firebase/database/database.kt

@@ -0,0 +1,20 @@
+package dev.gitlive.firebase.database
+
+import kotlinx.coroutines.*
+import platform.Foundation.*
+
+actual val emulatorHost: String = "localhost"
+
+actual val context: Any = Unit
+
+actual fun runTest(test: suspend () -> Unit) = runBlocking {
+    val testRun = MainScope().async { test() }
+    while (testRun.isActive) {
+        NSRunLoop.mainRunLoop.runMode(
+            NSDefaultRunLoopMode,
+            beforeDate = NSDate.create(timeInterval = 1.0, sinceDate = NSDate())
+        )
+        yield()
+    }
+    testRun.await()
+}

firebase-database/src/jsTest/kotlin/dev/gitlive/firebase/database/database.kt

@@ -0,0 +1,26 @@
+package dev.gitlive.firebase.database
+
+import kotlinx.coroutines.GlobalScope
+import kotlinx.coroutines.promise
+
+actual val emulatorHost: String = "localhost"
+
+actual val context: Any = Unit
+
+actual fun runTest(test: suspend () -> Unit) = GlobalScope
+    .promise {
+        try {
+            test()
+        } catch (e: Throwable) {
+            e.log()
+            throw e
+        }
+    }.asDynamic()
+
+internal fun Throwable.log() {
+    console.error(this)
+    cause?.let {
+        console.error("Caused by:")
+        it.log()
+    }
+}