private void waitPeerOpened() {
int i = 50;
while (i != 0) {
if (_peer.isOpen()) return;
try {
Thread.sleep(20);
} catch (InterruptedException e) {
}
i++;
}
assertTrue(_peer.isOpen());
}
@Test
public void testTimeout() throws Throwable {
// Log.getLog().setDebugEnabled(true);
TestDiameterHandler serverHandler =
new TestDiameterHandler() {
@Override
public void doHandle(DiameterMessage message) throws Throwable {
Thread.sleep(300);
((DiameterServletRequest) message).createAnswer(Common.DIAMETER_SUCCESS).send();
}
};
_server.setHandler(serverHandler);
_server.start();
TestDiameterHandler clientHandler =
new TestDiameterHandler() {
@Override
public void doHandle(DiameterMessage message) throws Throwable {}
};
_client.setHandler(clientHandler);
_client.setRequestTimeout(200);
_client.start();
waitPeerOpened();
newUdr().send();
assertEquals(1, clientHandler.waitNoAnswer());
serverHandler.assertDone();
Thread.sleep(150);
clientHandler.assertDone(0);
assertTrue(_peer.isOpen());
}
@Test
public void testConnect() throws Exception {
// org.eclipse.jetty.util.log.Log.getLog().setDebugEnabled(true);
_server.start();
_client.start();
waitPeerOpened();
Peer clientPeer = _server.getPeer("client");
assertNotNull(clientPeer);
assertTrue(clientPeer.isOpen());
_peer.stop();
Thread.sleep(100);
assertTrue(_peer.isClosed());
assertTrue(clientPeer.isClosed());
}
