Server: Implement temporary links

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

@@ -0,0 +1,81 @@
+/*
+ * 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 kotlinx.coroutines.*
+import org.isoron.uhabits.sync.*
+import org.isoron.uhabits.sync.links.*
+import org.junit.Test
+import org.mockito.Mockito.*
+import kotlin.test.*
+
+class LinksModuleTest : BaseApplicationTest() {
+    private val link = Link(
+        id = "ABC123",
+        syncKey = "SECRET",
+        createdAt = System.currentTimeMillis(),
+    )
+
+    @Test
+    fun `when POST is successful should return link`(): Unit = runBlocking {
+        `when`(server.registerLink("SECRET")).thenReturn(link)
+        withTestApplication(app()) {
+            handlePost("/links", LinkRegisterRequestData(syncKey = "SECRET")).apply {
+                assertEquals(HttpStatusCode.OK, response.status())
+                assertEquals(link.toJson(), response.content)
+            }
+        }
+    }
+
+    @Test
+    fun `when GET is successful should return link`(): Unit = runBlocking {
+        `when`(server.getLink("ABC123")).thenReturn(link)
+        withTestApplication(app()) {
+            handleGet("/links/ABC123").apply {
+                assertEquals(HttpStatusCode.OK, response.status())
+                assertEquals(link.toJson(), response.content)
+            }
+        }
+    }
+
+    @Test
+    fun `GET with invalid link id should return 404`(): Unit = runBlocking {
+        `when`(server.getLink("ABC123")).thenThrow(KeyNotFoundException())
+        withTestApplication(app()) {
+            handleGet("/links/ABC123").apply {
+                assertEquals(HttpStatusCode.NotFound, response.status())
+            }
+        }
+    }
+
+    private fun TestApplicationEngine.handlePost(url: String, data: LinkRegisterRequestData): TestApplicationCall {
+        return handleRequest(HttpMethod.Post, url) {
+            addHeader(HttpHeaders.ContentType, ContentType.Application.Json.toString())
+            setBody(data.toJson())
+        }
+    }
+
+    private fun TestApplicationEngine.handleGet(url: String): TestApplicationCall {
+        return handleRequest(HttpMethod.Get, url)
+    }
+
+}

server/test/org/isoron/uhabits/sync/links/LinkManagerTest.kt

@@ -0,0 +1,44 @@
+/*
+ * 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.links
+
+import org.isoron.uhabits.sync.*
+import org.junit.Test
+import kotlin.test.*
+
+class LinkManagerTest {
+
+    @Test
+    fun testUsage() {
+        val manager = LinkManager(timeoutInMillis = 250)
+        val originalLink = manager.register(syncKey = "SECRET")
+        val retrievedLink = manager.get(originalLink.id)
+        assertEquals(originalLink, retrievedLink)
+
+        Thread.sleep(260) // wait until expiration
+        assertFailsWith<KeyNotFoundException> {
+            manager.get(originalLink.id)
+        }
+
+        assertFailsWith<KeyNotFoundException> {
+            manager.get("INVALID")
+        }
+    }
+}

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

@@ -17,11 +17,11 @@
  * with this program. If not, see <http://www.gnu.org/licenses/>.
  */
 
-package org.isoron.uhabits.sync
+package org.isoron.uhabits.sync.server
 
 import kotlinx.coroutines.*
+import org.isoron.uhabits.sync.*
 import org.isoron.uhabits.sync.repository.*
-import org.isoron.uhabits.sync.server.*
 import org.junit.Test
 import java.nio.file.*
 import kotlin.test.*