+ * Copyright Aug 17, 2009 + *
+ */ +public class HttpContentTypeTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(HttpContentTypeTestCase.= class); + = + protected static boolean firstTime =3D true; + = + 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(Level.INFO); + 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); = + } + = + TestMarshaller.marshallers.clear(); + TestUnMarshaller.unmarshallers.clear(); + MarshalFactory.addMarshaller("test", new TestMarshaller(), new TestU= nMarshaller()); + } + + = + public void tearDown() + { + } + = + = + public void testOrdinaryInvocation() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connections. + assertEquals("abc", client.invoke("abc")); + log.info("connection is good"); + = + // Check remoting content type handling. + validateOrdinaryInvocation(); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + public void testRawStringMessage() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + clientConfig.put(InvokerLocator.FORCE_REMOTE, "true"); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, clientConfig); + client.connect(); + log.info("client is connected"); + = + // Test connections. + Map metadata =3D new HashMap(); + metadata.put(Client.RAW, "true"); + assertEquals("abc", client.invoke("abc", metadata)); + log.info("connection is good"); + = + // Check remoting content type handling. + validateRawStringMessage(); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + protected void validateOrdinaryInvocation() + { + assertEquals(6, TestMarshaller.marshallers.size()); = + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_NON_STRING,= ((TestMarshaller)TestMarshaller.marshallers.get(3)).type); + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_NON_STRING,= ((TestMarshaller)TestMarshaller.marshallers.get(5)).type); + assertEquals(4, TestUnMarshaller.unmarshallers.size()); + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_NON_STRING,= ((TestUnMarshaller)TestUnMarshaller.unmarshallers.get(1)).type); + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_NON_STRING,= ((TestUnMarshaller)TestUnMarshaller.unmarshallers.get(3)).type); + } + = + = + protected void validateRawStringMessage() + { + assertEquals(6, TestMarshaller.marshallers.size()); = + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_STRING, ((T= estMarshaller)TestMarshaller.marshallers.get(3)).type); + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_STRING, ((T= estMarshaller)TestMarshaller.marshallers.get(5)).type); + assertEquals(4, TestUnMarshaller.unmarshallers.size()); + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_STRING, ((T= estUnMarshaller)TestUnMarshaller.unmarshallers.get(1)).type); + assertEquals(HTTPMetadataConstants.REMOTING_CONTENT_TYPE_STRING, ((T= estUnMarshaller)TestUnMarshaller.unmarshallers.get(3)).type); + } + = + = + 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; + locatorURI +=3D "/?datatype=3Dtest"; + String metadata =3D System.getProperty("remoting.metadata"); + if (metadata !=3D null) + { + locatorURI +=3D "&" + metadata; + } + 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 + { + Map responseMap =3D invocation.getReturnPayload(); + responseMap.put(HTTPMetadataConstants.CONTENTTYPE, "text/html"); + return invocation.getParameter(); + } + public void removeListener(InvokerCallbackHandler callbackHandler) {} + public void setMBeanServer(MBeanServer server) {} + public void setInvoker(ServerInvoker invoker) {} + } + = + + public static class TestMarshaller extends HTTPMarshaller + { + public static ArrayList marshallers =3D new ArrayList(); + private static final long serialVersionUID =3D -7528137229006015488L; + public String type; + = + public void write(Object dataObject, OutputStream output, int versio= n) throws IOException + { + log.info(this + " writing " + dataObject); + type =3D (dataObject instanceof String) ? HTTPMetadataConstants.R= EMOTING_CONTENT_TYPE_STRING : HTTPMetadataConstants.REMOTING_CONTENT_TYPE_N= ON_STRING; + super.write(dataObject, output, version); + } + = + public Marshaller cloneMarshaller() throws CloneNotSupportedException + { + TestMarshaller marshaller =3D new TestMarshaller(); + marshallers.add(marshaller); + log.info("returning " + marshaller); + return marshaller; + } + } + = + public static class TestUnMarshaller extends HTTPUnMarshaller + { + public static ArrayList unmarshallers =3D new ArrayList(); + private static final long serialVersionUID =3D -6422222480047910351L; + public String type; + = + public Object read(InputStream inputStream, Map metadata, int versio= n) throws IOException, ClassNotFoundException + { + Object o =3D metadata.get(HTTPMetadataConstants.REMOTING_CONTENT_= TYPE); + if (o instanceof List) + { + type =3D (String) ((List) o).get(0); + } + else if (o instanceof String) + { + type =3D (String) o; + } + else = + { + log.warn(this + " unrecognized remotingContentType: " + o); + } + = + o =3D super.read(inputStream, metadata, version); + log.info(this + " read " + o); + return o; + } + = + public UnMarshaller cloneUnMarshaller() throws CloneNotSupportedExce= ption + { + TestUnMarshaller unmarshaller =3D new TestUnMarshaller(); + unmarshallers.add(unmarshaller); + unmarshaller.setClassLoader(this.customClassLoader); + log.info("returning " + unmarshaller); + return unmarshaller; + } + } +} \ No newline at end of file Property changes on: remoting2/branches/2.x/src/tests/org/jboss/test/remoti= ng/transport/http/marshal/HttpContentTypeTestCase.java ___________________________________________________________________ Name: svn:mergeinfo + = --===============3272294884911524277==--