test: AsyncAPI v3 yaml completion - variants provider test

asyncapi · Dec 11, 2023 · de0714e · de0714e
1 parent ba1075e
commit de0714e
Showing 1 changed file with 126 additions and 0 deletions.
diff --git a/src/test/kotlin/com/asyncapi/plugin/idea/psi/reference/v3/YamlFileVariantsProviderTest.kt b/src/test/kotlin/com/asyncapi/plugin/idea/psi/reference/v3/YamlFileVariantsProviderTest.kt
@@ -0,0 +1,126 @@
+package com.asyncapi.plugin.idea.psi.reference.v3
+
+import com.asyncapi.plugin.idea._core.xpath.JsonFileXPath
+import com.asyncapi.plugin.idea._core.xpath.YamlFileXPath
+import com.asyncapi.plugin.idea.extensions.psi.reference.YamlFileVariantsProvider
+import com.intellij.psi.PsiFileFactory
+import com.intellij.testFramework.fixtures.BasePlatformTestCase
+import junit.framework.TestCase
+import org.jetbrains.yaml.YAMLLanguage
+import org.jetbrains.yaml.psi.YAMLFile
+import org.jetbrains.yaml.psi.YAMLPsiElement
+
+/*
+    LightPlatformCodeInsightTestCase
+    LightCompletionTestCase
+    CodeInsightFixtureTestCase
+ */
+
+/**
+ * @author Pavel Bodiachevskii
+ */
+//class JsonSchemaReferenceCompletionTest: LightJavaCodeInsightFixtureTestCase() {
+//class JsonSchemaReferenceCompletionTest: CompletionAutoPopupTestCase() {
+//class JsonSchemaReferenceCompletionTest: BasePlatformTestCase() {
+class YamlFileVariantsProviderTest: BasePlatformTestCase() {
+
+    override fun getTestDataPath(): String {
+        return "src/test/testData"
+    }
+
+    fun `test completion - 3_0_0`() {
+        testCompletion("asyncapi-3.0.0")
+    }
+
+    fun testCompletion(asyncapi: String) {
+        val psiFileFactory = PsiFileFactory.getInstance(project)
+        val asyncAPISpecification = this.javaClass.getResource("/$asyncapi.yaml").readText()
+        val asyncAPIPSI = psiFileFactory.createFileFromText(
+                "$asyncapi.yaml",
+                YAMLLanguage.INSTANCE,
+                asyncAPISpecification
+        ) as YAMLFile
+
+        TestCase.assertEquals(listOf("#/components", "#/components/messages"), collectVariants("#/co", asyncAPIPSI))
+        TestCase.assertEquals(listOf("#/components", "#/components/messages"), collectVariants("#/comp", asyncAPIPSI))
+        TestCase.assertEquals(listOf("#/components", "#/components/messages"), collectVariants("#/components", asyncAPIPSI))
+        TestCase.assertEquals(
+            listOf(
+                "#/components/messages",
+                "#/components/messages/UserSignedUp",
+            ),
+            collectVariants("#/components/m", asyncAPIPSI)
+        )
+        TestCase.assertEquals(
+            listOf(
+                "#/components/messages/UserSignedUp",
+                "#/components/messages/UserSignedUp/payload",
+            ),
+            collectVariants("#/components/messages/u", asyncAPIPSI)
+        )
+
+        TestCase.assertEquals(
+            listOf(
+                "#/info",
+                "#/info/title",
+                "#/info/version",
+                "#/info/description",
+                "#/operations",
+                "#/operations/processUserSignups",
+                "#/asyncapi"
+            ),
+            collectVariants("#/i", asyncAPIPSI)
+        )
+        TestCase.assertEquals(
+            listOf(
+                "#/info",
+                "#/info/title",
+                "#/info/version",
+                "#/info/description",
+            ),
+            collectVariants("#/in", asyncAPIPSI)
+        )
+        TestCase.assertEquals(
+            listOf(
+                "#/info",
+                "#/info/title",
+                "#/info/version",
+                "#/info/description",
+            ),
+            collectVariants("#/inf", asyncAPIPSI)
+        )
+
+        TestCase.assertEquals(
+            listOf(
+                "#/info",
+                "#/info/title",
+                "#/info/version",
+                "#/info/description",
+                "#/channels",
+                "#/channels/userSignedup",
+                "#/operations",
+                "#/operations/processUserSignups",
+                "#/components",
+                "#/components/messages",
+                "#/asyncapi",
+            ),
+            collectVariants("#/", asyncAPIPSI)
+        )
+
+        TestCase.assertTrue("In case of wrong property name variants must be empty", collectVariants("#/qwerty", asyncAPIPSI).isEmpty())
+        TestCase.assertTrue("In case of wrong property name variants must be empty", collectVariants("#/123", asyncAPIPSI).isEmpty())
+        TestCase.assertTrue("In case of wrong property name variants must be empty", collectVariants("#/$%^^", asyncAPIPSI).isEmpty())
+    }
+
+    private fun collectVariants(xpath: String, specification: YAMLFile): List<String> {
+        val psiPath = JsonFileXPath.compileXPath(xpath)
+        val foundPsiElements = YamlFileXPath.findPsi(specification, psiPath, true)
+
+        return YamlFileVariantsProvider(
+                foundPsiElements.filterIsInstance<YAMLPsiElement>(),
+                specification.name,
+                psiPath
+        ).variants().map { it.lookupString }
+    }
+
+}