From 9cd5b19b8b1fe983f5d28529c188a4a239f4487c Mon Sep 17 00:00:00 2001
From: Arjen Poutsma <apoutsma@vmware.com>
Date: Tue, 19 Feb 2008 12:32:38 +0000
Subject: [PATCH] SWS-289

---
 .../ws/transport/context/DefaultTransportContext.java    | 5 +++++
 .../ws/transport/context/TransportContextHolder.java     | 9 ++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/core/src/main/java/org/springframework/ws/transport/context/DefaultTransportContext.java b/core/src/main/java/org/springframework/ws/transport/context/DefaultTransportContext.java
index 6626ca1c..edfc7a6c 100644
--- a/core/src/main/java/org/springframework/ws/transport/context/DefaultTransportContext.java
+++ b/core/src/main/java/org/springframework/ws/transport/context/DefaultTransportContext.java
@@ -16,6 +16,7 @@
 
 package org.springframework.ws.transport.context;
 
+import org.springframework.util.Assert;
 import org.springframework.ws.transport.WebServiceConnection;
 
 /**
@@ -30,6 +31,7 @@ public class DefaultTransportContext implements TransportContext {
 
     /** Creates a new <code>DefaultTransportContext</code> that exposes the given connection. */
     public DefaultTransportContext(WebServiceConnection connection) {
+        Assert.notNull(connection, "'connection' must not be null");
         this.connection = connection;
     }
 
@@ -37,4 +39,7 @@ public class DefaultTransportContext implements TransportContext {
         return connection;
     }
 
+    public String toString() {
+        return connection.toString();
+    }
 }
diff --git a/core/src/main/java/org/springframework/ws/transport/context/TransportContextHolder.java b/core/src/main/java/org/springframework/ws/transport/context/TransportContextHolder.java
index c01abb99..3deccd6b 100644
--- a/core/src/main/java/org/springframework/ws/transport/context/TransportContextHolder.java
+++ b/core/src/main/java/org/springframework/ws/transport/context/TransportContextHolder.java
@@ -25,7 +25,7 @@ package org.springframework.ws.transport.context;
  */
 public abstract class TransportContextHolder {
 
-    private static final ThreadLocal transportContextHolder = new ThreadLocal();
+    private static final ThreadLocal transportContextHolder = new TransportThreadLocal();
 
     /**
      * Associate the given <code>TransportContext</code> with the current thread.
@@ -44,4 +44,11 @@ public abstract class TransportContextHolder {
     public static TransportContext getTransportContext() {
         return (TransportContext) transportContextHolder.get();
     }
+
+    private static class TransportThreadLocal extends ThreadLocal {
+
+        public String toString() {
+            return "Transport State";
+        }
+    }
 }