Add more compiler plugin tests

compiler-plugin-test/src/test/java/io/ballerina/lib/data/yaml/compiler/CompilerPluginTest.java

@@ -146,4 +146,33 @@ public void testComplexUnionTypeWithUnsupportedTypeAndDuplicateFields() {
                 "invalid field: duplicate field found");
         Assert.assertEquals(errorDiagnosticsList.get(2).diagnosticInfo().messageFormat(), UNSUPPORTED_TYPE);
     }
+
+    @Test
+    public void testPackageWithErroneousCompilation() {
+        DiagnosticResult diagnosticResult =
+                CompilerPluginTestUtils.loadPackage("sample_package_11").getCompilation().diagnosticResult();
+        List<Diagnostic> errorDiagnosticsList = diagnosticResult.diagnostics().stream()
+                .filter(r -> r.diagnosticInfo().severity().equals(DiagnosticSeverity.ERROR)).toList();
+        Assert.assertEquals(errorDiagnosticsList.size(), 1);
+    }
+
+    @Test
+    public void testPackageWithOtherModuleImports() {
+        DiagnosticResult diagnosticResult =
+                CompilerPluginTestUtils.loadPackage("sample_package_12").getCompilation().diagnosticResult();
+        List<Diagnostic> errorDiagnosticsList = diagnosticResult.diagnostics().stream()
+                .filter(r -> r.diagnosticInfo().severity().equals(DiagnosticSeverity.ERROR)).toList();
+        Assert.assertEquals(errorDiagnosticsList.size(), 0);
+    }
+
+    @Test
+    public void testUnsupportedTupleType() {
+        DiagnosticResult diagnosticResult =
+                CompilerPluginTestUtils.loadPackage("sample_package_13").getCompilation().diagnosticResult();
+        List<Diagnostic> errorDiagnosticsList = diagnosticResult.diagnostics().stream()
+                .filter(r -> r.diagnosticInfo().severity().equals(DiagnosticSeverity.ERROR)).toList();
+        Assert.assertEquals(errorDiagnosticsList.size(), 2);
+        Assert.assertEquals(errorDiagnosticsList.get(0).diagnosticInfo().messageFormat(), UNSUPPORTED_TYPE);
+        Assert.assertEquals(errorDiagnosticsList.get(1).diagnosticInfo().messageFormat(), UNSUPPORTED_TYPE);
+    }
 }

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_1/sample.bal

@@ -16,6 +16,8 @@
 
 import ballerina/data.yaml;
 
+const VALUE = "1";
+
 public function main() returns error? {
     int|table<record {|string a;|}>|record {| int b;|} val = check yaml:parseString("1");
 }

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_11/Ballerina.toml

@@ -0,0 +1,4 @@
+[package]
+org = "yamldata_test"
+name = "sample_11"
+version = "0.1.0"

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_11/sample.bal

@@ -0,0 +1,23 @@
+// Copyright (c) 2024, WSO2 LLC. (https://www.wso2.com).
+//
+// WSO2 LLC. licenses this file to you 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.
+
+import ballerina/data.yaml;
+
+function call(string data) returns error? {
+    string value = "value";
+    anydata _ = check yaml:parseString(value);
+    int _ = "3";
+}

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_12/Ballerina.toml

@@ -0,0 +1,4 @@
+[package]
+org = "yamldata_test"
+name = "sample_12"
+version = "0.1.0"

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_12/sample.bal

@@ -0,0 +1,24 @@
+// Copyright (c) 2024, WSO2 LLC. (https://www.wso2.com).
+//
+// WSO2 LLC. licenses this file to you 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.
+
+import ballerina/lang.'int as i;
+import ballerina/data.yaml;
+
+function call(stream<byte[], error?> strm) returns error? {
+    string value = "value";
+    anydata _ = check yaml:parseStream(strm);
+    int _ = i:abs(12);
+}

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_13/Ballerina.toml

@@ -0,0 +1,4 @@
+[package]
+org = "yamldata_test"
+name = "sample_13"
+version = "0.1.0"

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_13/sample.bal

@@ -0,0 +1,26 @@
+// Copyright (c) 2024, WSO2 LLC. (https://www.wso2.com).
+//
+// WSO2 LLC. licenses this file to you 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.
+
+import ballerina/data.yaml;
+
+type Tuple [xml, xml...];
+type Tuple2 [xml, xml];
+
+public function main() returns error? {
+    byte[] bytes = [];
+    Tuple _ = check yaml:parseBytes(bytes);
+    Tuple2 _ = check yaml:parseBytes(bytes);
+}

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_5/sample.bal

@@ -22,4 +22,8 @@ type Data record {
     }
     string A;
     string B;
+    @display {
+        label: "C"
+    }
+    string C;
 };

compiler-plugin-test/src/test/resources/ballerina_sources/sample_package_8/sample.bal

@@ -21,6 +21,6 @@ type UnionType table<record {|int a;|}>|record {|string b;|};
 type IntersectionType UnionType & readonly;
 
 public function main() returns error? {
-    string str = string `{"a": 1, "b": "str"}`;
-    IntersectionType _ = check yaml:parseString(str);
+    byte[] bytes = [];
+    IntersectionType _ = check yaml:parseBytes(bytes);
 }

compiler-plugin/src/main/java/io/ballerina/lib/data/yaml/compiler/Constants.java

@@ -25,11 +25,13 @@
  */
 public class Constants {
 
+    private Constants() {
+    }
+
     static final String BALLERINA = "ballerina";
     static final String PARSE_STRING = "parseString";
     static final String PARSE_BYTES = "parseBytes";
     static final String PARSE_STREAM = "parseStream";
-    static final String TO_YAML_STRING = "toYamlString";
     static final String NAME = "Name";
     static final String YAML = "yaml";
     static final String DATA_YAML = "data.yaml";

compiler-plugin/src/main/java/io/ballerina/lib/data/yaml/compiler/YamlDataTypeValidator.java

@@ -113,6 +113,7 @@ public void perform(SyntaxNodeAnalysisContext ctx) {
                         processModuleVariableDeclarationNode((ModuleVariableDeclarationNode) member, ctx);
                 case TYPE_DEFINITION ->
                         processTypeDefinitionNode((TypeDefinitionNode) member, ctx);
+                default -> { }
             }
         }
     }
@@ -206,6 +207,8 @@ private void validateTupleType(TupleTypeSymbol tupleTypeSymbol, SyntaxNodeAnalys
         for (TypeSymbol memberType : tupleTypeSymbol.memberTypeDescriptors()) {
             validateExpectedType(memberType, ctx);
         }
+        Optional<TypeSymbol> restTypeSymbol = tupleTypeSymbol.restTypeDescriptor();
+        restTypeSymbol.ifPresent(typeSymbol -> validateExpectedType(typeSymbol, ctx));
     }
 
     private void validateRecordType(RecordTypeSymbol recordTypeSymbol, SyntaxNodeAnalysisContext ctx) {