Refactor SQL references and enhance method signatures in test cases and SQL files

Author: xterao

src/test/kotlin/org/domaframework/doma/intellij/reference/SqlReferenceTestCase.kt

@@ -27,15 +27,15 @@ import org.domaframework.doma.intellij.psi.SqlTypes
 import org.jetbrains.kotlin.idea.core.util.toPsiFile
 
 class SqlReferenceTestCase : DomaSqlTest() {
-    val testPackage = "reference"
-    val testDaoName = "ReferenceTestDao"
+    private val testPackage = "reference"
+    private val testDaoName = "ReferenceTestDao"
 
-    val forItemResolve = "SqlElIdExprImpl(EL_ID_EXPR)"
-    val forItemFieldAccessResolve = "SqlElFieldAccessExprImpl(EL_FIELD_ACCESS_EXPR)"
-    val daoParameterResolve = "PsiParameter"
-    val fieldResolve = "PsiField"
-    val methodResolve = "PsiMethod"
-    val classResolve = "PsiClass"
+    private val forItemResolve = "SqlElIdExprImpl(EL_ID_EXPR)"
+    private val forItemFieldAccessResolve = "SqlElFieldAccessExprImpl(EL_FIELD_ACCESS_EXPR)"
+    private val daoParameterResolve = "PsiParameter"
+    private val fieldResolve = "PsiField"
+    private val methodResolve = "PsiMethod"
+    private val classResolve = "PsiClass"
 
     override fun setUp() {
         super.setUp()
@@ -67,11 +67,15 @@ class SqlReferenceTestCase : DomaSqlTest() {
             mapOf(
                 "detail" to listOf("$daoParameterResolve:detail"),
                 "getFirstEmployee" to listOf("$methodResolve:getFirstEmployee"),
+                "getCustomNumber" to listOf("$methodResolve:getCustomNumber"),
+                "projectCategory" to listOf("$fieldResolve:projectCategory"),
                 "projectNumber" to listOf("$fieldResolve:projectNumber"),
                 "projects" to listOf("$daoParameterResolve:projects"),
                 "project" to listOf(forItemResolve),
                 "id" to listOf(null),
+                "null" to listOf(null),
                 "employeeId" to listOf("$fieldResolve:employeeId"),
+                "isNotBlank" to listOf("$methodResolve:isNotBlank"),
             ),
         )
     }
@@ -85,7 +89,10 @@ class SqlReferenceTestCase : DomaSqlTest() {
                 "projectCategory" to listOf("$fieldResolve:projectCategory"),
                 "getTermNumber" to listOf("$methodResolve:getTermNumber"),
                 "getFirstEmployee" to listOf("$methodResolve:getFirstEmployee"),
+                "getCustomNumber" to listOf("$methodResolve:getCustomNumber"),
+                "detail" to listOf("$daoParameterResolve:detail"),
                 "employeeId" to listOf("$fieldResolve:employeeId"),
+                "toString" to listOf("$methodResolve:toString"),
             ),
         )
     }
@@ -179,6 +186,7 @@ class SqlReferenceTestCase : DomaSqlTest() {
         for (reference in references) {
             val resolveResult = reference.references.firstOrNull()?.resolve()
             val expectedResults = resolveExpects[reference.text]
+            println("Reference: ${reference.text}, Resolve Result: ${resolveResult?.toString()}, Expected Results: $expectedResults")
             assertTrue(expectedResults?.contains(resolveResult?.toString()) == true)
         }
     }

src/test/testData/src/main/java/doma/example/dao/reference/ReferenceTestDao.java

@@ -14,7 +14,7 @@ public interface ReferenceTestDao {
   Project referenceEntityProperty(List<Project> projects, ProjectDetail detail);
 
   @Select
-  ProjectDetail referenceStaticField();
+  ProjectDetail referenceStaticField(ProjectDetail detail);
 
   @Select
   List<Employee> referenceListFieldMethod(List<Employee> employeesList);

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

@@ -39,4 +39,8 @@ public void addTermNumber(){
     projectNumber++;
   }
 
+  public static String getCustomNumber(String prefix) {
+    return prefix + projectNumber.toString();
+  }
+
 }

src/test/testData/src/main/resources/META-INF/doma/example/dao/reference/ReferenceTestDao/referenceEntityProperty.sql

@@ -1,7 +1,11 @@
 SELECT *
   FROM project_detail
  WHERE detail_id = /* detail.getFirstEmployee().employeeId */0
+   /*%if (detail != null && @isNotBlank(detail.projectCategory)) */
+   AND detail_id = /* detail.getCustomNumber(detail.projectCategory) */'number'
    /*%for project : projects */
    AND project_id = /* project.id */0
+   AND detail_sub_id = /* detail.getCustomNumber(project.projectCategory) */'sub'
    /*%end*/
-OR number = /* detail.projectNumber */
+   /*%end*/
+    OR number = /* detail.projectNumber */0

src/test/testData/src/main/resources/META-INF/doma/example/dao/reference/ReferenceTestDao/referenceStaticField.sql

@@ -1,4 +1,5 @@
 SELECT *
   FROM project_detail
  WHERE category = /* @doma.example.entity.ProjectDetail@projectCategory */'category'
-   AND number = /* @doma.example.entity.Project@getFirstEmployee().employeeId */'9999'
+   AND number = /* @doma.example.entity.Project@getFirstEmployee().employeeId.toString() */'9999'
+    OR number = /* @doma.example.entity.ProjectDetail@getCustomNumber(detail.projectCategory) */'number'