11:07 a.m.
Author: pferraro
Date: 2008-12-12 12:07:03 -0500 (Fri, 12 Dec 2008)
New Revision: 2147
Added:
trunk/mod_cluster/src/test/java/org/jboss/modcluster/test/
trunk/mod_cluster/src/test/java/org/jboss/modcluster/test/MockProxy.java
Log:
Mock proxy for use in integration testing.
Added: trunk/mod_cluster/src/test/java/org/jboss/modcluster/test/MockProxy.java
===================================================================
--- trunk/mod_cluster/src/test/java/org/jboss/modcluster/test/MockProxy.java
(rev 0)
+++ trunk/mod_cluster/src/test/java/org/jboss/modcluster/test/MockProxy.java 2008-12-12
17:07:03 UTC (rev 2147)
@@ -0,0 +1,232 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2008, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY 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 along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.jboss.modcluster.test;
+
+import java.io.BufferedReader;
+import java.io.BufferedWriter;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.io.OutputStreamWriter;
+import java.net.InetAddress;
+import java.net.ServerSocket;
+import java.net.Socket;
+import java.net.SocketTimeoutException;
+import java.net.URLDecoder;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.concurrent.BlockingQueue;
+import java.util.concurrent.LinkedBlockingQueue;
+
+import javax.management.MBeanServerConnection;
+import javax.management.ObjectName;
+
+import org.jboss.modcluster.mcmp.MCMPRequest;
+import org.jboss.modcluster.mcmp.MCMPRequestType;
+import org.jboss.modcluster.mcmp.MCMPServerState;
+
+/**
+ * Mock httpd proxy that queues all received messages
+ * @author Paul Ferraro
+ */
+@SuppressWarnings("boxing")
+public class MockProxy implements Runnable
+{
+ private static final String MOD_CLUSTER_SERVICE =
"jboss.web:service=ModCluster";
+
+ private final BlockingQueue<MCMPRequest> requests = new
LinkedBlockingQueue<MCMPRequest>();
+ private final int port = 6666;
+ private final int socketTimeout = 10000;
+
+ private ServerSocket server;
+ private Thread worker;
+
+ private volatile MCMPServerState.State state = MCMPServerState.State.OK;
+
+ private static final Map<String, MCMPRequestType> requestTypes = new
HashMap<String, MCMPRequestType>();
+ {
+ for (MCMPRequestType type: MCMPRequestType.values())
+ {
+ requestTypes.put(type.getCommand(), type);
+ }
+ }
+
+ public void run()
+ {
+ try
+ {
+ while (!Thread.currentThread().isInterrupted())
+ {
+ try
+ {
+ Socket socket = this.server.accept();
+ System.out.println("Socket connection accepted");
+ socket.setSoTimeout(this.socketTimeout);
+
+ try
+ {
+ BufferedReader reader = new BufferedReader(new
InputStreamReader(socket.getInputStream()));
+ BufferedWriter writer = new BufferedWriter(new
OutputStreamWriter(socket.getOutputStream()));
+
+ String line = reader.readLine();
+
+ while ((line != null) &&
!Thread.currentThread().isInterrupted())
+ {
+ if (line.length() == 0)
+ {
+ line = reader.readLine();
+ continue;
+ }
+
+ String[] parts = line.split("\\s");
+ MCMPRequestType type = requestTypes.get(parts[0]);
+ boolean wildcard = parts[1].endsWith("*");
+
+ line = reader.readLine();
+
+ while ((line != null) && (line.length() > 0))
+ {
+ // Ignore headers
+ line = reader.readLine();
+ }
+
+ if (line != null)
+ {
+ line = reader.readLine();
+ }
+
+ Map<String, String> parameters = new HashMap<String,
String>();
+ String jvmRoute = null;
+
+ if (line != null)
+ {
+ if (line.length() > 0)
+ {
+ for (String parameter: line.split("&"))
+ {
+ parts = parameter.split("=");
+
+ String name = parts[0];
+ String value = URLDecoder.decode(parts[1],
"UTF-8");
+
+ if (name.equals("JVMRoute"))
+ {
+ jvmRoute = value;
+ }
+ else
+ {
+ parameters.put(name, value);
+ }
+ }
+ }
+
+ this.requests.add(new MCMPRequest(type, wildcard, jvmRoute,
parameters));
+
+ if (this.state == MCMPServerState.State.OK)
+ {
+ writer.write("HTTP/1.0 200 OK");
+ }
+ else
+ {
+ writer.write("HTTP/1.0 500 ERROR");
+
+ if (this.state == MCMPServerState.State.DOWN)
+ {
+ writer.newLine();
+ writer.write("Type: SYNTAX");
+ }
+ }
+
+ writer.newLine();
+ writer.newLine();
+ writer.flush();
+
+ line = reader.readLine();
+ }
+ }
+ }
+ finally
+ {
+ socket.close();
+ }
+ }
+ catch (SocketTimeoutException e)
+ {
+ System.out.println("Socket accept timeout");
+ }
+ }
+ }
+ catch (Throwable e)
+ {
+ e.printStackTrace(System.err);
+ }
+ }
+
+ public void setState(MCMPServerState.State state)
+ {
+ this.state = state;
+ }
+
+ public BlockingQueue<MCMPRequest> getRequests()
+ {
+ return this.requests;
+ }
+
+ public void start() throws Exception
+ {
+ this.server = new ServerSocket(this.port, 50, null);
+ this.server.setSoTimeout(this.socketTimeout);
+
+ this.worker = new Thread(this);
+ this.worker.start();
+
+ MBeanServerConnection server = new MBeanServerConnector(0).getServer();
+
+ ObjectName name = ObjectName.getInstance(MOD_CLUSTER_SERVICE);
+
+ String host = InetAddress.getLocalHost().getHostName();
+ int port = this.server.getLocalPort();
+
+ server.invoke(name, "addProxy", new Object[] { host, port }, new String[]
{ String.class.getName(), Integer.TYPE.getName() });
+ }
+
+ public void close() throws IOException
+ {
+ System.out.println("Begin close " + new java.util.Date());
+ if (this.worker != null) this.worker.interrupt();
+
+ // Break ServerSocket.accept()
+ if (this.server != null) this.server.close();
+
+ try
+ {
+ // Wait for worker to finish
+ this.worker.join();
+ }
+ catch (InterruptedException e)
+ {
+ Thread.currentThread().interrupt();
+ }
+
+ this.requests.clear();
+ System.out.println("End close " + new java.util.Date());
+ }
+}
\ No newline at end of file