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()); }