X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=impl%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fxmlrouter%2Fimpl%2FXMLRouterTest.java;h=64aefa1a8ca9f77e8746d7b0f94c5eacb192a00c;hb=13b475f8216ad5d74d64df4122ea6812554f7a14;hp=baa63a16c7a6de2cc7121ce85407a8131fbf4a13;hpb=f70baadfd579f4d3aa2e8c9ee7d758fb37d7872f;p=xmlrouter

diff --git a/impl/src/test/java/org/wamblee/xmlrouter/impl/XMLRouterTest.java b/impl/src/test/java/org/wamblee/xmlrouter/impl/XMLRouterTest.java
index baa63a1..64aefa1 100644
--- a/impl/src/test/java/org/wamblee/xmlrouter/impl/XMLRouterTest.java
+++ b/impl/src/test/java/org/wamblee/xmlrouter/impl/XMLRouterTest.java
@@ -275,8 +275,7 @@ public class XMLRouterTest {
         when(transformation.getToType()).thenReturn("bla");
         when(transformation.transform(same(source1))).thenReturn(source2);
         routerConfig.transformationConfig().add(transformation);
-        config.getTransformations().replaceTransformations(
-            routerConfig.transformationConfig().map());
+        config.setRouterConfig(routerConfig);
 
         Destination destination = mock(Destination.class);
         when(
@@ -339,9 +338,7 @@ public class XMLRouterTest {
             source1, null);
 
         routerConfig.transformationConfig().add(transformation);
-        config.getTransformations().replaceTransformations(
-            routerConfig.transformationConfig().map());
-
+        config.setRouterConfig(routerConfig);
         Destination destination = mock(Destination.class);
         when(
             destination.chooseFromTargetTypes((Collection<String>) anyObject()))
@@ -359,8 +356,7 @@ public class XMLRouterTest {
             source1, source2);
 
         routerConfig.transformationConfig().add(transformation2);
-        config.getTransformations().replaceTransformations(
-            routerConfig.transformationConfig().map());
+        config.setRouterConfig(routerConfig);
         when(
             destination.chooseFromTargetTypes((Collection<String>) anyObject()))
             .thenReturn(Arrays.asList("bla", "bla2"));
@@ -405,8 +401,7 @@ public class XMLRouterTest {
         Transformation transformation = createTransformation("any", "other",
             source1, source2);
         routerConfig.transformationConfig().add(transformation);
-        config.getTransformations().replaceTransformations(
-            routerConfig.transformationConfig().map());
+        config.setRouterConfig(routerConfig);
 
         router.publish("source", source1);
         verify(listener, times(2)).delivered(any(EventInfo.class),
@@ -428,8 +423,7 @@ public class XMLRouterTest {
         Transformation t2 = createTransformation("intermediate", "other",
             source2, source3);
         routerConfig.transformationConfig().add(t2);
-        config.getTransformations().replaceTransformations(
-            routerConfig.transformationConfig().map());
+        config.setRouterConfig(routerConfig);
 
         router.publish("source", source1);
         verify(listener).delivered(any(EventInfo.class),