test module

Author: tangzx

src/test/kotlin/com/tang/intellij/test/FileTree.kt

@@ -132,6 +132,8 @@ class TestProject(
     val root: VirtualFile,
     val filesWithCaret: List<String>
 ) {
+    val fileWithCaret: String get() = filesWithCaret.singleOrNull()!!
+
     fun doFindElementInFile(path: String): PsiElement {
         val vFile = root.findFileByRelativePath(path)
             ?: error("No `$path` file in test project")
@@ -180,5 +182,5 @@ private fun findElementInFile(file: PsiFile, marker: String): PsiElement {
         error { "No element found, offset = $elementOffset" }
 }
 
-fun replaceCaretMarker(text: String): String = text.replace("/*caret*/", "<caret>")
-fun hasCaretMarker(text: String): Boolean = text.contains("/*caret*/")
+fun replaceCaretMarker(text: String): String = text.replace("--[[caret]]", "<caret>")
+fun hasCaretMarker(text: String): Boolean = text.contains("--[[caret]]")

src/test/kotlin/com/tang/intellij/test/LuaTestBase.kt

@@ -46,4 +46,10 @@ abstract class LuaTestBase : LightPlatformCodeInsightFixtureTestCase() {
 
     protected fun FileTree.create(): TestProject =
             create(myFixture.project, myFixture.findFileInTempDir(""))
+
+    protected fun FileTree.createAndOpenFileWithCaretMarker(): TestProject {
+        val testProject = create()
+        myFixture.configureFromTempProjectFile(testProject.fileWithCaret)
+        return testProject
+    }
 }

src/test/kotlin/com/tang/intellij/test/completion/TestModule.kt

@@ -0,0 +1,64 @@
+/*
+ * Copyright (c) 2017. tangzx([email protected])
+ *
+ * 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 com.tang.intellij.test.completion
+
+import com.intellij.openapi.fileEditor.FileDocumentManager
+import com.tang.intellij.test.LuaTestBase
+import com.tang.intellij.test.fileTreeFromText
+
+class TestModule : LuaTestBase() {
+
+    fun `test module type`() {
+        fileTreeFromText("""
+             --- moduleA.lua
+             ---@module TypeA
+             module("TypeA")
+
+             name = "a"
+
+             --- B.lua
+             local a ---@type TypeA
+             a.--[[caret]]
+        """).createAndOpenFileWithCaretMarker()
+
+        FileDocumentManager.getInstance().saveAllDocuments()
+        myFixture.completeBasic()
+        val elementStrings = myFixture.lookupElementStrings
+        assertTrue(elementStrings!!.contains("name"))
+    }
+
+
+    fun `test module members visibility`() {
+        fileTreeFromText("""
+             --- moduleA.lua
+             ---@module TypeA
+             module("TypeA")
+
+             myFieldName = "a"
+
+             function myFunction() end
+
+             --- B.lua
+             --[[caret]]
+        """).createAndOpenFileWithCaretMarker()
+
+        FileDocumentManager.getInstance().saveAllDocuments()
+        myFixture.completeBasic()
+        val elementStrings = myFixture.lookupElementStrings
+        assertFalse(elementStrings!!.containsAll(listOf("myFieldName", "myFunction")))
+    }
+}