Fix failing tests

ballerina-platform · Jul 22, 2024 · 1dd1eb9 · 1dd1eb9
1 parent 5df9d7d
commit 1dd1eb9
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/compiler/ballerina-lang/src/main/java/io/ballerina/compiler/api/impl/ReferenceFinder.java b/compiler/ballerina-lang/src/main/java/io/ballerina/compiler/api/impl/ReferenceFinder.java
@@ -898,7 +898,6 @@ public void visit(BLangInvocation invocationExpr) {
     @Override
     public void visit(BLangTypeInit typeInit) {
         find(typeInit.userDefinedType);
-        find(typeInit.argsExpr);
         find(typeInit.initInvocation);
     }
 

diff --git a/...est/src/test/java/io/ballerina/semantic/api/test/allreferences/FindAllReferencesTest.java b/...est/src/test/java/io/ballerina/semantic/api/test/allreferences/FindAllReferencesTest.java
@@ -26,6 +26,7 @@
 import io.ballerina.tools.text.LinePosition;
 import io.ballerina.tools.text.LineRange;
 import org.ballerinalang.test.BCompileUtil;
+import org.testng.annotations.AfterClass;
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
@@ -143,4 +144,9 @@ protected static void assertLocations(List<Location> locations, List<Location> e
             assertTrue(lineRanges.contains(expLocation.lineRange()));
         }
     }
+
+    @AfterClass
+    public void tearDown() {
+        model = null;
+    }
 }
diff --git a/...src/test/java/io/ballerina/semantic/api/test/allreferences/FindRefsInClassObjectTest.java b/...src/test/java/io/ballerina/semantic/api/test/allreferences/FindRefsInClassObjectTest.java
@@ -18,6 +18,7 @@
 
 package io.ballerina.semantic.api.test.allreferences;
 
+import org.testng.annotations.AfterClass;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
 
@@ -54,7 +55,8 @@ public Object[][] getLookupPositions() {
                         List.of(location(24, 13, 20))
                 },
                 {30, 13, location(30, 13, 17),
-                        List.of(location(30, 13, 17))
+                        List.of(location(30, 13, 17),
+                                location(52, 17, 20))
                 },
                 {34, 13, location(34, 13, 20),
                         List.of(location(34, 13, 20),
@@ -79,4 +81,9 @@ public Object[][] getLookupPositions() {
     public String getTestSourcePath() {
         return "test-src/find-all-ref/find_refs_in_class_object.bal";
     }
+
+    @AfterClass
+    public void tearDown() {
+        super.tearDown();
+    }
 }