test(model-server): add test for node expansion

Author: mhuster23

model-server/src/test/kotlin/org/modelix/model/server/handlers/ContentExplorerTest.kt

@@ -19,25 +19,32 @@ package org.modelix.model.server.handlers
 import io.ktor.client.call.body
 import io.ktor.client.request.get
 import io.ktor.client.request.post
+import io.ktor.client.request.setBody
+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.client.successful
 import org.modelix.model.lazy.CLVersion
 import org.modelix.model.server.api.v2.VersionDelta
 import org.modelix.model.server.store.InMemoryStoreClient
 import org.modelix.model.server.store.LocalModelClient
 import kotlin.test.Test
+import kotlin.test.assertNotNull
 import kotlin.test.assertTrue
 import io.ktor.client.plugins.contentnegotiation.ContentNegotiation as ClientContentNegotiation
 
 class ContentExplorerTest {
 
-    private val repoId = "test-repo"
     private val modelClient = LocalModelClient(InMemoryStoreClient())
     private val repoManager = RepositoriesManager(modelClient)
 
@@ -62,7 +69,7 @@ class ContentExplorerTest {
             install(ClientContentNegotiation) { json() }
         }
 
-        val delta: VersionDelta = client.post("/v2/repositories/$repoId/init").body()
+        val delta: VersionDelta = client.post("/v2/repositories/node-inspector/init").body()
 
         val versionHash = delta.versionHash
         val version = CLVersion.loadFromHash(versionHash, modelClient.storeCache)
@@ -71,4 +78,33 @@ class ContentExplorerTest {
         val response = client.get("/content/$versionHash/$nodeId/")
         assertTrue(response.successful)
     }
+
+    @Test
+    fun `nodes can be expanded`() = runTest {
+        val client = createClient {
+            install(ClientContentNegotiation) { json() }
+        }
+
+        val delta: VersionDelta = client.post("/v2/repositories/node-expansion/init").body()
+
+        val versionHash = delta.versionHash
+        val version = CLVersion.loadFromHash(versionHash, modelClient.storeCache)
+        val nodeId = checkNotNull(version.getTree().root?.id)
+
+        val expandedNodes = ContentExplorerExpandedNodes(setOf(nodeId.toString()), false)
+
+        val response = client.post("/content/$versionHash/") {
+            contentType(ContentType.Application.Json)
+            setBody(expandedNodes)
+        }
+
+        val html = Jsoup.parse(response.bodyAsText())
+        val root: Element? = html.body().firstElementChild()
+
+        assertTrue { response.successful }
+        assertNotNull(root)
+        assertTrue { root.`is`(Evaluator.Tag("ul")) }
+        assertTrue { root.`is`(Evaluator.Class("treeRoot")) }
+        assertTrue { root.childrenSize() > 0 }
+    }
 }