+ * Copyright Apr 29, 2011 + *
+ */ +public class LocalStreamTestCase extends TestCase +{ + private static Logger log =3D Logger.getLogger(LocalStreamTestCase.clas= s); + = + private static boolean firstTime =3D true; + = + protected String host; + protected int port; + protected String locatorURI; + protected InvokerLocator serverLocator; + protected Connector connector; + protected File testFile =3D null; + protected FileInputStream fileInput =3D null; + protected TestStreamInvocationHandler 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); = + } + } + + = + public void tearDown() + { + } + = + = + public void testLocalStream() throws Throwable + { + log.info("entering " + getName()); + = + // Start server. + setupServer(); + = + // Create client. + InvokerLocator clientLocator =3D new InvokerLocator(locatorURI); + HashMap clientConfig =3D new HashMap(); + addExtraClientConfig(clientConfig); + Client client =3D new Client(clientLocator, "test_stream", clientCon= fig); + client.connect(); + log.info("client is connected"); + = + // Test sending stream. + URL fileURL =3D this.getClass().getResource("../test.txt"); + if(fileURL =3D=3D null) + { + throw new Exception("Can not find file test.txt"); + } + testFile =3D new File(fileURL.getFile()); + fileInput =3D new FileInputStream(testFile); + + String param =3D "foobar"; + long fileLength =3D testFile.length(); + log.info("File size =3D " + fileLength); + Object ret =3D client.invoke(fileInput, param); + + Map responseMap =3D (Map)ret; + String subSys =3D (String)responseMap.get("subsystem"); + String clientId =3D (String)responseMap.get("clientid"); + String paramVal =3D (String)responseMap.get("paramval"); + + assertEquals("test_stream".toUpperCase(), subSys); + assertEquals(client.getSessionId(), clientId); + assertEquals("foobar", paramVal); + + Object response =3D client.invoke("get_size"); + int returnedFileLength =3D ((Integer) response).intValue(); + log.info("Invocation response: " + response); + log.info("returnedFileLength: " + returnedFileLength); + assertEquals(fileLength, returnedFileLength); + = + // Test for LocalServerInvoker. + ServerInvoker[] serverInvokers =3D InvokerRegistry.getServerInvokers= (); + assertEquals(2, serverInvokers.length); + assertTrue(serverInvokers[0] instanceof SocketServerInvoker || serve= rInvokers[1] instanceof SocketServerInvoker); + assertTrue(serverInvokers[0] instanceof LocalServerInvoker || server= Invokers[1] instanceof LocalServerInvoker); + = + client.disconnect(); + shutdownServer(); + log.info(getName() + " PASSES"); + } + = + = + protected String getTransport() + { + return "socket"; + } + + = + 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; + 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 TestStreamInvocationHandler(); + connector.addInvocationHandler("test_stream", invocationHandler); + connector.start(); + } + = + = + protected void shutdownServer() throws Exception + { + if (connector !=3D null) + connector.stop(); + } + = + = + public static class TestStreamInvocationHandler implements StreamInvoca= tionHandler + { + private InputStream stream =3D null; + + private int streamSize =3D 0; + + public Object invoke(InvocationRequest invocation) throws Throwable + { + // Print out the invocation request + log.info("Invocation request is: " + invocation.getParameter()); + + // Just going to return static string as this is just simple exam= ple code. + return new Integer(streamSize); + } + + public void addListener(InvokerCallbackHandler callbackHandler) + { + } + public void removeListener(InvokerCallbackHandler callbackHandler) + { + } + public void setMBeanServer(MBeanServer server) + { + } + public void setInvoker(ServerInvoker invoker) + { + } + + public Object handleStream(InputStream stream, InvocationRequest par= am) + { + this.stream =3D stream; + + try + { + ByteArrayOutputStream out =3D new ByteArrayOutputStream(); + byte buf[] =3D new byte[4096]; + while(true) + { + int c =3D this.stream.read(buf); + if(c < 0) + { + break; + } + out.write(buf, 0, c); + } + byte[] bytes =3D out.toByteArray(); + streamSize =3D bytes.length; + log.info("Read stream. Length is: " + streamSize); + } + catch(IOException e) + { + e.printStackTrace(); + } + // Not closing the InputStream to keep the StreamServer alive. +// finally +// { +// try +// { +//// log.info("closing stream"); +//// stream.close(); +// } +// catch(IOException e) +// { +// log.error("error closing stream", e); +// } +// } + // build return map + Map retMap =3D new HashMap(); + retMap.put("subsystem", param.getSubsystem()); + retMap.put("clientid", param.getSessionId()); + retMap.put("paramval", param.getParameter()); + return retMap; + } + } +} \ No newline at end of file --===============7467573643182886370==--