Created
August 2, 2011 09:37
-
-
Save justinwyer/1119897 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
/* | |
* JBoss, Home of Professional Open Source | |
* Copyright 2011 Red Hat Inc. and/or its affiliates and other contributors | |
* as indicated by the @authors tag. All rights reserved. | |
* See the copyright.txt in the distribution for a | |
* full listing of individual contributors. | |
* | |
* This copyrighted material is made available to anyone wishing to use, | |
* modify, copy, or redistribute it subject to the terms and conditions | |
* of the GNU Lesser General Public License, v. 2.1. | |
* This program is distributed in the hope that it will be useful, but WITHOUT A | |
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A | |
* PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. | |
* You should have received a copy of the GNU Lesser General Public License, | |
* v.2.1 along with this distribution; if not, write to the Free Software | |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, | |
* MA 02110-1301, USA. | |
*/ | |
package org.switchyard.test.helpdeskdemo; | |
import java.io.IOException; | |
import java.io.InputStream; | |
import java.io.UnsupportedEncodingException; | |
import java.net.Socket; | |
import java.net.SocketException; | |
import java.util.List; | |
import java.util.Locale; | |
import javax.persistence.EntityManagerFactory; | |
import javax.persistence.Persistence; | |
import org.apache.commons.httpclient.HttpClient; | |
import org.apache.commons.httpclient.HttpMethod; | |
import org.apache.commons.httpclient.methods.InputStreamRequestEntity; | |
import org.apache.commons.httpclient.methods.PostMethod; | |
import org.apache.commons.httpclient.methods.StringRequestEntity; | |
import org.drools.agent.impl.DoNothingSystemEventListener; | |
import org.jboss.arquillian.api.Deployment; | |
import org.jboss.arquillian.junit.Arquillian; | |
import org.jboss.shrinkwrap.api.spec.JavaArchive; | |
import org.jbpm.task.Status; | |
import org.jbpm.task.User; | |
import org.jbpm.task.query.TaskSummary; | |
import org.jbpm.task.service.TaskClient; | |
import org.jbpm.task.service.TaskServer; | |
import org.jbpm.task.service.TaskService; | |
import org.jbpm.task.service.TaskServiceSession; | |
import org.jbpm.task.service.mina.MinaTaskClientConnector; | |
import org.jbpm.task.service.mina.MinaTaskClientHandler; | |
import org.jbpm.task.service.mina.MinaTaskServer; | |
import org.jbpm.task.service.responsehandlers.BlockingTaskOperationResponseHandler; | |
import org.jbpm.task.service.responsehandlers.BlockingTaskSummaryResponseHandler; | |
import org.junit.Assert; | |
import org.junit.Test; | |
import org.junit.runner.RunWith; | |
import org.switchyard.test.ArquillianUtil; | |
/** | |
* | |
* @author <a href="mailto:tom.fennelly@gmail.com">tom.fennelly@gmail.com</a> | |
*/ | |
@RunWith(Arquillian.class) | |
public class HelpDeskDemoQuickstartTest { | |
@Deployment(testable = false) | |
public static JavaArchive createDeployment() { | |
return ArquillianUtil.createDemoDeployment("switchyard-quickstart-demo-helpdesk"); | |
} | |
@Test | |
public void test() throws Exception { | |
HumanTaskServerHelper ht = new HumanTaskServerHelper(); | |
if (ht.startTaskServer("Developer", "User")) { | |
try { | |
HttpHelper http = new HttpHelper(); | |
http.postStringAndTestXML("http://localhost:18001/HelpDeskService", SOAP_REQUEST, EXPECTED_SOAP_RESPONSE); | |
boolean keepWorking = true; | |
while (keepWorking) { | |
keepWorking = ht.completeTasksForUsers("Developer", "User"); | |
} | |
} finally { | |
ht.stopTaskServer(); | |
} | |
} | |
} | |
private static final String SOAP_REQUEST = "<soap:Envelope xmlns:soap=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns:helpdesk=\"urn:switchyard-quickstart-demo:helpdesk:1.0\">\n" | |
+ " <soap:Body>\n" | |
+ " <helpdesk:openTicket>\n" | |
+ " <ticket>\n" | |
+ " <id>TCKT-17761852</id>\n" | |
+ " </ticket>\n" | |
+ " </helpdesk:openTicket>\n" | |
+ " </soap:Body>\n" | |
+ "</soap:Envelope>\n"; | |
private static final String EXPECTED_SOAP_RESPONSE = "<SOAP-ENV:Envelope xmlns:SOAP-ENV=\"http://schemas.xmlsoap.org/soap/envelope/\" xmlns:bpm=\"urn:switchyard-component-bpm:process:1.0\" xmlns:helpdesk=\"urn:switchyard-quickstart-demo:helpdesk:1.0\">\n" | |
+ " <SOAP-ENV:Header>\n" | |
+ " <bpm:processInstanceId>1</bpm:processInstanceId>\n" | |
+ " </SOAP-ENV:Header>\n" | |
+ " <SOAP-ENV:Body>\n" | |
+ " <helpdesk:openTicketResponse>\n" | |
+ " <ticketAck>\n" | |
+ " <id>TCKT-17761852</id>\n" | |
+ " <received>true</received>\n" | |
+ " </ticketAck>\n" | |
+ " </helpdesk:openTicketResponse>\n" | |
+ " </SOAP-ENV:Body>\n" | |
+ "</SOAP-ENV:Envelope>"; | |
class HttpHelper { | |
private HttpClient _httpClient; | |
private String _contentType = "text/xml"; | |
/** | |
* Set the content type. | |
* <p/> | |
* Default content type is "text/xml". | |
* | |
* @param contentType The content type. | |
* @return This HTTPMixIn instance. | |
*/ | |
public HttpHelper() { | |
initialize(); | |
} | |
public HttpHelper setContentType(String contentType) { | |
this._contentType = contentType; | |
return this; | |
} | |
public void initialize() { | |
_httpClient = new HttpClient(); | |
} | |
/** | |
* POST the specified request payload to the specified HTTP endpoint. | |
* @param endpointURL The HTTP endpoint URL. | |
* @param request The request payload. | |
* @return The HTTP response payload. | |
*/ | |
public String postString(String endpointURL, String request) { | |
PostMethod postMethod = new PostMethod(endpointURL); | |
try { | |
postMethod.setRequestEntity(new StringRequestEntity(request, _contentType, "UTF-8")); | |
} catch (UnsupportedEncodingException e) { | |
e.printStackTrace(); | |
} | |
try { | |
return execute(postMethod); | |
} finally { | |
postMethod.releaseConnection(); | |
} | |
} | |
/** | |
* POST the specified String request to the specified HTTP endpoint and perform an XML compare | |
* between the HTTP response and the specified expected response String. | |
* @param endpointURL The HTTP endpoint URL. | |
* @param request The classpath resource to be posted to the endpoint. | |
* @param expectedResponse The String to use to perform the XML test on the response. | |
* @return The HTTP response payload. | |
*/ | |
public String postStringAndTestXML(String endpointURL, String request, String expectedResponse) { | |
String response = postString(endpointURL, request); | |
System.err.println(response); | |
return response; | |
} | |
/** | |
* POST the specified classpath resource to the specified HTTP endpoint. | |
* @param endpointURL The HTTP endpoint URL. | |
* @param requestResource The classpath resource to be posted to the endpoint. | |
* @return The HTTP response payload. | |
*/ | |
public String postResource(String endpointURL, String requestResource) { | |
PostMethod postMethod = new PostMethod(endpointURL); | |
InputStream requestStream = this.getClass().getResourceAsStream(requestResource); | |
try { | |
postMethod.setRequestEntity(new InputStreamRequestEntity(requestStream, _contentType + "; charset=utf-8")); | |
return execute(postMethod); | |
} finally { | |
try { | |
requestStream.close(); | |
} catch (IOException e) { | |
Assert.fail("Unexpected exception closing HTTP request resource stream."); | |
} finally { | |
postMethod.releaseConnection(); | |
} | |
} | |
} | |
/** | |
* POST the specified classpath resource to the specified HTTP endpoint and perform an XML compare | |
* between the HTTP response and the specified expected classpath response resource. | |
* @param endpointURL The HTTP endpoint URL. | |
* @param requestResource The classpath resource to be posted to the endpoint. | |
* @param expectedResponseResource The classpath resource to use to perform the XML test on the response. | |
* @return The HTTP response payload. | |
*/ | |
public String postResourceAndTestXML(String endpointURL, String requestResource, String expectedResponseResource) { | |
String response = postResource(endpointURL, requestResource); | |
System.err.println(response); | |
return response; | |
} | |
/** | |
* Execute the supplied HTTP Method. | |
* <p/> | |
* Does not release the {@link org.apache.commons.httpclient.HttpMethod#releaseConnection() HttpMethod connection}. | |
* | |
* @param method The HTTP Method. | |
* @return The HTTP Response. | |
*/ | |
public String execute(HttpMethod method) { | |
try { | |
_httpClient.executeMethod(method); | |
return method.getResponseBodyAsString(); | |
} catch (Exception e) { | |
e.printStackTrace(); | |
} | |
return null; | |
} | |
} | |
/** | |
* Human Task Server Helper. Based off the BPMMixIn from the jbpm-component. | |
* Helps with <a href="http://docs.jboss.org/jbpm/v5.0/userguide/ch.Human_Tasks.html">jBPM 5 Human Tasks</a>. | |
* | |
* @author Justin Wyer <<a href="mailto:justin@lifeasageek.com">justin@lifeasageek.com</a>> (C) 2011 Red Hat Inc. | |
*/ | |
class HumanTaskServerHelper { | |
/** The default host. */ | |
public static final String DEFAULT_HOST = "127.0.0.1"; | |
/** The default port. */ | |
public static final int DEFAULT_PORT = 9123; | |
/** The Administrator user id. */ | |
public static final String ADMINISTRATOR = "Administrator"; | |
private TaskServer _server = null; | |
private TaskClient _client = null; | |
private boolean _clientConnected = false; | |
/** | |
* Starts the task server with the default host and port, and the specified user ids to pre-populate the database with. | |
* @param userIds the user ids to pre-populate the database with | |
* @return whether the task server started successfully | |
*/ | |
public boolean startTaskServer(String... userIds) { | |
return startTaskServer(DEFAULT_HOST, DEFAULT_PORT, userIds); | |
} | |
/** | |
* Starts the task server with the specified host and port, and the specified user ids to pre-populate the database with. | |
* @param host the task server host | |
* @param port the task server port | |
* @param userIds the user ids to pre-populate the database with | |
* @return whether the task server started successfully | |
*/ | |
public boolean startTaskServer(String host, int port, String... userIds) { | |
try { | |
boolean ready = false; | |
while (!ready) { | |
Socket socket = null; | |
try { | |
socket = new Socket(host, port); | |
if (socket.isConnected()) { | |
Thread.sleep(1000); | |
} else { | |
ready = true; | |
} | |
} catch (SocketException se) { | |
ready = true; | |
} finally { | |
if (socket != null) { | |
socket.close(); | |
socket = null; | |
} | |
} | |
} | |
EntityManagerFactory emf = Persistence.createEntityManagerFactory("org.jbpm.task"); | |
TaskService taskService = new TaskService(emf, new DoNothingSystemEventListener()); | |
TaskServiceSession taskServiceSession = taskService.createSession(); | |
taskServiceSession.addUser(new User(ADMINISTRATOR)); | |
for (String userId : userIds) { | |
if (!ADMINISTRATOR.equals(userId)) { | |
taskServiceSession.addUser(new User(userId)); | |
} | |
} | |
taskServiceSession.dispose(); | |
_server = new MinaTaskServer(taskService, port, host); | |
new Thread(_server).start(); | |
//_server.start(); | |
return true; | |
} catch (Throwable t) { | |
System.err.println("*** Could not start task server: " + t.getMessage() + " ***"); | |
return false; | |
} | |
} | |
/** | |
* Stops the task server. | |
*/ | |
public void stopTaskServer() { | |
if (_server != null) { | |
try { | |
_server.stop(); | |
} catch (Throwable t) { | |
// just to keep checkstyle happy ("Must have at least one statement.") | |
t.getMessage(); | |
} | |
_server = null; | |
} | |
} | |
/** | |
* Starts the task client with the default host and port of the task server to connect to. | |
* @throws Exception if a problem occurred | |
*/ | |
public void startTaskClient() throws Exception { | |
startTaskClient(DEFAULT_HOST, DEFAULT_PORT); | |
} | |
/** | |
* Starts the task client with the specified host and port of the task server to connect to. | |
* @param host the task server host | |
* @param port the task server port | |
* @throws Exception if a problem occurred | |
*/ | |
public void startTaskClient(String host, int port) throws Exception { | |
if (_client == null) { | |
_client = new TaskClient(new MinaTaskClientConnector(HumanTaskServerHelper.class.getSimpleName(), new MinaTaskClientHandler(new DoNothingSystemEventListener()))); | |
} | |
if (!_clientConnected) { | |
_client.connect(host, port); | |
_clientConnected = true; | |
} | |
} | |
/** | |
* Stops the task client. | |
*/ | |
public void stopTaskClient() { | |
if (_client != null) { | |
if (_clientConnected) { | |
try { | |
_client.disconnect(); | |
} catch (Throwable t) { | |
// just to keep checkstyle happy ("Must have at least one statement.") | |
t.getMessage(); | |
} | |
_clientConnected = false; | |
} | |
_client = null; | |
} | |
} | |
/** | |
* Completes all tasks for the specified user ids, auto-connecting to the task server with the default host and port. | |
* @param userIds the user ids to complete the tasks for | |
* @return if there might be more user tasks to complete | |
* @throws Exception if a problem occurred | |
*/ | |
public boolean completeTasksForUsers(String... userIds) throws Exception { | |
return completeTasksForUsers(DEFAULT_HOST, DEFAULT_PORT, userIds); | |
} | |
/** | |
* Completes all tasks for the specified user ids, connecting to the task server with the specified host and port. | |
* @param host the task server host | |
* @param port the task server port | |
* @param userIds the user ids to complete the tasks for | |
* @return if there might be more user tasks to complete | |
* @throws Exception if a problem occurred | |
*/ | |
public boolean completeTasksForUsers(String host, int port, String... userIds) throws Exception { | |
boolean keepWorking = false; | |
startTaskClient(host, port); | |
for (String userId : userIds) { | |
BlockingTaskSummaryResponseHandler btsrh = new BlockingTaskSummaryResponseHandler(); | |
_client.getTasksOwned(userId, Locale.getDefault().toString().replaceAll("_", "-"), btsrh); | |
List<TaskSummary> tasks = btsrh.getResults(); | |
for (TaskSummary task : tasks) { | |
if (Status.Completed.equals(task.getStatus())) { | |
continue; | |
} | |
User user = task.getActualOwner(); | |
BlockingTaskOperationResponseHandler btorh = new BlockingTaskOperationResponseHandler(); | |
_client.start(task.getId(), user.getId(), btorh); | |
btorh.waitTillDone(10000); | |
btorh = new BlockingTaskOperationResponseHandler(); | |
_client.complete(task.getId(), user.getId(), null, btorh); | |
btorh.waitTillDone(10000); | |
keepWorking = true; | |
} | |
} | |
stopTaskClient(); | |
Thread.sleep(1000); | |
return keepWorking; | |
} | |
} | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment