- * Copyright (c) November 24, 2007 - *
- * - */ -public class ResponseCodeTestCase extends TestCase -{ - private static Logger log =3D Logger.getLogger(ResponseCodeTestCase.cla= ss); - = - private static boolean firstTime =3D true; - = - protected static String RESPONSE_KEY =3D "responseKey"; - protected static String RESPONSE_VALUE =3D "responseValue"; - protected static String RETURN_RESPONSE_HEADER =3D "returnResponseHeade= r"; - = - protected String host; - protected int port; - protected String locatorURI; - protected InvokerLocator serverLocator; - protected Connector connector; - protected TestInvocationHandler invocationHandler; - - = - public void setUp() throws Exception - { - if (firstTime) - { - firstTime =3D false; - Logger.getLogger("org.jboss.remoting").setLevel(XLevel.TRACE); - Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO); - String pattern =3D "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n"; - PatternLayout layout =3D new PatternLayout(pattern); - ConsoleAppender consoleAppender =3D new ConsoleAppender(layout); - Logger.getRootLogger().addAppender(consoleAppender); = - } - } - - = - public void tearDown() - { - } - = - = - public void testResponseHeaders() throws Throwable - { - log.info("entering " + getName()); - = - // Start server. - setupServer(); - = - // Create client. - InvokerLocator clientLocator1 =3D new InvokerLocator(locatorURI); - HashMap clientConfig =3D new HashMap(); - clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); - addExtraClientConfig(clientConfig); - Client client =3D new Client(clientLocator1, clientConfig); - client.connect(); - log.info("client is connected"); - = - // Test connections. - HashMap metadata =3D new HashMap(); - assertEquals(RETURN_RESPONSE_HEADER, client.invoke(RETURN_RESPONSE_H= EADER, metadata)); - log.info("connection is good"); - log.info("metadata: " + metadata); - log.info("metadata type: " + metadata.getClass()); - = - // Verify that all entries in the HTTPMetadataConstants.RESPONSE_HEA= DERS value - // also appear in metadata map. - Object o =3D metadata.remove(HTTPMetadataConstants.RESPONSE_HEADERS); - assertTrue(o instanceof Map); - Map responseHeaders =3D (Map) o; - Set responseHeadersEntries =3D responseHeaders.entrySet(); - Set metadataEntries =3D metadata.entrySet(); - = - Iterator it =3D responseHeadersEntries.iterator(); - while (it.hasNext()) - { - Map.Entry entry =3D (Map.Entry) it.next(); - log.info("entry: " + entry); - if (entry.getKey() =3D=3D null) - continue; - assertTrue("metadata does not contain entry " + entry, metadataEn= tries.contains(entry)); - } - = - // Verify response map entry generated in invocation handler - // was added as a response header. - assertTrue(responseHeaders.keySet().contains(RESPONSE_KEY)); - assertTrue(responseHeaders.get(RESPONSE_KEY) instanceof Collection); - List responses =3D (List) responseHeaders.get(RESPONSE_KEY); - assertTrue(responses.size() =3D=3D 1); - assertEquals(RESPONSE_VALUE, responses.get(0)); - = - client.disconnect(); - shutdownServer(); - log.info(getName() + " PASSES"); - } - = - = - protected String getTransport() - { - return "http"; - } - = - = - protected void addExtraClientConfig(Map config) {} - protected void addExtraServerConfig(Map config) {} - = - - protected void setupServer() throws Exception - { - host =3D InetAddress.getLocalHost().getHostAddress(); - port =3D PortUtil.findFreePort(host); - locatorURI =3D getTransport() + "://" + host + ":" + port; = - serverLocator =3D new InvokerLocator(locatorURI); - log.info("Starting remoting server with locator uri of: " + locatorU= RI); - HashMap config =3D new HashMap(); - config.put(InvokerLocator.FORCE_REMOTE, "true"); - addExtraServerConfig(config); - connector =3D new Connector(serverLocator, config); - connector.create(); - invocationHandler =3D new TestInvocationHandler(); - connector.addInvocationHandler("test", invocationHandler); - connector.start(); - } - = - = - = - = - protected void shutdownServer() throws Exception - { - if (connector !=3D null) - connector.stop(); - } - = - = - static class TestInvocationHandler implements ServerInvocationHandler - { - public void addListener(InvokerCallbackHandler callbackHandler) {} - public Object invoke(final InvocationRequest invocation) throws Thro= wable - { - if (RETURN_RESPONSE_HEADER.equals(invocation.getParameter())) - invocation.getReturnPayload().put(RESPONSE_KEY, RESPONSE_VALUE= ); - return invocation.getParameter(); - } - public void removeListener(InvokerCallbackHandler callbackHandler) {} - public void setMBeanServer(MBeanServer server) {} - public void setInvoker(ServerInvoker invoker) {} - } - = - = - static class TestCallbackHandler implements InvokerCallbackHandler - { - public void handleCallback(Callback callback) throws HandleCallbackE= xception - { - log.info("received callback"); - } = - } +/* +* JBoss, Home of Professional Open Source +* Copyright 2005, JBoss Inc., and individual contributors as indicated +* by the @authors tag. See the copyright.txt in the distribution for a +* full listing of individual contributors. +* +* This is free software; you can redistribute it and/or modify it +* under the terms of the GNU Lesser General Public License as +* published by the Free Software Foundation; either version 2.1 of +* the License, or (at your option) any later version. +* +* This software is distributed in the hope that it will be useful, +* but WITHOUT ANY WARRANTY; without even the implied warranty of +* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +* Lesser General Public License for more details. +* +* You should have received a copy of the GNU Lesser General Public +* License along with this software; if not, write to the Free +* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA +* 02110-1301 USA, or see the FSF site: http://www.fsf.org. +*/ +package org.jboss.test.remoting.transport.http.headers; + +import java.net.InetAddress; +import java.util.Collection; +import java.util.HashMap; +import java.util.Iterator; +import java.util.List; +import java.util.Map; +import java.util.Set; + +import javax.management.MBeanServer; + +import junit.framework.TestCase; + +import org.apache.log4j.ConsoleAppender; +import org.apache.log4j.Level; +import org.apache.log4j.Logger; +import org.apache.log4j.PatternLayout; +import org.jboss.logging.XLevel; +import org.jboss.remoting.Client; +import org.jboss.remoting.InvocationRequest; +import org.jboss.remoting.InvokerLocator; +import org.jboss.remoting.ServerInvocationHandler; +import org.jboss.remoting.ServerInvoker; +import org.jboss.remoting.callback.Callback; +import org.jboss.remoting.callback.HandleCallbackException; +import org.jboss.remoting.callback.InvokerCallbackHandler; +import org.jboss.remoting.transport.Connector; +import org.jboss.remoting.transport.PortUtil; +import org.jboss.remoting.transport.http.HTTPMetadataConstants; + +/** + * Unit test for JBREM-728. + * = + * @author Ron Sigal + *+ * Copyright (c) November 24, 2007 + *
+ * + */ +public class ResponseCodeTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(ResponseCodeTestCase.cla= ss); + = + private static boolean firstTime =3D true; + = + protected static String RESPONSE_KEY =3D "responseKey"; + protected static String RESPONSE_VALUE =3D "responseValue"; + protected static String RETURN_RESPONSE_HEADER =3D "returnResponseHeade= r"; + = + protected String host; + protected int port; + protected String locatorURI; + protected InvokerLocator serverLocator; + protected Connector connector; + protected TestInvocationHandler invocationHandler; + + = + public void setUp() throws Exception + { + if (firstTime) + { + firstTime =3D false; + Logger.getLogger("org.jboss.remoting").setLevel(XLevel.TRACE); + Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO); + String pattern =3D "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n"; + PatternLayout layout =3D new PatternLayout(pattern); + ConsoleAppender consoleAppender =3D new ConsoleAppender(layout); + Logger.getRootLogger().addAppender(consoleAppender); = + } + } + + = + public void tearDown() + { + } + = + = + public void testResponseHeaders() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator1 =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator1, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connections. + HashMap metadata =3D new HashMap(); + assertEquals(RETURN_RESPONSE_HEADER, client.invoke(RETURN_RESPONSE_H= EADER, metadata)); + log.info("connection is good"); + log.info("metadata: " + metadata); + log.info("metadata type: " + metadata.getClass()); + = + // Verify that all entries in the HTTPMetadataConstants.RESPONSE_HEA= DERS value + // also appear in metadata map. + Object o =3D metadata.remove(HTTPMetadataConstants.RESPONSE_HEADERS); + assertTrue(o instanceof Map); + Map responseHeaders =3D (Map) o; + Set responseHeadersEntries =3D responseHeaders.entrySet(); + Set metadataEntries =3D metadata.entrySet(); + = + Iterator it =3D responseHeadersEntries.iterator(); + while (it.hasNext()) + { + Map.Entry entry =3D (Map.Entry) it.next(); + log.info("entry: " + entry); + if (entry.getKey() =3D=3D null) + continue; + assertTrue("metadata does not contain entry " + entry, metadataEn= tries.contains(entry)); + } + = + // Verify response map entry generated in invocation handler + // was added as a response header. + assertTrue(responseHeaders.keySet().contains(RESPONSE_KEY)); + assertTrue(responseHeaders.get(RESPONSE_KEY) instanceof Collection); + List responses =3D (List) responseHeaders.get(RESPONSE_KEY); + assertTrue(responses.size() =3D=3D 1); + assertEquals(RESPONSE_VALUE, responses.get(0)); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + protected String getTransport() + { + return "http"; + } + = + = + protected void addExtraClientConfig(Map config) {} + protected void addExtraServerConfig(Map config) {} + = + + protected void setupServer() throws Exception + { + host =3D InetAddress.getLocalHost().getHostAddress(); + port =3D PortUtil.findFreePort(host); + locatorURI =3D getTransport() + "://" + host + ":" + port; = + serverLocator =3D new InvokerLocator(locatorURI); + log.info("Starting remoting server with locator uri of: " + locatorU= RI); + HashMap config =3D new HashMap(); + config.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraServerConfig(config); + connector =3D new Connector(serverLocator, config); + connector.create(); + invocationHandler =3D new TestInvocationHandler(); + connector.addInvocationHandler("test", invocationHandler); + connector.start(); + } + = + = + = + = + protected void shutdownServer() throws Exception + { + if (connector !=3D null) + connector.stop(); + } + = + = + static class TestInvocationHandler implements ServerInvocationHandler + { + public void addListener(InvokerCallbackHandler callbackHandler) {} + public Object invoke(final InvocationRequest invocation) throws Thro= wable + { + if (RETURN_RESPONSE_HEADER.equals(invocation.getParameter())) + invocation.getReturnPayload().put(RESPONSE_KEY, RESPONSE_VALUE= ); + return invocation.getParameter(); + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + = + = + static class TestCallbackHandler implements InvokerCallbackHandler + { + public void handleCallback(Callback callback) throws HandleCallbackE= xception + { + log.info("received callback"); + } = + } } \ No newline at end of file Property changes on: remoting2/branches/2.2/src/tests/org/jboss/test/remoti= ng/transport/http/headers/ResponseCodeTestCase.java ___________________________________________________________________ Name: svn:eol-style + native --===============2292733614062720626==--