Apache
/
activemq
mirror of https://github.com/apache/activemq.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add tests for Websockets using selenium 

git-svn-id: https://svn.apache.org/repos/asf/activemq/trunk@1342813 13f79535-47bb-0310-9956-ffa450edef68
This commit is contained in:
Timothy A. Bish 2012-05-25 21:07:52 +00:00
parent a1b235d275
commit afcbc4e5a3
3 changed files with 250 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
activemq-optional/pom.xml
View File

@ -202,6 +202,24 @@
<artifactId>gentlyweb-utils</artifactId>
</dependency>
<dependency>
<groupId>org.seleniumhq.selenium</groupId>
<artifactId>selenium-java</artifactId>
<version>2.21.0</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.seleniumhq.selenium</groupId>
<artifactId>selenium-chrome-driver</artifactId>
<version>2.21.0</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.seleniumhq.selenium</groupId>
<artifactId>selenium-firefox-driver</artifactId>
<version>2.21.0</version>
<scope>test</scope>
</dependency>
</dependencies>

187
activemq-optional/src/test/java/org/apache/activemq/transport/ws/WSTransportTest.java
View File

@ -17,29 +17,190 @@
package org.apache.activemq.transport.ws;
import static org.junit.Assert.assertTrue;
import java.io.File;
import java.io.IOException;
import java.net.Socket;
import java.net.URI;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.UnknownHostException;
import org.apache.activemq.broker.BrokerFactory;
import org.apache.activemq.broker.BrokerService;
import org.apache.activemq.transport.stomp.StompConnection;
import org.apache.activemq.util.Wait;
import org.junit.After;
import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.firefox.FirefoxDriver;
import org.openqa.selenium.firefox.FirefoxProfile;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class WSTransportTest {
protected String getBindLocation() {
return "ws://localhost:61614";
private static final Logger LOG = LoggerFactory.getLogger(WSTransportTest.class);
private static final int MESSAGE_COUNT = 1000;
private BrokerService broker;
private WebDriver driver;
private File profileDir;
private String stompUri;
private String wsUri;
private StompConnection stompConnection = new StompConnection();
protected BrokerService createBroker(boolean deleteMessages) throws Exception {
BrokerService broker = BrokerFactory.createBroker(
new URI("broker:()/localhost?persistent=false&useJmx=false"));
stompUri = broker.addConnector("stomp://localhost:0").getPublishableConnectString();
wsUri = broker.addConnector("ws://127.0.0.1:61623").getPublishableConnectString();
broker.setDeleteAllMessagesOnStartup(deleteMessages);
broker.start();
broker.waitUntilStarted();
return broker;
}
protected void stopBroker() throws Exception {
if (broker != null) {
broker.stop();
broker.waitUntilStopped();
broker = null;
}
}
@Before
public void setUp() throws Exception {
profileDir = new File("activemq-data/profiles");
broker = createBroker(true);
stompConnect();
}
@After
public void tearDown() throws Exception {
try {
stompDisconnect();
} catch(Exception e) {
// Some tests explicitly disconnect from stomp so can ignore
} finally {
stopBroker();
if (driver != null) {
driver.quit();
driver = null;
}
}
}
@Test
public void testBrokerStart() throws Exception {
BrokerService broker = BrokerFactory.createBroker(new URI("broker:()/localhost?persistent=false&useJmx=false"));
broker.addConnector(getBindLocation());
broker.start();
broker.waitUntilStarted();
Thread.sleep(2000);
//System.in.read();
broker.stop();
broker.waitUntilStopped();
assertTrue(broker.isStarted());
}
@Ignore
@Test
public void testFireFoxWebSockets() throws Exception {
driver = createFireFoxWebDriver();
doTestWebSockets(driver);
}
@Ignore
@Test
public void testChromeWebSockets() throws Exception {
driver = createChromeWebDriver();
doTestWebSockets(driver);
}
protected WebDriver createChromeWebDriver() throws Exception {
File profile = new File(profileDir, "chrome");
profile.mkdirs();
ChromeOptions options = new ChromeOptions();
options.addArguments("--enable-udd-profiles",
"--user-data-dir=" + profile,
"--allow-file-access-from-files");
return new ChromeDriver(options);
}
protected WebDriver createFireFoxWebDriver() throws Exception {
File profile = new File(profileDir, "firefox");
profile.mkdirs();
return new FirefoxDriver(new FirefoxProfile(profile));
}
private void stompConnect() throws IOException, URISyntaxException, UnknownHostException {
URI connectUri = new URI(stompUri);
stompConnection.open(createSocket(connectUri));
}
private Socket createSocket(URI connectUri) throws IOException {
return new Socket("127.0.0.1", connectUri.getPort());
}
private void stompDisconnect() throws IOException {
if (stompConnection != null) {
stompConnection.close();
stompConnection = null;
}
}
public void doTestWebSockets(WebDriver driver) throws Exception {
URL url = getClass().getResource("websocket.html");
LOG.info("working dir = ");
LOG.info("page url: " + url);
driver.get(url + "#" + wsUri);
final WebElement webStatus = driver.findElement(By.id("status"));
final WebElement webReceived = driver.findElement(By.id("received"));
while ("Loading" == webStatus.getText()) {
Thread.sleep(100);
}
// Skip test if browser does not support websockets..
if (webStatus.getText() != "No WebSockets") {
assertTrue("Should have connected", Wait.waitFor(new Wait.Condition() {
@Override
public boolean isSatisified() throws Exception {
return webStatus.getText().equals("Connected");
}
}));
stompConnection.connect("system", "manager");
stompConnection.send("/queue/websocket", "Hello");
assertTrue("Should have received message by now.", Wait.waitFor(new Wait.Condition() {
@Override
public boolean isSatisified() throws Exception {
return webReceived.getText().equals("Hello");
}
}));
for (int i = 1; i <= MESSAGE_COUNT; ++i) {
stompConnection.send("/queue/websocket", "messages #" + i);
}
assertTrue("Should have received messages by now.", Wait.waitFor(new Wait.Condition() {
@Override
public boolean isSatisified() throws Exception {
return webReceived.getText().equals("messages #" + MESSAGE_COUNT);
}
}));
}
}
}

58
activemq-optional/src/test/resources/org/apache/activemq/transport/ws/websocket.html Normal file
View File

@ -0,0 +1,58 @@
<!DOCTYPE html>
<!--
Licensed to the Apache Software Foundation (ASF) under one or more
contributor license agreements. See the NOTICE file distributed with
this work for additional information regarding copyright ownership.
The ASF licenses this file to You under the Apache License, Version 2.0
(the "License"); you may not use this file except in compliance with
the License. You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
-->
<html lang="en">
<head>
<meta charset="utf-8">
<title>Chat Example Using STOMP Over WebSockets</title>
<!--[if lt IE 9]>
<script src="http://html5shim.googlecode.com/svn/trunk/html5.js"></script>
<![endif]-->
<script src='http://ajax.googleapis.com/ajax/libs/jquery/1.4/jquery.min.js'></script>
<script src="../../../../../../../../activemq-web-demo/src/main/webapp/websocket/stomp.js"></script>
<script>//<![CDATA[
$(document).ready(function() {
if(window.WebSocket) {
var url = "ws://localhost:61623";
if(window.location.hash) {
url = window.location.hash.substring(1);
}
var login = "admin";
var passcode = "password";
destination = "/queue/websocket";
$("#status").html("Connecting");
client = Stomp.client(url);
// the client is notified when it is connected to the server.
client.connect(login, passcode, function(frame) {
$("#status").html("Connected");
client.subscribe(destination, function(message) {
$("#received").html(message.body);
});
});
} else {
$("#status").html("No WebSockets");
}
});
//]]></script>
</head>
<body>
<div id="status">Loading</div>
<div id="received"></div>
</body>
</html>