Initial version of Ktor sync server

server/test/org/isoron/uhabits/sync/MemorySyncServerTest.kt

@@ -0,0 +1,55 @@
+/*
+ * Copyright (C) 2016-2020 Alinson Santos Xavier <git@axavier.org>
+ *
+ * This file is part of Loop Habit Tracker.
+ *
+ * Loop Habit Tracker is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by the
+ * Free Software Foundation, either version 3 of the License, or (at your
+ * option) any later version.
+ *
+ * Loop Habit Tracker is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package org.isoron.uhabits.sync
+
+import org.junit.Test
+import kotlin.test.*
+
+class MemorySyncServerTest {
+
+    private val server = MemorySyncServer()
+    private val key = server.register()
+
+    @Test
+    fun testUsage() {
+        val data0 = SyncData(0, "")
+        assertEquals(server.get(key), data0)
+
+        val data1 = SyncData(1, "Hello world")
+        server.put(key, data1)
+        assertEquals(server.get(key), data1)
+
+        val data2 = SyncData(2, "Hello new world")
+        server.put(key, data2)
+        assertEquals(server.get(key), data2)
+
+        assertFailsWith<EditConflictException> {
+            server.put(key, data2)
+        }
+
+        assertFailsWith<KeyNotFoundException> {
+            server.get("INVALID")
+        }
+
+        assertFailsWith<KeyNotFoundException> {
+            server.put("INVALID", data0)
+        }
+    }
+}

server/test/org/isoron/uhabits/sync/app/BaseApplicationTest.kt

@@ -0,0 +1,35 @@
+/*
+ * Copyright (C) 2016-2020 Alinson Santos Xavier <git@axavier.org>
+ *
+ * This file is part of Loop Habit Tracker.
+ *
+ * Loop Habit Tracker is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by the
+ * Free Software Foundation, either version 3 of the License, or (at your
+ * option) any later version.
+ *
+ * Loop Habit Tracker is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package org.isoron.uhabits.sync.app
+
+import io.ktor.application.*
+import org.isoron.uhabits.sync.*
+import org.mockito.Mockito.*
+
+open class BaseApplicationTest {
+
+    protected val server: AbstractSyncServer = mock(AbstractSyncServer::class.java)
+
+    protected fun app(): Application.() -> Unit = {
+        SyncApplication(server).apply {
+            main()
+        }
+    }
+}

server/test/org/isoron/uhabits/sync/app/RegistrationModuleTest.kt

@@ -0,0 +1,49 @@
+/*
+ * Copyright (C) 2016-2020 Alinson Santos Xavier <git@axavier.org>
+ *
+ * This file is part of Loop Habit Tracker.
+ *
+ * Loop Habit Tracker is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by the
+ * Free Software Foundation, either version 3 of the License, or (at your
+ * option) any later version.
+ *
+ * Loop Habit Tracker is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package org.isoron.uhabits.sync.app
+
+import io.ktor.http.*
+import io.ktor.server.testing.*
+import org.isoron.uhabits.sync.*
+import org.junit.Test
+import org.mockito.*
+import org.mockito.Mockito.*
+import kotlin.test.*
+
+class RegistrationModuleTest : BaseApplicationTest() {
+    @Test
+    fun `when register succeeds should return generated key`() {
+        `when`(server.register()).thenReturn("ABCDEF")
+        withTestApplication(app()) {
+            val call = handleRequest(HttpMethod.Post, "/register")
+            assertEquals(HttpStatusCode.OK, call.response.status())
+            assertEquals("{\"key\":\"ABCDEF\"}", call.response.content)
+        }
+    }
+
+    @Test
+    fun `when registration is unavailable should return 503`() {
+        `when`(server.register()).thenThrow(RegistrationUnavailableException())
+        withTestApplication(app()) {
+            val call = handleRequest(HttpMethod.Post, "/register")
+            assertEquals(HttpStatusCode.ServiceUnavailable, call.response.status())
+        }
+    }
+}

server/test/org/isoron/uhabits/sync/app/StorageModuleTest.kt

@@ -0,0 +1,97 @@
+/*
+ * Copyright (C) 2016-2020 Alinson Santos Xavier <git@axavier.org>
+ *
+ * This file is part of Loop Habit Tracker.
+ *
+ * Loop Habit Tracker is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by the
+ * Free Software Foundation, either version 3 of the License, or (at your
+ * option) any later version.
+ *
+ * Loop Habit Tracker is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
+ * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package org.isoron.uhabits.sync.app
+
+import io.ktor.http.*
+import io.ktor.server.testing.*
+import org.isoron.uhabits.sync.*
+import org.junit.Test
+import org.mockito.Mockito.*
+import kotlin.test.*
+
+class StorageModuleTest : BaseApplicationTest() {
+    private val data1 = SyncData(1, "Hello world")
+    private val data2 = SyncData(2, "Hello new world")
+
+    @Test
+    fun `when get succeeds should return data`() {
+        `when`(server.get("k1")).thenReturn(data1)
+        withTestApplication(app()) {
+            handleGet("/db/k1").apply {
+                assertEquals(HttpStatusCode.OK, response.status())
+                assertEquals(data1.toJson(), response.content)
+            }
+        }
+    }
+
+    @Test
+    fun `when get with invalid key should return 404`() {
+        `when`(server.get("k1")).thenThrow(KeyNotFoundException())
+        withTestApplication(app()) {
+            handleGet("/db/k1").apply {
+                assertEquals(HttpStatusCode.NotFound, response.status())
+            }
+        }
+    }
+
+
+    @Test
+    fun `when put succeeds should return OK`() {
+        withTestApplication(app()) {
+            handlePut("/db/k1", data1).apply {
+                assertEquals(HttpStatusCode.OK, response.status())
+                verify(server).put("k1", data1)
+            }
+        }
+    }
+
+    @Test
+    fun `when put with invalid key should return 404`() {
+        `when`(server.put("k1", data1)).thenThrow(KeyNotFoundException())
+        withTestApplication(app()) {
+            handlePut("/db/k1", data1).apply {
+                assertEquals(HttpStatusCode.NotFound, response.status())
+            }
+        }
+    }
+
+    @Test
+    fun `when put with invalid version should return 409 and current data`() {
+        `when`(server.put("k1", data1)).thenThrow(EditConflictException())
+        `when`(server.get("k1")).thenReturn(data2)
+        withTestApplication(app()) {
+            handlePut("/db/k1", data1).apply {
+                assertEquals(HttpStatusCode.Conflict, response.status())
+                assertEquals(data2.toJson(), response.content)
+            }
+        }
+    }
+
+    private fun TestApplicationEngine.handlePut(url: String, data: SyncData): TestApplicationCall {
+        return handleRequest(HttpMethod.Put, url) {
+            addHeader(HttpHeaders.ContentType, ContentType.Application.Json.toString())
+            setBody(data.toJson())
+        }
+    }
+
+    private fun TestApplicationEngine.handleGet(url: String): TestApplicationCall {
+        return handleRequest(HttpMethod.Get, url)
+    }
+}