test(model-api-gen-gradle): added test case for generated enums

Author: mhuster23

model-api-gen-gradle-test/build.gradle.kts

@@ -39,6 +39,16 @@ val modelixCoreVersion: String = projectDir.resolve("../version.txt").readText()
 dependencies {
     mps("com.jetbrains:mps:2021.1.4")
     implementation("org.modelix:model-api-gen-runtime:$modelixCoreVersion")
+    testImplementation(kotlin("test"))
+    testImplementation("org.modelix:model-api:$modelixCoreVersion")
+    testImplementation("org.modelix:model-client:$modelixCoreVersion")
+    testImplementation(files("$buildDir/metamodel/kotlin_gen") {
+        builtBy("generateMetaModelSources")
+    })
+}
+
+tasks.test {
+    useJUnitPlatform()
 }
 
 val resolveMps by tasks.registering(Sync::class) {
@@ -58,7 +68,7 @@ metamodel {
     kotlinDir = kotlinGenDir
     kotlinProject = project
     includeNamespace("jetbrains")
-    //exportModules("jetbrains.mps.baseLanguage")
+    exportModules("jetbrains.mps.baseLanguage")
 
     names {
         languageClass.prefix = "L_"

model-api-gen-gradle-test/src/test/kotlin/GeneratedApiTest.kt

@@ -0,0 +1,46 @@
+import jetbrains.mps.lang.core.L_jetbrains_mps_lang_core
+import jetbrains.mps.lang.editor.*
+import org.modelix.metamodel.TypedLanguagesRegistry
+import org.modelix.metamodel.typed
+import org.modelix.metamodel.untyped
+import org.modelix.model.ModelFacade
+import org.modelix.model.api.INode
+import org.modelix.model.api.getRootNode
+import org.modelix.model.data.ModelData
+import java.io.File
+import kotlin.test.Test
+import kotlin.test.assertContains
+import kotlin.test.assertEquals
+
+class GeneratedApiTest {
+
+    @Test
+    fun `can handle enums`() {
+        val branch = ModelFacade.toLocalBranch(ModelFacade.newLocalTree())
+        TypedLanguagesRegistry.register(L_jetbrains_mps_lang_editor)
+        TypedLanguagesRegistry.register(L_jetbrains_mps_lang_core)
+        val data = ModelData.fromJson(File("build/metamodel/exported-modules/jetbrains.mps.baseLanguage.blTypes.json").readText())
+        branch.runWrite {
+            data.load(branch)
+            val node = findNodeWithStyleAttribute(branch.getRootNode())!!.typed(C_FontStyleStyleClassItem.getInstanceInterface())
+            assertContains(_FontStyle_Enum.values(), node.style)
+            val enumValue = _FontStyle_Enum.BOLD_ITALIC
+            node.style = enumValue
+            assertEquals(enumValue, node.untyped().typed(C_FontStyleStyleClassItem.getInstanceInterface()).style)
+        }
+    }
+
+    private fun findNodeWithStyleAttribute(node: INode) : INode? {
+        var found = node.allChildren.find { it.getPropertyRoles().contains("style") }
+        if (found != null) return found
+
+        for (child in node.allChildren) {
+            found = findNodeWithStyleAttribute(child)
+            if (found != null) {
+                return found
+            }
+        }
+        return null
+    }
+
+}