ServletClient.java.src

/*
 * Copyright (c) 2013, 2020 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 v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.ts.tests.jms.ee20.resourcedefs.annotations;

import com.sun.ts.lib.util.*;
import com.sun.ts.lib.porting.*;
import com.sun.ts.tests.jms.common.*;

import java.io.*;
import jakarta.servlet.*;
import jakarta.servlet.http.*;
import jakarta.servlet.annotation.WebServlet;
import java.util.Enumeration;
import java.util.Properties;
import java.util.ArrayList;
import jakarta.transaction.*;
import javax.naming.*;
import jakarta.jms.*;
import jakarta.inject.Inject;
import jakarta.annotation.Resource;

//-------------------------------------
// JMS Destination Resource Definitions
//-------------------------------------
@JMSDestinationDefinition(
      description="Define Queue ServletMyTestQueue",
      interfaceName="jakarta.jms.Queue",
      name="java:global/env/ServletMyTestQueue",
      destinationName="ServletMyTestQueue"
 )

@JMSDestinationDefinition(
      description="Define Topic ServletMyTestTopic",
      interfaceName="jakarta.jms.Topic",
      name="java:app/env/ServletMyTestTopic",
      destinationName="ServletMyTestTopic"
 )

//-------------------------------------------
// JMS ConnectionFactory Resource Definitions
//-------------------------------------------
@JMSConnectionFactoryDefinition(
      description="Define ConnectionFactory ServletMyTestConnectionFactory",
      interfaceName="jakarta.jms.ConnectionFactory",
      name="java:global/ServletMyTestConnectionFactory",
      user = "@user@",
      password = "@password@"
 )

@JMSConnectionFactoryDefinition(
      description="Define QueueConnectionFactory ServletMyTestQueueConnectionFactory",
      interfaceName="jakarta.jms.QueueConnectionFactory",
      name="java:app/ServletMyTestQueueConnectionFactory",
      user = "@user@",
      password = "@password@"
 )

@JMSConnectionFactoryDefinition(
      description="Define TopicConnectionFactory ServletMyTestTopicConnectionFactory",
      interfaceName="jakarta.jms.TopicConnectionFactory",
      name="java:module/ServletMyTestTopicConnectionFactory",
      user = "@user@",
      password = "@password@"
 )

@JMSConnectionFactoryDefinition(
      description="Define Durable TopicConnectionFactory ServletMyTestDurableTopicConnectionFactory",
      interfaceName="jakarta.jms.TopicConnectionFactory",
      name="java:comp/env/jms/ServletMyTestDurableTopicConnectionFactory",
      user = "@user@",
      password = "@password@",
      clientId = "MyClientID",
      properties = { "Property1=10", "Property2=20" },
      transactional = false,
      maxPoolSize = 30,
      minPoolSize = 20
 )

@WebServlet("/ServletTest")
public class ServletClient extends HttpServlet {
    private static final long serialVersionUID = 1L;

    // JMS objects
    protected transient ConnectionFactory	cf = null;
    protected transient ConnectionFactory	dcf = null;
    protected transient ConnectionFactory	cfra = null;
    protected transient QueueConnectionFactory 	qcf = null;
    protected transient TopicConnectionFactory 	tcf = null;
    protected transient TopicConnectionFactory 	dtcf = null;
    protected transient Topic 			topic = null;
    protected transient Topic 			topica = null;
    protected transient Queue 			queue = null;
    protected transient JMSContext              context = null;
    protected transient JMSConsumer             consumerQ = null;
    protected transient JMSProducer             producerQ = null;
    protected transient JMSConsumer             consumerT = null;
    protected transient JMSProducer             producerT = null;
    protected boolean			        queueTest = false;

    // Harness req's
    protected Properties              		harnessProps = null;

    // properties read from ts.jte file
    protected long                            	timeout;
    protected String                          	user;
    protected String                          	password;
    protected String                          	mode;

    /*
     * doCleanup()
     */
    private void doCleanup() throws Exception {
        try {
            if(queueTest && consumerQ != null) {
                TestUtil.logMsg("Flush any messages left on Queue");
		Message rmsg = consumerQ.receive(timeout);
                while (rmsg != null) {
                    rmsg = consumerQ.receiveNoWait();
                    if (rmsg == null) {
                        rmsg = consumerQ.receiveNoWait();
                    }
		}
	        consumerQ.close();
             }
            if(consumerT != null)
	        consumerT.close();
	    TestUtil.logMsg("Close JMSContext Objects");
            if(context != null)
                context.close();
	} catch (Exception e) {
	    TestUtil.logErr("Caught exception: " + e);
	    throw new Exception("doCleanup failed!", e);
	}
    }

    public void init(ServletConfig config) throws ServletException
    {
	super.init(config);
    }

    public void doGet(HttpServletRequest req, HttpServletResponse res)
                               throws ServletException, IOException
    {
	boolean pass = true;
	Properties p = new Properties();
	res.setContentType("text/plain");
	PrintWriter out = res.getWriter();

	try {
	    String test = harnessProps.getProperty("TEST");
	    System.out.println("doGet: test to execute is: "+test);
	    if(test.equals("sendAndRecvQueueTestFromServletClient")) {
                if(sendAndRecvQueueTestFromServletClient())
	            p.setProperty("TESTRESULT", "pass");
		else
	            p.setProperty("TESTRESULT", "fail");
	    } else if(test.equals("sendAndRecvTopicTestFromServletClient")) {
                if(sendAndRecvTopicTestFromServletClient())
	            p.setProperty("TESTRESULT", "pass");
		else
	            p.setProperty("TESTRESULT", "fail");
	    } else if(test.equals("checkClientIDTestFromServletClient")) {
                if(checkClientIDTestFromServletClient())
	            p.setProperty("TESTRESULT", "pass");
		else
	            p.setProperty("TESTRESULT", "fail");
	    } else {
	        p.setProperty("TESTRESULT", "fail");
	    }
	    doCleanup();
	    p.list(out);
	} catch (Exception e) {
	    TestUtil.logErr("doGet: Exception: " + e);
	    System.out.println("doGet: Exception: " + e);
	    p.setProperty("TESTRESULT", "fail");
	    p.list(out);
	}
	out.close();
    }

    public void doPost(HttpServletRequest req, HttpServletResponse res)
                               throws ServletException, IOException
    {
	harnessProps = new Properties();
	Enumeration enumlist = req.getParameterNames();
	while (enumlist.hasMoreElements()) {
	    String name = (String) enumlist.nextElement();
	    String value = req.getParameter(name);
	    harnessProps.setProperty(name, value);
	}

	try {
	    TestUtil.init(harnessProps);
            // get props
            timeout = Long.parseLong(harnessProps.getProperty("jms_timeout"));
            user = harnessProps.getProperty("user");
            password = harnessProps.getProperty("password");
            mode = harnessProps.getProperty("platform.mode");

            // check props for errors
            if (timeout < 1) {
                throw new Exception("'jms_timeout' (milliseconds) in ts.jte must be > 0");
            }
            if (user == null ) {
                throw new Exception("'user' in ts.jte must not be null ");
            }
            if (password == null ) {
                throw new Exception("'password' in ts.jte must not be null ");
            }
            if (mode == null) {
                throw new Exception("'platform.mode' in ts.jte must not be null");
            }
	    doLookupJMSObjects();
	} catch (Exception e) {
	    System.out.println("doPost: Exception: " + e);
	    e.printStackTrace();
	    throw new ServletException("unable to initialize remote logging");
	}
	doGet(req, res);
	harnessProps = null;
    }

    /*
     * Lookup JMS Connection Factory and Destination Objects
     */
    private void doLookupJMSObjects() throws Exception {
	try {
	    TestUtil.logMsg("Lookup JMS factories defined by @JMSConnectionFactoryDefinitions");
	    TestUtil.logMsg("Lookup JMS destinations defined by @JMSDestinationDefinitions");
	    TSNamingContext namingctx = new TSNamingContext();
	    TestUtil.logMsg("Lookup java:comp/DefaultJMSConnectionFactory");
            dcf = (ConnectionFactory) 
		namingctx.lookup("java:comp/DefaultJMSConnectionFactory");
	    TestUtil.logMsg("Lookup java:global/ServletMyTestConnectionFactory");
            cf = (ConnectionFactory) 
		namingctx.lookup("java:global/ServletMyTestConnectionFactory");
	    TestUtil.logMsg("Lookup java:app/ServletMyTestQueueConnectionFactory");
            qcf = (QueueConnectionFactory) 
		namingctx.lookup("java:app/ServletMyTestQueueConnectionFactory");
	    TestUtil.logMsg("Lookup java:module/ServletMyTestTopicConnectionFactory");
            tcf = (TopicConnectionFactory) 
		namingctx.lookup("java:module/ServletMyTestTopicConnectionFactory");
	    TestUtil.logMsg("Lookup java:comp/env/jms/ServletMyTestDurableTopicConnectionFactory");
            dtcf = (TopicConnectionFactory) 
		namingctx.lookup("java:comp/env/jms/ServletMyTestDurableTopicConnectionFactory");
	    TestUtil.logMsg("Lookup java:global/env/ServletMyTestQueue");
            queue = (Queue) namingctx.lookup("java:global/env/ServletMyTestQueue");
	    TestUtil.logMsg("Lookup java:app/env/ServletMyTestTopic");
            topic = (Topic) namingctx.lookup("java:app/env/ServletMyTestTopic");

	    TestUtil.logMsg("Create JMSContext, JMSProducer's and JMSConsumer's");
            context = cf.createContext(user, password, JMSContext.AUTO_ACKNOWLEDGE);
	    producerQ = context.createProducer();
	    consumerQ = context.createConsumer(queue);
	    producerT = context.createProducer();
	    consumerT = context.createConsumer(topic);
	} catch (Exception e) {
	    TestUtil.logErr("Caught exception: " + e);
	    throw e;
	}
    }


    public boolean sendAndRecvQueueTestFromServletClient() {
	boolean pass = true;
	String message = "Where are you!";
        try {
	    queueTest = true;
            // send and receive TextMessage
            TestUtil.logMsg("Creating TextMessage via JMSContext.createTextMessage(String)");
            TextMessage expTextMessage = context.createTextMessage(message);
            TestUtil.logMsg("Set some values in TextMessage");
            expTextMessage.setStringProperty("COM_SUN_JMS_TESTNAME", "sendAndRecvQueueTestFromServletClient");
	    TestUtil.logMsg("Sending TextMessage via JMSProducer.send(Destination, Message)");
            producerQ.send(queue, expTextMessage);
            TestUtil.logMsg("Receive TextMessage via JMSconsumer.receive(long)");
            TextMessage actTextMessage = (TextMessage) consumerQ.receive(timeout);
            if (actTextMessage == null) {
                TestUtil.logErr("Did not receive TextMessage");
		pass = false;
            } else {
                TestUtil.logMsg("Check the value in TextMessage");
	        if(actTextMessage.getText().equals(expTextMessage.getText())) {
		    TestUtil.logMsg("TextMessage is correct");
	        } else {
		    TestUtil.logErr("TextMessage is incorrect expected " + expTextMessage.getText() 
			+ ", received " + actTextMessage.getText());
		    pass = false;
	        }
	    }
        } catch (Exception e) {
	    TestUtil.logErr("Caught exception: " + e);
            pass = false;
        } 

	return pass;
    }

    public boolean sendAndRecvTopicTestFromServletClient() {
	boolean pass = true;
	String message = "Where are you!";
        try {
	    queueTest = false;
            // send and receive TextMessage
            TestUtil.logMsg("Creating TextMessage via JMSContext.createTextMessage(String)");
            TextMessage expTextMessage = context.createTextMessage(message);
            TestUtil.logMsg("Set some values in TextMessage");
            expTextMessage.setStringProperty("COM_SUN_JMS_TESTNAME", "sendAndRecvTopicTestFromServletClient");
	    TestUtil.logMsg("Sending TextMessage via JMSProducer.send(Destination, Message)");
            producerT.send(topic, expTextMessage);
            TestUtil.logMsg("Receive TextMessage via JMSconsumer.receive(long)");
            TextMessage actTextMessage = (TextMessage) consumerT.receive(timeout);
            if (actTextMessage == null) {
                TestUtil.logErr("Did not receive TextMessage");
		pass = false;
            } else {
                TestUtil.logMsg("Check the value in TextMessage");
	        if(actTextMessage.getText().equals(expTextMessage.getText())) {
		    TestUtil.logMsg("TextMessage is correct");
	        } else {
		    TestUtil.logErr("TextMessage is incorrect expected " + expTextMessage.getText() 
			+ ", received " + actTextMessage.getText());
		    pass = false;
		}
	    }
        } catch (Exception e) {
	    TestUtil.logErr("Caught exception: " + e);
            pass = false;
        } 
	return pass;
    }

    public boolean checkClientIDTestFromServletClient() {
	boolean pass = true;
	JMSContext context = null;
        try {
	    queueTest = false;
	    TestUtil.logMsg("Create JMSContext from durable topic connection factory");
	    TestUtil.logMsg("Check the client id which is configured as MyClientID in the " +
			"JMSConnectionFactoryDefinition annotation");
            context = dtcf.createContext(user, password, JMSContext.AUTO_ACKNOWLEDGE);
	    String clientid = context.getClientID();
	    if(clientid == null) {
		TestUtil.logErr("Client ID value is null (expected MyClientID)");
		pass = false;
	    } else if(clientid.equals("MyClientID")) {
		TestUtil.logMsg("Client ID value is correct (MyClientID)");
	    } else {
		TestUtil.logErr("Client ID value is incorrect (expected MyClientID, got "+clientid+")");
		pass = false;
	    }
        } catch (Exception e) {
	    TestUtil.logErr("Caught exception: " + e);
            pass = false;
        } finally {
	    try {
		if(context != null)
		    context.close();
                doCleanup();
	    } catch(Exception e) {
                TestUtil.logErr("Error in cleanup "+e);
            }
	}
	return pass;
    }

}