Add more tests for user defined import prefix

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

@@ -209,4 +209,20 @@ public void testComplexUnionTypeCaseWhenUserDefinedModulePrefix() {
         Assert.assertEquals(errorDiagnosticsList.get(0).diagnosticInfo().messageFormat(),
                 "invalid type: expected a record type");
     }
+
+    @Test
+    public void testComplexUnionTypeCaseWhenUserDefinedModulePrefix2() {
+        DiagnosticResult diagnosticResult =
+                CompilerPluginTestUtils.loadPackage("sample_package_12").getCompilation().diagnosticResult();
+        List<Diagnostic> errorDiagnosticsList = diagnosticResult.diagnostics().stream()
+                .filter(r -> r.diagnosticInfo().severity().equals(DiagnosticSeverity.ERROR))
+                .collect(Collectors.toList());
+        Assert.assertEquals(errorDiagnosticsList.size(), 3);
+        Assert.assertEquals(errorDiagnosticsList.get(0).diagnosticInfo().messageFormat(),
+                "invalid type: expected a record type");
+        Assert.assertEquals(errorDiagnosticsList.get(1).diagnosticInfo().messageFormat(),
+                "invalid type: expected a record type");
+        Assert.assertEquals(errorDiagnosticsList.get(2).diagnosticInfo().messageFormat(),
+                "invalid type: expected a record type");
+    }
 }

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

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

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

@@ -0,0 +1,6 @@
+import ballerina/data.xmldata as xd;
+
+public function testFunc() returns error? {
+    string str = string `{"a": 1, "b": "str"}`;
+    UnionType _ = check xd:parseString(str);
+}

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

@@ -0,0 +1,6 @@
+import ballerina/data.xmldata as xd2;
+
+public function testFunc2() returns error? {
+    string str = string `{"a": 1, "b": "str"}`;
+    UnionType _ = check xd2:parseString(str);
+}

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

@@ -0,0 +1,6 @@
+import ballerina/data.xmldata;
+
+public function testFunc3() returns error? {
+    string str = string `{"a": 1, "b": "str"}`;
+    UnionType _ = check xmldata:parseString(str);
+}

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

@@ -0,0 +1 @@
+type UnionType record {|int a;|}|record {|string b;|};