svn commit: r701712 - /webservices/commons/trunk/modules/transport/modules/testkit/src/main/java/org/apache/axis2/transport/testkit/http/JettyEchoEndpoint.java

[veithen]

Author: veithen
Date: Sat Oct  4 14:32:31 2008
New Revision: 701712

URL: http://svn.apache.org/viewvc?rev=701712&view=rev
Log:
HTTP transport tests: Added an assertion to check that the transport headers 
are set correctly in the response message context.

Modified:
    
webservices/commons/trunk/modules/transport/modules/testkit/src/main/java/org/apache/axis2/transport/testkit/http/JettyEchoEndpoint.java

Modified: 
webservices/commons/trunk/modules/transport/modules/testkit/src/main/java/org/apache/axis2/transport/testkit/http/JettyEchoEndpoint.java
URL: 
http://svn.apache.org/viewvc/webservices/commons/trunk/modules/transport/modules/testkit/src/main/java/org/apache/axis2/transport/testkit/http/JettyEchoEndpoint.java?rev=701712&r1=701711&r2=701712&view=diff
==============================================================================
--- 
webservices/commons/trunk/modules/transport/modules/testkit/src/main/java/org/apache/axis2/transport/testkit/http/JettyEchoEndpoint.java
 (original)
+++ 
webservices/commons/trunk/modules/transport/modules/testkit/src/main/java/org/apache/axis2/transport/testkit/http/JettyEchoEndpoint.java
 Sat Oct  4 14:32:31 2008
@@ -20,18 +20,30 @@
 package org.apache.axis2.transport.testkit.http;
 
 import java.io.IOException;
+import java.util.Map;
 
+import junit.framework.Assert;
+
+import org.apache.axis2.context.MessageContext;
+import org.apache.axis2.transport.testkit.axis2.MessageContextValidator;
 import org.apache.commons.io.IOUtils;
 import org.mortbay.http.HttpException;
 import org.mortbay.http.HttpRequest;
 import org.mortbay.http.HttpResponse;
 
-public class JettyEchoEndpoint extends JettyEndpoint {
+public class JettyEchoEndpoint extends JettyEndpoint implements 
MessageContextValidator {
     @Override
     protected void handle(String pathParams, HttpRequest request,
             HttpResponse response) throws HttpException, IOException {
         
         response.setContentType(request.getContentType());
+        response.addField("X-Test-Header", "test value");
         IOUtils.copy(request.getInputStream(), response.getOutputStream());
     }
+
+    public void validate(MessageContext msgContext, boolean isResponse) throws 
Exception {
+        Map<?,?> trpHeaders = 
(Map<?,?>)msgContext.getProperty(MessageContext.TRANSPORT_HEADERS);
+        Assert.assertNotNull(trpHeaders);
+        Assert.assertEquals("test value", trpHeaders.get("X-Test-Header"));
+    }
 }