Add Parameter Element Completion Test

Author: xterao

src/test/kotlin/org/domaframework/doma/intellij/complate/sql/SqlCompleteTest.kt

@@ -43,6 +43,10 @@ class SqlCompleteTest : DomaSqlTest() {
             "SqlCompleteTestDao/completeDirectiveFieldInsideFor.sql",
             "SqlCompleteTestDao/completeConcatenationOperator.sql",
             "SqlCompleteTestDao/completeComparisonOperator.sql",
+            "SqlCompleteTestDao/completeParameterFirst.sql",
+            "SqlCompleteTestDao/completeParameterFirstProperty.sql",
+            "SqlCompleteTestDao/completeParameterSecond.sql",
+            "SqlCompleteTestDao/completeParameterSecondProperty.sql",
         )
         myFixture.enableInspections(SqlBindVariableValidInspector())
     }
@@ -227,6 +231,32 @@ class SqlCompleteTest : DomaSqlTest() {
         )
     }
 
+    fun testCompleteParameter() {
+        innerDirectiveCompleteTest(
+            "SqlCompleteTestDao/completeParameterFirst.sql",
+            listOf("employee"),
+            listOf("employeeId", "department", "rank", "startWith"),
+        )
+
+        innerDirectiveCompleteTest(
+            "SqlCompleteTestDao/completeParameterFirstProperty.sql",
+            listOf("employeeId", "department", "rank"),
+            listOf("employee"),
+        )
+
+        innerDirectiveCompleteTest(
+            "SqlCompleteTestDao/completeParameterSecond.sql",
+            listOf("employee"),
+            listOf("employeeId", "department", "rank", "startWith"),
+        )
+
+        innerDirectiveCompleteTest(
+            "SqlCompleteTestDao/completeParameterSecondProperty.sql",
+            listOf("managerId"),
+            listOf("employee", "department", "rank"),
+        )
+    }
+
     private fun innerDirectiveCompleteTest(
         sqlFileName: String,
         expectedSuggestions: List<String>,

src/test/testData/src/main/java/doma/example/dao/SqlCompleteTestDao.java

@@ -57,4 +57,16 @@ interface SqlCompleteTestDao {
   @Insert(sqlFile = true)
   int completeConcatenationOperator(Employee employee,Integer point);
 
+  @Select
+  Employee completeParameterFirst(Employee employee);
+
+  @Select
+  Employee completeParameterFirstProperty(Employee employee);
+
+  @Select
+  Employee completeParameterSecond(Employee employee);
+
+  @Select
+  Employee completeParameterSecondProperty(Employee employee);
+
 }

src/test/testData/src/main/java/doma/example/entity/Employee.java

@@ -27,4 +27,7 @@ private String getEmployeeRank() {
     return rank;
   }
 
+  public Integer employeeParam(Integer p1, Integer p2) {
+    return p1 + p2;
+  }
 }

src/test/testData/src/main/resources/META-INF/doma/example/dao/SqlCompleteTestDao/completeParameterFirst.sql

@@ -0,0 +1,2 @@
+select * from employee
+where id = /* employee.department.startWith(<caret>) */1

src/test/testData/src/main/resources/META-INF/doma/example/dao/SqlCompleteTestDao/completeParameterFirstProperty.sql

@@ -0,0 +1 @@
+select * from employee where id = /* employee.department.startWith(employee.<caret>) */1

src/test/testData/src/main/resources/META-INF/doma/example/dao/SqlCompleteTestDao/completeParameterSecond.sql

@@ -0,0 +1 @@
+select * from employee where id = /* employee.employeeParam(3, <caret>) */1

src/test/testData/src/main/resources/META-INF/doma/example/dao/SqlCompleteTestDao/completeParameterSecondProperty.sql

@@ -0,0 +1,2 @@
+select * from employee
+where id = /* employee.employeeParam(employee.employeeId, employee.<caret>m) */1