Added Tests

Refer to: https://github.com/NiasSt90/jaxb-bug1135

Signed-off-by: Lin Gao <[email protected]>

jaxb-ri/runtime/impl/src/test/java/com/sun/xml/bind/v2/schemagen/MarshallingAbstractTest.java

@@ -0,0 +1,158 @@
+/*
+ * Copyright (c) 1997, 2019 Oracle and/or its affiliates. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Distribution License v. 1.0, which is available at
+ * http://www.eclipse.org/org/documents/edl-v10.php.
+ *
+ * SPDX-License-Identifier: BSD-3-Clause
+ */
+
+package com.sun.xml.bind.v2.schemagen;
+
+import org.junit.Assert;
+import org.junit.Test;
+
+import javax.xml.bind.annotation.XmlAccessType;
+import javax.xml.bind.annotation.XmlAccessorType;
+import javax.xml.bind.annotation.XmlElement;
+import javax.xml.bind.annotation.XmlElementWrapper;
+import javax.xml.bind.annotation.XmlRootElement;
+import javax.xml.bind.annotation.XmlSeeAlso;
+import javax.xml.bind.annotation.XmlType;
+import javax.xml.bind.annotation.XmlValue;
+import javax.xml.bind.JAXBContext;
+import javax.xml.bind.JAXBElement;
+import javax.xml.bind.Marshaller;
+import javax.xml.bind.Unmarshaller;
+import javax.xml.transform.stream.StreamSource;
+
+import java.io.StringReader;
+import java.io.StringWriter;
+import java.lang.reflect.Method;
+import java.util.ArrayList;
+import java.util.Collection;
+
+import junit.framework.TestCase;
+import junit.textui.TestRunner;
+
+public class MarshallingAbstractTest extends TestCase {
+    public static void main(String[] args) {
+        TestRunner.run(MarshallingAbstractTest.class);
+    }
+
+    @XmlSeeAlso({B.class, C.class})
+    abstract static class A {
+        //marshal/unmarshal of list elements containing elements of type = A will work without @XmlValue annotation...
+        @XmlValue
+        protected String value;
+    }
+
+    @XmlAccessorType(XmlAccessType.FIELD)
+    @XmlType(name = "ClassType1")
+    public static class B extends A {
+        public B() {}
+        public B(String value) {
+            this.value = value;
+        }
+    }
+
+    @XmlAccessorType(XmlAccessType.FIELD)
+    @XmlType(name = "ClassType2")
+    public static class C extends A {
+        public C() {}
+        public C(String value) {
+            this.value = value;
+        }
+    }
+
+    @XmlRootElement(name="root")
+    @XmlAccessorType(XmlAccessType.FIELD)
+    static class Mapping {
+        @XmlElementWrapper(name = "list")
+        @XmlElement(name="element")
+        Collection<A> list = new ArrayList<A>();
+        A element1;
+        A element2;
+    }
+
+    @Test
+    public void testMarshalSingleElement() throws Exception {
+        JAXBContext jc = JAXBContext.newInstance(Mapping.class);
+        Marshaller marshaller = jc.createMarshaller();
+        marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
+        StringWriter resultWriter = new StringWriter();
+        Mapping mapping = new Mapping();
+        mapping.element1 = new B("B1");
+        mapping.element2 = new C("C1");
+        marshaller.marshal(mapping, resultWriter);
+
+        String expectedXml1 = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n<root>\n"
+                + "    <list/>\n"
+                + "    <element1 xsi:type=\"ClassType1\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">B1</element1>\n"
+                + "    <element2 xsi:type=\"ClassType2\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">C1</element2>\n"
+                + "</root>\n";
+        Assert.assertEquals(resultWriter.toString(), expectedXml1);
+    }
+
+    @Test
+    public void testMarshalCollection() throws Exception {
+        JAXBContext jc = JAXBContext.newInstance(Mapping.class);
+        Marshaller marshaller = jc.createMarshaller();
+        marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
+        StringWriter resultWriter = new StringWriter();
+        Mapping mapping = new Mapping();
+        mapping.element1 = new B("B1");
+        mapping.element2 = new C("C1");
+        mapping.list.add(new B("B"));
+        mapping.list.add(new C("C"));
+        String expectedXml2 = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n<root>\n"
+                + "    <list>\n"
+                + "        <element xsi:type=\"ClassType1\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">B</element>\n"
+                + "        <element xsi:type=\"ClassType2\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">C</element>\n"
+                + "    </list>\n"
+                + "    <element1 xsi:type=\"ClassType1\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">B1</element1>\n"
+                + "    <element2 xsi:type=\"ClassType2\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">C1</element2>\n"
+                + "</root>\n";
+        marshaller.marshal(mapping, resultWriter);
+        Assert.assertEquals(resultWriter.toString(), expectedXml2);
+    }
+
+    @Test
+    public void testUnmarshalSingleElement() throws Exception {
+        JAXBContext jc = JAXBContext.newInstance(Mapping.class);
+        Unmarshaller unmarshaller = jc.createUnmarshaller();
+        //works without list..
+        String sourceXml1 = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n<root>\n"
+                + "    <element1 xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:type=\"ClassType1\">B1</element1>\n"
+                + "    <element2 xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:type=\"ClassType2\">C1</element2>\n"
+                + "</root>\n";
+        JAXBElement<Mapping> element = unmarshaller.unmarshal(new StreamSource(new StringReader(sourceXml1)), Mapping.class);
+        Assert.assertNotNull(element.getValue());
+        Assert.assertEquals(B.class, element.getValue().element1.getClass());
+        Assert.assertEquals(C.class, element.getValue().element2.getClass());
+    }
+
+    @Test
+    public void testUnmarshalCollection() throws Exception {
+        JAXBContext jc = JAXBContext.newInstance(Mapping.class);
+        Unmarshaller unmarshaller = jc.createUnmarshaller();
+        //don't work -> try to instantiate the abstract class
+        String sourceXml2 = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n<root>\n"
+                + "    <list>\n"
+                + "        <element xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:type=\"ClassType1\">B</element>\n"
+                + "        <element xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:type=\"ClassType2\">C</element>\n"
+                + "    </list>\n"
+                + "    <element1 xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:type=\"ClassType1\">B1</element1>\n"
+                + "    <element2 xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:type=\"ClassType2\">C1</element2>\n"
+                + "</root>\n";
+
+        try {
+            JAXBElement<Mapping> element = unmarshaller.unmarshal(new StreamSource(new StringReader(sourceXml2)), Mapping.class);
+            Assert.assertEquals(2, element.getValue().list.size());
+        }
+        catch (Throwable e) {
+            Assert.fail();
+        }
+    }
+}