/*******************************************************************************
* Copyright (c) 1998, 2015 Oracle and/or its affiliates. All rights reserved.
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License v1.0 and Eclipse Distribution License v. 1.0
* which accompanies this distribution.
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
* and the Eclipse Distribution License is available at
* http://www.eclipse.org/org/documents/edl-v10.php.
*
* Contributors:
* David McCann - 2.3 - Initial implementation
******************************************************************************/
package dbws.testing.sqlascollection;
//javase imports
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.StringReader;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.HashMap;
import java.util.Map;
import org.w3c.dom.Document;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
//java eXtension imports
import javax.annotation.PostConstruct;
import javax.annotation.PreDestroy;
import javax.servlet.ServletContext;
import javax.wsdl.WSDLException;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import javax.xml.namespace.QName;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.soap.MessageFactory;
import javax.xml.soap.SOAPException;
import javax.xml.soap.SOAPMessage;
import javax.xml.soap.SOAPPart;
import javax.xml.transform.Source;
import javax.xml.transform.TransformerException;
import javax.xml.transform.dom.DOMResult;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamSource;
import javax.xml.ws.BindingProvider;
import javax.xml.ws.Dispatch;
import javax.xml.ws.Endpoint;
import javax.xml.ws.Provider;
import javax.xml.ws.Service;
import javax.xml.ws.ServiceMode;
import javax.xml.ws.WebServiceProvider;
import static javax.xml.ws.BindingProvider.ENDPOINT_ADDRESS_PROPERTY;
import static javax.xml.ws.Service.Mode.MESSAGE;
import static javax.xml.ws.soap.SOAPBinding.SOAP11HTTP_BINDING;
//JUnit4 imports
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
//EclipseLink imports
import org.eclipse.persistence.descriptors.ClassDescriptor;
import org.eclipse.persistence.internal.databaseaccess.Platform;
import org.eclipse.persistence.internal.dbws.ProviderHelper;
import org.eclipse.persistence.internal.helper.ConversionManager;
import org.eclipse.persistence.internal.jpa.deployment.PersistenceUnitProcessor;
import org.eclipse.persistence.internal.jpa.metadata.MetadataProcessor;
import org.eclipse.persistence.internal.sessions.DatabaseSessionImpl;
import org.eclipse.persistence.internal.xr.ProjectHelper;
import org.eclipse.persistence.internal.xr.XRDynamicClassLoader;
import org.eclipse.persistence.internal.xr.XmlBindingsModel;
import org.eclipse.persistence.jaxb.JAXBContextProperties;
import org.eclipse.persistence.jaxb.xmlmodel.XmlBindings;
import org.eclipse.persistence.logging.AbstractSessionLog;
import org.eclipse.persistence.logging.SessionLog;
import org.eclipse.persistence.oxm.XMLContext;
import org.eclipse.persistence.oxm.XMLDescriptor;
import org.eclipse.persistence.oxm.XMLLogin;
import org.eclipse.persistence.oxm.XMLUnmarshaller;
import org.eclipse.persistence.platform.xml.XMLComparer;
import org.eclipse.persistence.platform.xml.XMLParser;
import org.eclipse.persistence.platform.xml.XMLPlatform;
import org.eclipse.persistence.platform.xml.XMLPlatformFactory;
import org.eclipse.persistence.sessions.DatabaseLogin;
import org.eclipse.persistence.sessions.DatabaseSession;
import org.eclipse.persistence.sessions.DatasourceLogin;
import org.eclipse.persistence.sessions.Project;
import org.eclipse.persistence.tools.dbws.DBWSBuilder;
import org.eclipse.persistence.tools.dbws.DBWSBuilderModel;
import org.eclipse.persistence.tools.dbws.DBWSBuilderModelProject;
import org.eclipse.persistence.tools.dbws.JSR109WebServicePackager;
import static org.eclipse.persistence.tools.dbws.DBWSBuilder.NO_SESSIONS_FILENAME;
import static org.eclipse.persistence.tools.dbws.DBWSBuilder.SESSIONS_FILENAME_KEY;
import static org.eclipse.persistence.tools.dbws.DBWSPackager.ArchiveUse.noArchive;
import static org.eclipse.persistence.tools.dbws.Util.DOM_PLATFORM_CLASSNAME;
import static org.eclipse.persistence.tools.dbws.Util.OR_PRJ_SUFFIX;
import static org.eclipse.persistence.tools.dbws.Util.TYPE_STR;
import static org.eclipse.persistence.tools.dbws.XRPackager.__nullStream;
//testing imports
import static dbws.testing.DBWSTestSuite.DATABASE_DDL_CREATE_KEY;
import static dbws.testing.DBWSTestSuite.DATABASE_DDL_DEBUG_KEY;
import static dbws.testing.DBWSTestSuite.DATABASE_DDL_DROP_KEY;
import static dbws.testing.DBWSTestSuite.DATABASE_DRIVER;
import static dbws.testing.DBWSTestSuite.DATABASE_PLATFORM;
import static dbws.testing.DBWSTestSuite.DATABASE_USERNAME_KEY;
import static dbws.testing.DBWSTestSuite.DATABASE_PASSWORD_KEY;
import static dbws.testing.DBWSTestSuite.DATABASE_URL_KEY;
import static dbws.testing.DBWSTestSuite.DEFAULT_DATABASE_DDL_CREATE;
import static dbws.testing.DBWSTestSuite.DEFAULT_DATABASE_DDL_DEBUG;
import static dbws.testing.DBWSTestSuite.DEFAULT_DATABASE_DDL_DROP;
import static dbws.testing.DBWSTestSuite.DEFAULT_DATABASE_PASSWORD;
import static dbws.testing.DBWSTestSuite.DEFAULT_DATABASE_URL;
import static dbws.testing.DBWSTestSuite.DEFAULT_DATABASE_USERNAME;
import static dbws.testing.DBWSTestSuite.SQLCOLLECTION;
import static dbws.testing.DBWSTestSuite.SQLCOLLECTION_NAMESPACE;
import static dbws.testing.DBWSTestSuite.SQLCOLLECTION_SERVICE_NAMESPACE;
import static dbws.testing.DBWSTestSuite.SQLCOLLECTION_SERVICE;
import static dbws.testing.DBWSTestSuite.SQLCOLLECTION_SERVICE_PORT;
import static dbws.testing.DBWSTestSuite.buildConnection;
import static dbws.testing.DBWSTestSuite.documentToString;
import static dbws.testing.DBWSTestSuite.runDdl;
@WebServiceProvider(
targetNamespace = SQLCOLLECTION_SERVICE_NAMESPACE,
serviceName = SQLCOLLECTION_SERVICE,
portName = SQLCOLLECTION_SERVICE_PORT
)
@ServiceMode(MESSAGE)
public class SQLAsCollectionTestSuite extends ProviderHelper implements Provider<SOAPMessage> {
static final String CREATE_SQLCOLLECTION_TABLE =
"CREATE TABLE IF NOT EXISTS sqlascollection (" +
"\nID NUMERIC NOT NULL," +
"\nNAME varchar(25)," +
"\nSINCE date," +
"\nPRIMARY KEY (ID)" +
"\n)";
static String[] POPULATE_SQLCOLLECTION_TABLE = new String[] {
"INSERT INTO sqlascollection (ID, NAME, SINCE) VALUES (1, 'mike', '2001-12-25')",
"INSERT INTO sqlascollection (ID, NAME, SINCE) VALUES (2, null,'2001-12-25')",
"INSERT INTO sqlascollection (ID, NAME, SINCE) VALUES (3, 'rick',NULL)"
};
static final String DROP_SQLCOLLECTION_TABLE =
"DROP TABLE sqlascollection";
static final String ENDPOINT_ADDRESS = "http://localhost:9999/" + SQLCOLLECTION + "Test";
// JUnit test fixtures
static Connection conn = null;
static ByteArrayOutputStream DBWS_SERVICE_STREAM = new ByteArrayOutputStream();
static ByteArrayOutputStream DBWS_SCHEMA_STREAM = new ByteArrayOutputStream();
static ByteArrayOutputStream DBWS_OR_STREAM = new ByteArrayOutputStream();
static ByteArrayOutputStream DBWS_OX_STREAM = new ByteArrayOutputStream();
static ByteArrayOutputStream DBWS_WSDL_STREAM = new ByteArrayOutputStream();
static XMLComparer comparer = new XMLComparer();
static XMLPlatform xmlPlatform = XMLPlatformFactory.getInstance().getXMLPlatform();
static XMLParser xmlParser = xmlPlatform.newXMLParser();
static Endpoint endpoint = null;
static QName portQName = null;
static Service testService = null;
static DBWSBuilder builder = new DBWSBuilder();
static final String username = System.getProperty(DATABASE_USERNAME_KEY, DEFAULT_DATABASE_USERNAME);
static final String password = System.getProperty(DATABASE_PASSWORD_KEY, DEFAULT_DATABASE_PASSWORD);
static final String url = System.getProperty(DATABASE_URL_KEY, DEFAULT_DATABASE_URL);
static boolean ddlCreate = false;
static boolean ddlDrop = false;
static boolean ddlDebug = false;
@BeforeClass
public static void setUp() throws WSDLException {
try {
conn = buildConnection();
}
catch (Exception e) {
e.printStackTrace();
}
String ddlCreateProp = System.getProperty(DATABASE_DDL_CREATE_KEY, DEFAULT_DATABASE_DDL_CREATE);
if ("true".equalsIgnoreCase(ddlCreateProp)) {
ddlCreate = true;
}
String ddlDropProp = System.getProperty(DATABASE_DDL_DROP_KEY, DEFAULT_DATABASE_DDL_DROP);
if ("true".equalsIgnoreCase(ddlDropProp)) {
ddlDrop = true;
}
String ddlDebugProp = System.getProperty(DATABASE_DDL_DEBUG_KEY, DEFAULT_DATABASE_DDL_DEBUG);
if ("true".equalsIgnoreCase(ddlDebugProp)) {
ddlDebug = true;
}
if (ddlCreate) {
runDdl(conn, CREATE_SQLCOLLECTION_TABLE, ddlDebug);
try {
Statement stmt = conn.createStatement();
for (int i = 0; i < POPULATE_SQLCOLLECTION_TABLE.length; i++) {
stmt.addBatch(POPULATE_SQLCOLLECTION_TABLE[i]);
}
stmt.executeBatch();
}
catch (SQLException e) {
if (ddlDebug) {
e.printStackTrace();
}
}
}
String builderString = DBWS_BUILDER_XML_USERNAME + username + DBWS_BUILDER_XML_PASSWORD +
password + DBWS_BUILDER_XML_URL + url + DBWS_BUILDER_XML_DRIVER + DATABASE_DRIVER +
DBWS_BUILDER_XML_PLATFORM + DATABASE_PLATFORM + DBWS_BUILDER_XML_MAIN;
XMLContext context = new XMLContext(new DBWSBuilderModelProject());
XMLUnmarshaller unmarshaller = context.createUnmarshaller();
DBWSBuilderModel builderModel =
(DBWSBuilderModel)unmarshaller.unmarshal(new StringReader(builderString));
builder.quiet = true;
//builder.setLogLevel(SessionLog.FINE_LABEL);
builder.setLogLevel(SessionLog.OFF_LABEL);
builder.properties = builderModel.properties;
builder.getProperties().put(SESSIONS_FILENAME_KEY, NO_SESSIONS_FILENAME);
builder.getTargetNamespace();
builder.operations = builderModel.operations;
builder.setPackager(new JSR109WebServicePackager(null, "WebServiceTestPackager", noArchive) {
@Override
public void start() {
}
});
builder.build(DBWS_SCHEMA_STREAM, __nullStream, DBWS_SERVICE_STREAM, DBWS_OR_STREAM,
DBWS_OX_STREAM, __nullStream, __nullStream, DBWS_WSDL_STREAM, __nullStream,
__nullStream, __nullStream, __nullStream, null);
endpoint = Endpoint.create(new SQLAsCollectionTestSuite());
endpoint.publish(ENDPOINT_ADDRESS);
QName serviceQName = new QName(SQLCOLLECTION_SERVICE_NAMESPACE, SQLCOLLECTION_SERVICE);
portQName = new QName(SQLCOLLECTION_SERVICE_NAMESPACE, SQLCOLLECTION_SERVICE_PORT);
testService = Service.create(serviceQName);
testService.addPort(portQName, SOAP11HTTP_BINDING, ENDPOINT_ADDRESS);
}
@AfterClass
public static void teardown() {
if (endpoint != null) {
endpoint.stop();
}
if (ddlDrop) {
runDdl(conn, DROP_SQLCOLLECTION_TABLE, ddlDebug);
}
}
@PreDestroy
public void destroy() {
super.destroy();
}
@Override
protected InputStream initXRServiceStream(ClassLoader parentClassLoader, ServletContext sc) {
return new ByteArrayInputStream(DBWS_SERVICE_STREAM.toByteArray());
}
@Override
protected InputStream initXRSchemaStream(ClassLoader parentClassLoader, ServletContext sc) {
return new ByteArrayInputStream(DBWS_SCHEMA_STREAM.toByteArray());
}
@Override
protected InputStream initWSDLInputStream(ClassLoader parentClassLoader, ServletContext sc) {
return new ByteArrayInputStream(DBWS_WSDL_STREAM.toByteArray());
}
@PostConstruct
public void init() {
super.init(new XRDynamicClassLoader(Thread.currentThread().getContextClassLoader()),
null, false);
}
@Override
public void logoutSessions() {
if (xrService.getORSession() != null) {
((DatabaseSession)xrService.getORSession()).logout();
}
if (xrService.getOXSession() != null) {
((DatabaseSession)xrService.getOXSession()).logout();
}
}
@Override
public void buildSessions() {
XRDynamicClassLoader xrdecl = new XRDynamicClassLoader(parentClassLoader);
DatasourceLogin login = new DatabaseLogin();
login.setUserName(username);
login.setPassword(password);
((DatabaseLogin) login).setConnectionString(url);
((DatabaseLogin) login).setDriverClassName(DATABASE_PLATFORM);
Platform platform = builder.getDatabasePlatform();
ConversionManager conversionManager = platform.getConversionManager();
if (conversionManager != null) {
conversionManager.setLoader(xrdecl);
}
login.setDatasourcePlatform(platform);
((DatabaseLogin)login).bindAllParameters();
((DatabaseLogin)login).setUsesStreamsForBinding(true);
Project orProject = null;
if (DBWS_OR_STREAM.size() != 0) {
MetadataProcessor processor = new MetadataProcessor(new XRPersistenceUnitInfo(xrdecl),
new DatabaseSessionImpl(login), xrdecl, false, true, false, false, false, null, null);
processor.setMetadataSource(new JPAMetadataSource(xrdecl, new StringReader(DBWS_OR_STREAM.toString())));
PersistenceUnitProcessor.processORMetadata(processor, true, PersistenceUnitProcessor.Mode.ALL);
processor.addNamedQueries();
orProject = processor.getProject().getProject();
} else {
orProject = new Project();
}
orProject.setName(builder.getProjectName().concat(OR_PRJ_SUFFIX));
orProject.setDatasourceLogin(login);
DatabaseSession databaseSession = orProject.createDatabaseSession();
if ("off".equalsIgnoreCase(builder.getLogLevel())) {
databaseSession.dontLogMessages();
} else {
databaseSession.setLogLevel(AbstractSessionLog.translateStringToLoggingLevel(builder.getLogLevel()));
}
xrService.setORSession(databaseSession);
orProject.convertClassNamesToClasses(xrdecl);
Project oxProject = null;
Map<String, OXMMetadataSource> metadataMap = new HashMap<String, OXMMetadataSource>();
StreamSource xml = new StreamSource(new StringReader(DBWS_OX_STREAM.toString()));
try {
JAXBContext jc = JAXBContext.newInstance(XmlBindingsModel.class);
Unmarshaller unmarshaller = jc.createUnmarshaller();
JAXBElement<XmlBindingsModel> jaxbElt = unmarshaller.unmarshal(xml, XmlBindingsModel.class);
XmlBindingsModel model = jaxbElt.getValue();
for (XmlBindings xmlBindings : model.getBindingsList()) {
metadataMap.put(xmlBindings.getPackageName(), new OXMMetadataSource(xmlBindings));
}
} catch (JAXBException jaxbex) {
jaxbex.printStackTrace();
}
Map<String, Map<String, OXMMetadataSource>> properties = new HashMap<String, Map<String, OXMMetadataSource>>();
properties.put(JAXBContextProperties.OXM_METADATA_SOURCE, metadataMap);
try {
org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext jCtx =
org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContextFactory.createContextFromOXM(parentClassLoader, properties);
oxProject = jCtx.getXMLContext().getSession(0).getProject();
} catch (JAXBException e) {
e.printStackTrace();
}
((XMLLogin)oxProject.getDatasourceLogin()).setPlatformClassName(DOM_PLATFORM_CLASSNAME);
((XMLLogin)oxProject.getDatasourceLogin()).setEqualNamespaceResolvers(false);
prepareDescriptors(oxProject, orProject, xrdecl);
ProjectHelper.fixOROXAccessors(orProject, oxProject);
xrService.setORSession(databaseSession);
xrService.setXMLContext(new XMLContext(oxProject));
xrService.setOXSession(xrService.getXMLContext().getSession(0));
}
static final String GETDATA_REQUEST =
"<env:Envelope xmlns:env=\"http://schemas.xmlsoap.org/soap/envelope/\">" +
"<env:Body>" +
"<srvc:"+SQLCOLLECTION_SERVICE+" xmlns:srvc=\"" + SQLCOLLECTION_SERVICE_NAMESPACE + "\" xmlns=\"" + SQLCOLLECTION_NAMESPACE + "\"/>" +
"</env:Body>" +
"</env:Envelope>";
@Test
public void getData() throws SOAPException, ParserConfigurationException, SAXException, IOException, TransformerException {
MessageFactory factory = MessageFactory.newInstance();
SOAPMessage request = factory.createMessage();
SOAPPart part = request.getSOAPPart();
DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
dbf.setNamespaceAware(true);
DocumentBuilder db = dbf.newDocumentBuilder();
DOMSource domSource = new DOMSource(db.parse(new InputSource(new StringReader(GETDATA_REQUEST))));
part.setContent(domSource);
Dispatch<SOAPMessage> dispatch = testService.createDispatch(portQName, SOAPMessage.class, Service.Mode.MESSAGE);
BindingProvider bp = dispatch;
Map<String, Object> rc = bp.getRequestContext();
rc.put(ENDPOINT_ADDRESS_PROPERTY, ENDPOINT_ADDRESS);
SOAPMessage response = null;
try {
response = dispatch.invoke(request);
}
catch (Exception e) {
e.printStackTrace();
fail("An unexpected exception occurred: " + e.getMessage());
}
if (response != null) {
Source src = response.getSOAPPart().getContent();
DOMResult result = new DOMResult();
getTransformer().transform(src, result);
Document resultDoc = (Document)result.getNode();
String resultString = documentToString(resultDoc);
Document controlDoc = xmlParser.parse(new StringReader(GETDATA_RESPONSE));
String controlString = documentToString(controlDoc);
assertTrue("Control document not same as instance document.\n Expected:\n" + controlString + "\nActual:\n" + resultString, controlString.equals(resultString));
}
}
static final String DBWS_BUILDER_XML_USERNAME =
"<?xml version=\"1.0\" encoding=\"UTF-8\"?>" +
"<dbws-builder xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\">" +
"<properties>" +
"<property name=\"projectName\">" + SQLCOLLECTION + "</property>" +
"<property name=\"logLevel\">off</property>" +
"<property name=\"username\">";
static final String DBWS_BUILDER_XML_PASSWORD =
"</property><property name=\"password\">";
static final String DBWS_BUILDER_XML_URL =
"</property><property name=\"url\">";
static final String DBWS_BUILDER_XML_DRIVER =
"</property><property name=\"driver\">";
static final String DBWS_BUILDER_XML_PLATFORM =
"</property><property name=\"platformClassname\">";
static final String DBWS_BUILDER_XML_MAIN =
"</property>" +
"</properties>" +
"<sql " +
"name=\""+SQLCOLLECTION_SERVICE+"\" " +
"returnType=\"sRecord\" " +
"isCollection=\"true\">" +
"<statement><![CDATA[select * from sqlascollection]]></statement>" +
"<build-statement><![CDATA[select * from sqlascollection where 0=1]]></build-statement>" +
"</sql>" +
"</dbws-builder>";
static final String GETDATA_RESPONSE =
"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n" +
"<SOAP-ENV:Envelope xmlns:SOAP-ENV=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns:xsd=\"http://www.w3.org/2001/XMLSchema\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\">\n" +
"<SOAP-ENV:Header/>\n" +
"<SOAP-ENV:Body>\n" +
"<srvc:"+SQLCOLLECTION_SERVICE+"Response xmlns=\"" + SQLCOLLECTION_NAMESPACE + "\" xmlns:srvc=\"" + SQLCOLLECTION_SERVICE_NAMESPACE + "\">\n" +
"<srvc:result>\n" +
"<sRecord>\n" +
"<id>1</id>\n" +
"<name>mike</name>\n" +
"<since>2001-12-25</since>\n" +
"</sRecord>\n" +
"<sRecord>\n" +
"<id>2</id>\n" +
"<name xsi:nil=\"true\"/>\n" +
"<since>2001-12-25</since>\n" +
"</sRecord>\n" +
"<sRecord>\n" +
"<id>3</id>\n" +
"<name>rick</name>\n" +
"<since xsi:nil=\"true\"/>\n" +
"</sRecord>\n" +
"</srvc:result>\n" +
"</srvc:"+SQLCOLLECTION_SERVICE+"Response>\n" +
"</SOAP-ENV:Body>\n" +
"</SOAP-ENV:Envelope>";
}