SYNAPSE-1109: fix for unit test
diff --git a/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java b/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java
index eadf01c..8d4e42e 100755
--- a/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java
+++ b/modules/core/src/test/java/org/apache/synapse/util/MessageHelperTest.java
@@ -22,6 +22,7 @@
import javax.activation.DataHandler;
import junit.framework.TestCase;
+import org.apache.axis2.addressing.EndpointReference;
import org.apache.axis2.context.MessageContext;
/**
@@ -37,7 +38,7 @@
Object result = newMc.getProperty(key);
assertEquals(result, "propValue");
}
-
+
// Regression test for SYNAPSE-309
public void testClonePartiallyWithAttachments() throws Exception {
MessageContext origMc = new MessageContext();
@@ -52,10 +53,10 @@
public void testClonePartiallyWithFrom() throws Exception {
String fromValue = "uri://some-from-value";
MessageContext origMc = new MessageContext();
- origMc.setFrom(fromValue);
+ origMc.setFrom(new EndpointReference(fromValue));
MessageContext newMc = MessageHelper.clonePartially(origMc);
- Object result = newMc.getFrom();
- assertEquals(fromValue, result);
+ EndpointReference result = newMc.getFrom();
+ assertEquals(fromValue, result.getAddress());
}
}