| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF 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. |
| */ |
| |
| package org.apache.axis2.json.gson; |
| |
| import com.google.gson.Gson; |
| import org.apache.axiom.om.OMAbstractFactory; |
| import org.apache.axiom.om.OMElement; |
| import org.apache.axiom.om.OMFactory; |
| import org.apache.axiom.om.OMNamespace; |
| import org.apache.axiom.om.OMOutputFormat; |
| import org.apache.axiom.soap.SOAPEnvelope; |
| import org.apache.axiom.soap.SOAPFactory; |
| import org.apache.axis2.Constants; |
| import org.apache.axis2.context.ConfigurationContext; |
| import org.apache.axis2.context.MessageContext; |
| import org.apache.axis2.description.AxisMessage; |
| import org.apache.axis2.description.AxisOperation; |
| import org.apache.axis2.description.AxisOperationFactory; |
| import org.apache.axis2.description.AxisService; |
| import org.apache.axis2.engine.AxisConfiguration; |
| import org.apache.axis2.json.factory.JsonConstant; |
| import org.apache.axis2.wsdl.WSDLConstants; |
| import org.apache.ws.commons.schema.XmlSchema; |
| import org.apache.ws.commons.schema.XmlSchemaCollection; |
| import org.junit.After; |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| import javax.xml.namespace.QName; |
| import javax.xml.transform.stream.StreamSource; |
| import java.io.ByteArrayOutputStream; |
| import java.io.FileInputStream; |
| import java.io.InputStream; |
| import java.io.OutputStream; |
| |
| public class JsonFormatterTest { |
| MessageContext outMsgContext; |
| String contentType; |
| String jsonString; |
| SOAPEnvelope soapEnvelope; |
| OMOutputFormat outputFormat; |
| OutputStream outputStream; |
| @Before |
| public void setUp() throws Exception { |
| contentType = "application/json-impl"; |
| SOAPFactory soapFactory = OMAbstractFactory.getSOAP11Factory(); |
| soapEnvelope = soapFactory.getDefaultEnvelope(); |
| outputFormat = new OMOutputFormat(); |
| outputStream = new ByteArrayOutputStream(); |
| |
| outMsgContext = new MessageContext(); |
| outMsgContext.setProperty(Constants.Configuration.CHARACTER_SET_ENCODING, "UTF-8"); |
| } |
| |
| @After |
| public void tearDown() throws Exception { |
| outputStream.close(); |
| } |
| |
| @Test |
| public void testWriteToFaultMessage() throws Exception { |
| jsonString = "{\"Fault\":{\"faultcode\":\"soapenv:Server\",\"faultstring\":\"javax.xml.stream.XMLStreamException\",\"detail\":\"testFaultMsg\"}}"; |
| outMsgContext.setProcessingFault(true); |
| soapEnvelope.getBody().addChild(createFaultOMElement()); |
| outMsgContext.setEnvelope(soapEnvelope); |
| JsonFormatter jsonFormatter = new JsonFormatter(); |
| jsonFormatter.writeTo(outMsgContext, outputFormat, outputStream, false); |
| String faultMsg = outputStream.toString(); |
| Assert.assertEquals(jsonString , faultMsg); |
| } |
| |
| |
| @Test |
| public void testWriteToXMLtoJSON() throws Exception { |
| jsonString = "{\"response\":{\"return\":{\"name\":\"kate\",\"age\":\"35\",\"gender\":\"female\"}}}"; |
| String fileName = "test-resources/custom_schema/testSchema_1.xsd"; |
| InputStream is = new FileInputStream(fileName); |
| XmlSchemaCollection schemaCol = new XmlSchemaCollection(); |
| XmlSchema schema = schemaCol.read(new StreamSource(is)); |
| QName elementQName = new QName("http://www.w3schools.com", "response"); |
| ConfigurationContext configCtxt = new ConfigurationContext(new AxisConfiguration()); |
| outMsgContext.setConfigurationContext(configCtxt); |
| AxisOperation axisOperation = AxisOperationFactory.getAxisOperation(AxisOperation.MEP_CONSTANT_IN_OUT); |
| AxisMessage message = new AxisMessage(); |
| message.setElementQName(elementQName); |
| axisOperation.addMessage(message , WSDLConstants.MESSAGE_LABEL_OUT_VALUE); |
| outMsgContext.setAxisOperation(axisOperation); |
| AxisService axisService = new AxisService("testService"); |
| axisService.addSchema(schema); |
| outMsgContext.setAxisService(axisService); |
| soapEnvelope.getBody().addChild(getResponseOMElement()); |
| outMsgContext.setEnvelope(soapEnvelope); |
| JsonFormatter jsonFormatter = new JsonFormatter(); |
| jsonFormatter.writeTo(outMsgContext, outputFormat , outputStream , false); |
| String response = outputStream.toString(); |
| Assert.assertEquals(jsonString, response); |
| } |
| |
| |
| @Test |
| public void testWriteToJSON() throws Exception { |
| Person person = new Person(); |
| person.setName("Leo"); |
| person.setAge(27); |
| person.setGender("Male"); |
| person.setSingle(true); |
| outMsgContext.setProperty(JsonConstant.RETURN_OBJECT, person); |
| outMsgContext.setProperty(JsonConstant.RETURN_TYPE, Person.class); |
| jsonString = "{\""+ JsonConstant.RESPONSE +"\":" + new Gson().toJson(person, Person.class) + "}"; |
| |
| JsonFormatter jsonFormatter = new JsonFormatter(); |
| jsonFormatter.writeTo(outMsgContext, outputFormat, outputStream, false); |
| String personString = outputStream.toString(); |
| Assert.assertEquals(jsonString, personString); |
| |
| } |
| |
| |
| private OMElement createFaultOMElement() { |
| OMFactory omFactory = OMAbstractFactory.getOMFactory(); |
| OMNamespace ns = omFactory.createOMNamespace("", ""); |
| OMElement faultCode = omFactory.createOMElement("faultcode", ns); |
| faultCode.setText("soapenv:Server"); |
| OMElement faultString = omFactory.createOMElement("faultstring", ns); |
| faultString.setText("javax.xml.stream.XMLStreamException"); |
| OMElement detail = omFactory.createOMElement("detail", ns); |
| detail.setText("testFaultMsg"); |
| OMElement fault = omFactory.createOMElement("Fault", ns); |
| fault.addChild(faultCode); |
| fault.addChild(faultString); |
| fault.addChild(detail); |
| return fault; |
| } |
| |
| private OMElement getResponseOMElement() { |
| OMFactory omFactory = OMAbstractFactory.getOMFactory(); |
| OMNamespace ns = omFactory.createOMNamespace("", ""); |
| |
| OMElement response = omFactory.createOMElement("response", ns); |
| OMElement ret = omFactory.createOMElement("return", ns); |
| OMElement name = omFactory.createOMElement("name", ns); |
| name.setText("kate"); |
| OMElement age = omFactory.createOMElement("age", ns); |
| age.setText("35"); |
| OMElement gender = omFactory.createOMElement("gender", ns); |
| gender.setText("female"); |
| ret.addChild(name); |
| ret.addChild(age); |
| ret.addChild(gender); |
| response.addChild(ret); |
| return response; |
| } |
| |
| public static class Person { |
| |
| private String name; |
| private int age; |
| private String gender; |
| private boolean single; |
| |
| public String getName() { |
| return name; |
| } |
| |
| public void setName(String name) { |
| this.name = name; |
| } |
| |
| public int getAge() { |
| return age; |
| } |
| |
| public void setAge(int age) { |
| this.age = age; |
| } |
| |
| public String getGender() { |
| return gender; |
| } |
| |
| public void setGender(String gender) { |
| this.gender = gender; |
| } |
| |
| public boolean isSingle() { |
| return single; |
| } |
| |
| public void setSingle(boolean single) { |
| this.single = single; |
| } |
| } |
| } |