test(model-server): deduplicate code

modelix · Apr 3, 2024 · 1533d6e · 1533d6e
1 parent 41326f3
commit 1533d6e
Show file tree

Hide file tree

Showing 6 changed files with 56 additions and 81 deletions.
diff --git a/model-server/src/test/kotlin/org/modelix/model/server/ModelClientV2Test.kt b/model-server/src/test/kotlin/org/modelix/model/server/ModelClientV2Test.kt
@@ -15,14 +15,8 @@
 
 package org.modelix.model.server
 
-import io.ktor.serialization.kotlinx.json.json
-import io.ktor.server.application.install
-import io.ktor.server.plugins.contentnegotiation.ContentNegotiation
-import io.ktor.server.resources.Resources
-import io.ktor.server.routing.IgnoreTrailingSlash
 import io.ktor.server.testing.ApplicationTestBuilder
 import io.ktor.server.testing.testApplication
-import io.ktor.server.websocket.WebSockets
 import org.modelix.authorization.installAuthentication
 import org.modelix.model.api.IConceptReference
 import org.modelix.model.api.ITree
@@ -50,23 +44,12 @@ class ModelClientV2Test {
     private fun runTest(block: suspend ApplicationTestBuilder.() -> Unit) = testApplication {
         application {
             installAuthentication(unitTestMode = true)
-            install(ContentNegotiation) {
-                json()
-            }
-            install(WebSockets)
-            install(Resources)
-            install(IgnoreTrailingSlash)
+            installDefaultServerPlugins()
             ModelReplicationServer(InMemoryStoreClient()).init(this)
         }
         block()
     }
 
-    private suspend fun ApplicationTestBuilder.createModelClient(): ModelClientV2 {
-        val url = "http://localhost/v2"
-        val modelClient = ModelClientV2.builder().url(url).client(client).build().also { it.init() }
-        return modelClient
-    }
-
     @Test
     fun test_t1() = runTest {
         val client = createModelClient()

diff --git a/model-server/src/test/kotlin/org/modelix/model/server/ModelServerTestUtil.kt b/model-server/src/test/kotlin/org/modelix/model/server/ModelServerTestUtil.kt
@@ -0,0 +1,39 @@
+/*
+ * Copyright (c) 2024.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.modelix.model.server
+
+import io.ktor.serialization.kotlinx.json.json
+import io.ktor.server.application.Application
+import io.ktor.server.application.install
+import io.ktor.server.plugins.contentnegotiation.ContentNegotiation
+import io.ktor.server.resources.Resources
+import io.ktor.server.routing.IgnoreTrailingSlash
+import io.ktor.server.testing.ApplicationTestBuilder
+import io.ktor.server.websocket.WebSockets
+import org.modelix.model.client2.ModelClientV2
+
+suspend fun ApplicationTestBuilder.createModelClient(): ModelClientV2 {
+    val url = "http://localhost/v2"
+    return ModelClientV2.builder().url(url).client(client).build().also { it.init() }
+}
+
+fun Application.installDefaultServerPlugins() {
+    install(WebSockets)
+    install(ContentNegotiation) { json() }
+    install(Resources)
+    install(IgnoreTrailingSlash)
+}
diff --git a/model-server/src/test/kotlin/org/modelix/model/server/PullPerformanceTest.kt b/model-server/src/test/kotlin/org/modelix/model/server/PullPerformanceTest.kt
@@ -16,14 +16,8 @@
 
 package org.modelix.model.server
 
-import io.ktor.serialization.kotlinx.json.json
-import io.ktor.server.application.install
-import io.ktor.server.plugins.contentnegotiation.ContentNegotiation
-import io.ktor.server.resources.Resources
-import io.ktor.server.routing.IgnoreTrailingSlash
 import io.ktor.server.testing.ApplicationTestBuilder
 import io.ktor.server.testing.testApplication
-import io.ktor.server.websocket.WebSockets
 import kotlinx.coroutines.coroutineScope
 import org.modelix.authorization.installAuthentication
 import org.modelix.model.api.IChildLink
@@ -48,12 +42,7 @@ class PullPerformanceTest {
         val repositoriesManager = RepositoriesManager(LocalModelClient(storeClientWithStatistics))
         application {
             installAuthentication(unitTestMode = true)
-            install(ContentNegotiation) {
-                json()
-            }
-            install(WebSockets)
-            install(Resources)
-            install(IgnoreTrailingSlash)
+            installDefaultServerPlugins()
             ModelReplicationServer(repositoriesManager).init(this)
             KeyValueLikeModelServer(repositoriesManager).init(this)
         }

diff --git a/model-server/src/test/kotlin/org/modelix/model/server/ReplicatedRepositoryTest.kt b/model-server/src/test/kotlin/org/modelix/model/server/ReplicatedRepositoryTest.kt
@@ -15,14 +15,8 @@
 
 package org.modelix.model.server
 
-import io.ktor.serialization.kotlinx.json.json
-import io.ktor.server.application.install
-import io.ktor.server.plugins.contentnegotiation.ContentNegotiation
-import io.ktor.server.resources.Resources
-import io.ktor.server.routing.IgnoreTrailingSlash
 import io.ktor.server.testing.ApplicationTestBuilder
 import io.ktor.server.testing.testApplication
-import io.ktor.server.websocket.WebSockets
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.coroutineScope
 import kotlinx.coroutines.delay
@@ -73,12 +67,7 @@ class ReplicatedRepositoryTest {
     private fun runTest(block: suspend ApplicationTestBuilder.(scope: CoroutineScope) -> Unit) = testApplication {
         application {
             installAuthentication(unitTestMode = true)
-            install(ContentNegotiation) {
-                json()
-            }
-            install(WebSockets)
-            install(Resources)
-            install(IgnoreTrailingSlash)
+            installDefaultServerPlugins()
             val repositoriesManager = RepositoriesManager(LocalModelClient(InMemoryStoreClient()))
             ModelReplicationServer(repositoriesManager).init(this)
             KeyValueLikeModelServer(repositoriesManager).init(this)

diff --git a/model-server/src/test/kotlin/org/modelix/model/server/handlers/ContentExplorerTest.kt b/model-server/src/test/kotlin/org/modelix/model/server/handlers/ContentExplorerTest.kt
@@ -24,24 +24,21 @@ import io.ktor.client.statement.bodyAsText
 import io.ktor.http.ContentType
 import io.ktor.http.contentType
 import io.ktor.serialization.kotlinx.json.json
-import io.ktor.server.plugins.contentnegotiation.ContentNegotiation
-import io.ktor.server.resources.Resources
-import io.ktor.server.routing.IgnoreTrailingSlash
 import io.ktor.server.testing.ApplicationTestBuilder
 import io.ktor.server.testing.testApplication
-import io.ktor.server.websocket.WebSockets
 import org.jsoup.Jsoup
 import org.jsoup.nodes.Element
 import org.jsoup.select.Evaluator
 import org.modelix.model.api.IReferenceLink
 import org.modelix.model.api.ITree
 import org.modelix.model.api.NodeReferenceById
 import org.modelix.model.client.successful
-import org.modelix.model.client2.ModelClientV2
 import org.modelix.model.client2.runWrite
 import org.modelix.model.lazy.CLVersion
 import org.modelix.model.lazy.RepositoryId
 import org.modelix.model.server.api.v2.VersionDelta
+import org.modelix.model.server.createModelClient
+import org.modelix.model.server.installDefaultServerPlugins
 import org.modelix.model.server.store.InMemoryStoreClient
 import org.modelix.model.server.store.LocalModelClient
 import kotlin.test.Test
@@ -55,32 +52,22 @@ class ContentExplorerTest {
     private val modelClient = LocalModelClient(InMemoryStoreClient())
     private val repoManager = RepositoriesManager(modelClient)
 
-    private fun runTest(body: suspend (ApplicationTestBuilder.() -> Unit)) {
-        testApplication {
-            install(WebSockets)
-            install(ContentNegotiation) { json() }
-            install(Resources)
-            install(IgnoreTrailingSlash)
-            application {
-                ModelReplicationServer(repoManager).init(this)
-                ContentExplorer(modelClient, repoManager).init(this)
-            }
-
-            body()
+    private fun runTest(body: suspend (ApplicationTestBuilder.() -> Unit)) = testApplication {
+        application {
+            installDefaultServerPlugins()
+            ModelReplicationServer(repoManager).init(this)
+            ContentExplorer(modelClient, repoManager).init(this)
         }
+        body()
     }
 
-    private suspend fun ApplicationTestBuilder.createModelClient(): ModelClientV2 {
-        val url = "http://localhost/v2"
-        val modelClient = ModelClientV2.builder().url(url).client(client).build().also { it.init() }
-        return modelClient
+    private fun ApplicationTestBuilder.createHttpClient() = createClient {
+        install(ClientContentNegotiation) { json() }
     }
 
     @Test
     fun `node inspector finds root node`() = runTest {
-        val client = createClient {
-            install(ClientContentNegotiation) { json() }
-        }
+        val client = createHttpClient()
 
         val delta: VersionDelta = client.post("/v2/repositories/node-inspector/init").body()
 
@@ -122,9 +109,7 @@ class ContentExplorerTest {
 
     @Test
     fun `nodes can be expanded`() = runTest {
-        val client = createClient {
-            install(ClientContentNegotiation) { json() }
-        }
+        val client = createHttpClient()
 
         val delta: VersionDelta = client.post("/v2/repositories/node-expansion/init").body()
 

diff --git a/model-server/src/test/kotlin/org/modelix/model/server/handlers/ModelReplicationServerTest.kt b/model-server/src/test/kotlin/org/modelix/model/server/handlers/ModelReplicationServerTest.kt
@@ -18,14 +18,8 @@ package org.modelix.model.server.handlers
 import io.ktor.client.request.get
 import io.ktor.http.appendPathSegments
 import io.ktor.http.takeFrom
-import io.ktor.serialization.kotlinx.json.json
-import io.ktor.server.application.install
-import io.ktor.server.plugins.contentnegotiation.ContentNegotiation
-import io.ktor.server.resources.Resources
-import io.ktor.server.routing.IgnoreTrailingSlash
 import io.ktor.server.testing.ApplicationTestBuilder
 import io.ktor.server.testing.testApplication
-import io.ktor.server.websocket.WebSockets
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.coroutineScope
 import org.modelix.authorization.installAuthentication
@@ -35,6 +29,7 @@ import org.modelix.model.client2.readVersionDelta
 import org.modelix.model.client2.runWrite
 import org.modelix.model.client2.useVersionStreamFormat
 import org.modelix.model.lazy.RepositoryId
+import org.modelix.model.server.installDefaultServerPlugins
 import org.modelix.model.server.store.InMemoryStoreClient
 import org.modelix.model.server.store.LocalModelClient
 import kotlin.test.Test
@@ -45,12 +40,7 @@ class ModelReplicationServerTest {
     private fun runTest(block: suspend ApplicationTestBuilder.(scope: CoroutineScope) -> Unit) = testApplication {
         application {
             installAuthentication(unitTestMode = true)
-            install(ContentNegotiation) {
-                json()
-            }
-            install(WebSockets)
-            install(Resources)
-            install(IgnoreTrailingSlash)
+            installDefaultServerPlugins()
             val repositoriesManager = RepositoriesManager(LocalModelClient(InMemoryStoreClient()))
             ModelReplicationServer(repositoriesManager).init(this)
             KeyValueLikeModelServer(repositoriesManager).init(this)