svn commit: r1361116 - /camel/trunk/camel-core/src/test/java/org/apache/camel/component/mock/MockEndpointTest.java

[ningjiang]

Author: ningjiang
Date: Fri Jul 13 09:11:04 2012
New Revision: 1361116

URL: http://svn.apache.org/viewvc?rev=1361116&view=rev
Log:
CAMEL-5440 Added an unit test to show the issue

Modified:
    
camel/trunk/camel-core/src/test/java/org/apache/camel/component/mock/MockEndpointTest.java

Modified: 
camel/trunk/camel-core/src/test/java/org/apache/camel/component/mock/MockEndpointTest.java
URL: 
http://svn.apache.org/viewvc/camel/trunk/camel-core/src/test/java/org/apache/camel/component/mock/MockEndpointTest.java?rev=1361116&r1=1361115&r2=1361116&view=diff
==============================================================================
--- 
camel/trunk/camel-core/src/test/java/org/apache/camel/component/mock/MockEndpointTest.java
 (original)
+++ 
camel/trunk/camel-core/src/test/java/org/apache/camel/component/mock/MockEndpointTest.java
 Fri Jul 13 09:11:04 2012
@@ -16,6 +16,8 @@
  */
 package org.apache.camel.component.mock;
 
+import java.io.ByteArrayInputStream;
+import java.io.InputStream;
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
@@ -647,6 +649,22 @@ public class MockEndpointTest extends Co
 
         mock.assertIsSatisfied();
     }
+    
+    public void testSetMultipleExpectedHeaders4() throws Exception {
+        // to test the header value with Stream which can only be consumed once
+        InputStream is = new ByteArrayInputStream("Test".getBytes());
+        MockEndpoint mock = getMockEndpoint("mock:result");
+        mock.expectedMessageCount(1);
+        mock.expectedHeaderReceived("foo", 123);
+        mock.expectedHeaderReceived("bar", "Test");
+
+        Map<String, Object> map = new HashMap<String, Object>();
+        map.put("foo", 123);
+        map.put("bar", is);
+        template.sendBodyAndHeaders("direct:a", "Hello World", map);
+
+        mock.assertIsSatisfied();
+    }
 
     public void testSetMultipleExpectedHeadersShouldFail() throws Exception {
         MockEndpoint mock = getMockEndpoint("mock:result");