5 files changed, 34 insertions, 29 deletions
diff --git a/test/com/isode/stroke/serializer/AuthChallengeSerializerTest.java b/test/com/isode/stroke/serializer/AuthChallengeSerializerTest.java
index 44ed1b8..cc1523b 100644
--- a/test/com/isode/stroke/serializer/AuthChallengeSerializerTest.java
+++ b/test/com/isode/stroke/serializer/AuthChallengeSerializerTest.java
@@ -16,6 +16,7 @@ import org.junit.Test;
 import com.isode.stroke.serializer.AuthChallengeSerializer;
 import com.isode.stroke.elements.AuthChallenge;
 import com.isode.stroke.base.ByteArray;
+import com.isode.stroke.base.SafeByteArray;
 
 public class AuthChallengeSerializerTest {
 
@@ -33,9 +34,9 @@ public class AuthChallengeSerializerTest {
 		authChallenge.setValue(new ByteArray("foo"));
 
 		assertEquals(
-			"<challenge xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			new SafeByteArray("<challenge xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
 				"Zm9v" +
-			"</challenge>", testling.serialize(authChallenge));
+			"</challenge>"), testling.serialize(authChallenge));
 	}
 
 	@Test
@@ -44,7 +45,7 @@ public class AuthChallengeSerializerTest {
 		AuthChallenge authChallenge = new AuthChallenge();
 
 		assertEquals(
-			"<challenge xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\"></challenge>", testling.serialize(authChallenge));
+			new SafeByteArray("<challenge xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\"></challenge>"), testling.serialize(authChallenge));
 	}
 
 	@Test
@@ -54,8 +55,8 @@ public class AuthChallengeSerializerTest {
 		authChallenge.setValue(new ByteArray());
 
 		assertEquals(
-			"<challenge xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			new SafeByteArray("<challenge xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
 				"=" +
-			"</challenge>", testling.serialize(authChallenge));
+			"</challenge>"), testling.serialize(authChallenge));
 	}
 }
 \ No newline at end of file
diff --git a/test/com/isode/stroke/serializer/AuthRequestSerializerTest.java b/test/com/isode/stroke/serializer/AuthRequestSerializerTest.java
index eae4354..69d4442 100644
--- a/test/com/isode/stroke/serializer/AuthRequestSerializerTest.java
+++ b/test/com/isode/stroke/serializer/AuthRequestSerializerTest.java
@@ -16,6 +16,7 @@ import org.junit.Test;
 import com.isode.stroke.serializer.AuthRequestSerializer;
 import com.isode.stroke.elements.AuthRequest;
 import com.isode.stroke.base.ByteArray;
+import com.isode.stroke.base.SafeByteArray;
 
 public class AuthRequestSerializerTest {
 
@@ -30,12 +31,12 @@ public class AuthRequestSerializerTest {
 	public void testSerialize() {
 		AuthRequestSerializer testling = new AuthRequestSerializer();
 		AuthRequest authRequest = new AuthRequest("PLAIN");
-		authRequest.setMessage(new ByteArray("foo"));
+		authRequest.setMessage(new SafeByteArray("foo"));
 
 		assertEquals(
-			"<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"PLAIN\">" +
+			new SafeByteArray("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"PLAIN\">" +
 				"Zm9v" +
-			"</auth>", testling.serialize(authRequest));
+			"</auth>"), testling.serialize(authRequest));
 	}
 
 	@Test
@@ -44,19 +45,19 @@ public class AuthRequestSerializerTest {
 		AuthRequest authRequest = new AuthRequest("PLAIN");
 
 		assertEquals(
-			"<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"PLAIN\">" +
-			"</auth>", testling.serialize(authRequest));
+			new SafeByteArray("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"PLAIN\">" +
+			"</auth>"), testling.serialize(authRequest));
 	}
 
 	@Test
 	public void testSerialize_EmptyMessage() {
 		AuthRequestSerializer testling = new AuthRequestSerializer();
 		AuthRequest authRequest = new AuthRequest("PLAIN");
-		authRequest.setMessage(new ByteArray());
+		authRequest.setMessage(new SafeByteArray());
 
 		assertEquals(
-			"<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"PLAIN\">" +
+			new SafeByteArray("<auth xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\" mechanism=\"PLAIN\">" +
 				"=" +
-			"</auth>", testling.serialize(authRequest));
+			"</auth>"), testling.serialize(authRequest));
 	}
 }
 \ No newline at end of file
diff --git a/test/com/isode/stroke/serializer/AuthResponseSerializerTest.java b/test/com/isode/stroke/serializer/AuthResponseSerializerTest.java
index 9b0fe71..a80dc81 100644
--- a/test/com/isode/stroke/serializer/AuthResponseSerializerTest.java
+++ b/test/com/isode/stroke/serializer/AuthResponseSerializerTest.java
@@ -16,6 +16,7 @@ import org.junit.Test;
 import com.isode.stroke.serializer.AuthResponseSerializer;
 import com.isode.stroke.elements.AuthResponse;
 import com.isode.stroke.base.ByteArray;
