Ensure all resources shut down.

This commit is contained in:
Timothy Bish 2014-08-18 17:58:58 -04:00
parent 3aec42db4f
commit 9b017fdb39
1 changed files with 14 additions and 5 deletions

View File

@ -17,6 +17,8 @@
package org.apache.activemq.transport.ws;
import static org.junit.Assert.assertTrue;
import java.io.File;
import java.io.IOException;
import java.net.Socket;
@ -47,8 +49,6 @@ import org.openqa.selenium.firefox.FirefoxProfile;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import static org.junit.Assert.assertTrue;
public class WSTransportTest {
private static final Logger LOG = LoggerFactory.getLogger(WSTransportTest.class);
@ -137,13 +137,21 @@ public class WSTransportTest {
} catch(Exception e) {
// Some tests explicitly disconnect from stomp so can ignore
} finally {
try {
stopBroker();
} catch (Exception e) {
LOG.warn("Error on Broker stop.");
}
if (driver != null) {
try {
driver.quit();
} catch (Exception e) {}
driver = null;
}
if (server != null) {
try {
server.stop();
} catch (Exception e) {}
}
}
}
@ -250,6 +258,7 @@ public class WSTransportTest {
assertTrue("Should have disconnected", Wait.waitFor(new Wait.Condition() {
@Override
public boolean isSatisified() throws Exception {
return webStatus.getText().equals("Disconnected");
}