2
* JBoss, Home of Professional Open Source
3
* Copyright 2005, JBoss Inc., and individual contributors as indicated
4
* by the @authors tag. See the copyright.txt in the distribution for a
5
* full listing of individual contributors.
7
* This is free software; you can redistribute it and/or modify it
8
* under the terms of the GNU Lesser General Public License as
9
* published by the Free Software Foundation; either version 2.1 of
10
* the License, or (at your option) any later version.
12
* This software is distributed in the hope that it will be useful,
13
* but WITHOUT ANY WARRANTY; without even the implied warranty of
14
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15
* Lesser General Public License for more details.
17
* You should have received a copy of the GNU Lesser General Public
18
* License along with this software; if not, write to the Free
19
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
22
package org.jboss.test.remoting.transport.socket.localaddress;
24
import java.net.InetAddress;
25
import java.util.HashMap;
28
import javax.management.MBeanServer;
30
import junit.framework.TestCase;
32
import org.apache.log4j.ConsoleAppender;
33
import org.apache.log4j.Level;
34
import org.apache.log4j.Logger;
35
import org.apache.log4j.PatternLayout;
36
import org.jboss.remoting.Client;
37
import org.jboss.remoting.InvocationRequest;
38
import org.jboss.remoting.InvocationResponse;
39
import org.jboss.remoting.InvokerLocator;
40
import org.jboss.remoting.ServerInvocationHandler;
41
import org.jboss.remoting.ServerInvoker;
42
import org.jboss.remoting.callback.Callback;
43
import org.jboss.remoting.callback.HandleCallbackException;
44
import org.jboss.remoting.callback.InvokerCallbackHandler;
45
import org.jboss.remoting.transport.Connector;
46
import org.jboss.remoting.transport.PortUtil;
50
* Unit test for JBREM-781.
52
* @author <a href="ron.sigal@jboss.com">Ron Sigal</a>
53
* @version $Revision: 2980 $
55
* Copyright Jul 30, 2007
58
public class LocalAddressTestCase extends TestCase
60
private static Logger log = Logger.getLogger(LocalAddressTestCase.class);
62
private static boolean firstTime = true;
64
// remoting server connector
65
private Connector connector;
66
private InvokerLocator serverLocator;
67
private TestInvocationHandler invocationHandler;
71
* Sets up target remoting server.
73
public void setUp() throws Exception
78
Logger.getLogger("org.jboss.remoting").setLevel(Level.INFO);
79
Logger.getLogger("org.jboss.test.remoting").setLevel(Level.INFO);
80
String pattern = "[%d{ABSOLUTE}] [%t] %5p (%F:%L) - %m%n";
81
PatternLayout layout = new PatternLayout(pattern);
82
ConsoleAppender consoleAppender = new ConsoleAppender(layout);
83
Logger.getRootLogger().addAppender(consoleAppender);
88
public void tearDown()
94
* Verifies that the address returned by a "$GET_CLIENT_LOCAL_ADDRESS$"
97
public void testLocalAddressRequest() throws Throwable
99
log.info("entering " + getName());
102
String host = InetAddress.getLocalHost().getHostAddress();
103
int port = PortUtil.findFreePort(host);
104
String locatorURI = getTransport() + "://" + host + ":" + port;
105
serverLocator = new InvokerLocator(locatorURI);
106
log.info("Starting remoting server with locator uri of: " + locatorURI);
107
HashMap config = new HashMap();
108
config.put(InvokerLocator.FORCE_REMOTE, "true");
109
addExtraServerConfig(config);
110
connector = new Connector(serverLocator, config);
112
invocationHandler = new TestInvocationHandler();
113
connector.addInvocationHandler("sample", invocationHandler);
117
InvokerLocator clientLocator1 = new InvokerLocator(locatorURI + "/?timeout=1000");
118
HashMap clientConfig = new HashMap();
119
clientConfig.put(InvokerLocator.FORCE_REMOTE, "true");
120
addExtraClientConfig(clientConfig);
121
Client client = new Client(clientLocator1, clientConfig);
123
log.info("client is connected");
126
assertEquals("abc", client.invoke("abc"));
127
log.info("connection is good");
129
// Get address of server.
130
// Object o = client.invoke("$GET_CLIENT_LOCAL_ADDRESS$");
131
// assertTrue(o instanceof InvocationResponse);
132
// InvocationResponse response = (InvocationResponse) o;
133
// assertTrue(response.getResult() instanceof InetAddress);
134
// InetAddress newAddress = (InetAddress) response.getResult();
136
// This test is updated to conform to new treatment of this facility.
138
InetAddress newAddress = (InetAddress) client.invoke("$GET_CLIENT_LOCAL_ADDRESS$");
139
String newLocatorURI = getTransport() + "://" + newAddress.getHostAddress() + ":" + port;
140
log.info("new locator: " + newLocatorURI);
142
// Try to connect to locator constructed from returned address.
143
Client newClient = new Client(new InvokerLocator(newLocatorURI));
145
assertEquals("xyz", newClient.invoke("xyz"));
148
newClient.disconnect();
150
log.info(getName() + " PASSES");
154
protected String getTransport()
160
protected void addExtraClientConfig(Map config) {}
161
protected void addExtraServerConfig(Map config) {}
164
static class TestInvocationHandler implements ServerInvocationHandler
166
public void addListener(InvokerCallbackHandler callbackHandler) {}
167
public Object invoke(final InvocationRequest invocation) throws Throwable
169
return invocation.getParameter();
171
public void removeListener(InvokerCallbackHandler callbackHandler) {}
172
public void setMBeanServer(MBeanServer server) {}
173
public void setInvoker(ServerInvoker invoker) {}
177
static class TestCallbackHandler implements InvokerCallbackHandler
179
public void handleCallback(Callback callback) throws HandleCallbackException
181
log.info("received callback");
b'\\ No newline at end of file'