+import com.isode.stroke.base.SafeByteArray;
 
 public class AuthResponseSerializerTest {
 
@@ -30,12 +31,12 @@ public class AuthResponseSerializerTest {
 	public void testSerialize() {
 		AuthResponseSerializer testling = new AuthResponseSerializer();
 		AuthResponse authResponse = new AuthResponse();
-		authResponse.setValue(new ByteArray("foo"));
+		authResponse.setValue(new SafeByteArray("foo"));
 
 		assertEquals(
-			"<response xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			new SafeByteArray("<response xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
 				"Zm9v" +
-			"</response>", testling.serialize(authResponse));
+			"</response>"), testling.serialize(authResponse));
 	}
 
 	@Test
@@ -44,19 +45,19 @@ public class AuthResponseSerializerTest {
 		AuthResponse authResponse = new AuthResponse();
 
 		assertEquals(
-			"<response xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
-			"</response>", testling.serialize(authResponse));
+			new SafeByteArray("<response xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			"</response>"), testling.serialize(authResponse));
 	}
 
 	@Test
 	public void testSerialize_EmptyMessage() {
 		AuthResponseSerializer testling = new AuthResponseSerializer();
 		AuthResponse authResponse = new AuthResponse();
-		authResponse.setValue(new ByteArray());
+		authResponse.setValue(new SafeByteArray());
 
 		assertEquals(
-			"<response xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			new SafeByteArray("<response xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
 				"" +
-			"</response>", testling.serialize(authResponse));
+			"</response>"), testling.serialize(authResponse));
 	}
 }
 \ No newline at end of file
diff --git a/test/com/isode/stroke/serializer/AuthSuccessSerializerTest.java b/test/com/isode/stroke/serializer/AuthSuccessSerializerTest.java
index ee9b595..198080b 100644
--- a/test/com/isode/stroke/serializer/AuthSuccessSerializerTest.java
+++ b/test/com/isode/stroke/serializer/AuthSuccessSerializerTest.java
@@ -16,6 +16,7 @@ import org.junit.Test;
 import com.isode.stroke.serializer.AuthSuccessSerializer;
 import com.isode.stroke.elements.AuthSuccess;
 import com.isode.stroke.base.ByteArray;
+import com.isode.stroke.base.SafeByteArray;
 
 public class AuthSuccessSerializerTest {
 
@@ -33,9 +34,9 @@ public class AuthSuccessSerializerTest {
 		authSuccess.setValue(new ByteArray("foo"));
 
 		assertEquals(
-			"<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			new SafeByteArray("<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
 				"Zm9v" +
-			"</success>", testling.serialize(authSuccess));
+			"</success>"), testling.serialize(authSuccess));
 	}
 
 	@Test
@@ -44,8 +45,8 @@ public class AuthSuccessSerializerTest {
 		AuthSuccess authSuccess = new AuthSuccess();
 
 		assertEquals(
-			"<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
-			"</success>", testling.serialize(authSuccess));
+			new SafeByteArray("<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			"</success>"), testling.serialize(authSuccess));
 	}
 
 	@Test
@@ -55,8 +56,8 @@ public class AuthSuccessSerializerTest {
 		authSuccess.setValue(new ByteArray());
 
 		assertEquals(
-			"<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
+			new SafeByteArray("<success xmlns=\"urn:ietf:params:xml:ns:xmpp-sasl\">" +
 				"=" +
-			"</success>", testling.serialize(authSuccess));
+			"</success>"), testling.serialize(authSuccess));
 	}
 }
 \ No newline at end of file
diff --git a/test/com/isode/stroke/serializer/StreamFeaturesSerializerTest.java b/test/com/isode/stroke/serializer/StreamFeaturesSerializerTest.java
index 67d9bb0..76696b5 100644
--- a/test/com/isode/stroke/serializer/StreamFeaturesSerializerTest.java
+++ b/test/com/isode/stroke/serializer/StreamFeaturesSerializerTest.java
@@ -15,6 +15,7 @@ import static org.junit.Assert.assertEquals;
 import org.junit.Test;
 import com.isode.stroke.serializer.StreamFeaturesSerializer;
 import com.isode.stroke.elements.StreamFeatures;
+import com.isode.stroke.base.SafeByteArray;
 
 public class StreamFeaturesSerializerTest {
 
@@ -40,7 +41,7 @@ public class StreamFeaturesSerializerTest {
 		streamFeatures.setHasRosterVersioning();
 
 		assertEquals(
-		"<stream:features>"
+		new SafeByteArray("<stream:features>"
 	+		"<starttls xmlns=\"urn:ietf:params:xml:ns:xmpp-tls\"/>"
 	+		"<compression xmlns=\"http://jabber.org/features/compress\">"
 	+			"<method>zlib</method>"
@@ -54,6 +55,6 @@ public class StreamFeaturesSerializerTest {
 	+		"<session xmlns=\"urn:ietf:params:xml:ns:xmpp-session\"/>"
 	+		"<sm xmlns=\"urn:xmpp:sm:2\"/>"
 	+		"<ver xmlns=\"urn:xmpp:features:rosterver\"/>"
-	+	"</stream:features>", testling.serialize(streamFeatures));
+	+	"</stream:features>"), testling.serialize(streamFeatures));
 	}
 }
 \ No newline at end of file