X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=impl%2Fsrc%2Ftest%2Fjava%2Forg%2Fwamblee%2Fxmlrouter%2Fimpl%2FTransformationsTest.java;h=775e5000b8dc7c01121881f18e5d6731d17e3122;hb=19413a6699295b4bbebc1b3bdb9838fd4370e581;hp=230cd03d73c2e90e3c77fd00abd9c8c3b1579dd0;hpb=f77e29e9ae2505009ffc9270f4c4adfde331fa62;p=xmlrouter

diff --git a/impl/src/test/java/org/wamblee/xmlrouter/impl/TransformationsTest.java b/impl/src/test/java/org/wamblee/xmlrouter/impl/TransformationsTest.java
index 230cd03..775e500 100644
--- a/impl/src/test/java/org/wamblee/xmlrouter/impl/TransformationsTest.java
+++ b/impl/src/test/java/org/wamblee/xmlrouter/impl/TransformationsTest.java
@@ -18,6 +18,8 @@ package org.wamblee.xmlrouter.impl;
 import static junit.framework.Assert.*;
 
 import java.util.Collection;
+import java.util.LinkedHashMap;
+import java.util.Map;
 
 import javax.xml.transform.dom.DOMSource;
 
@@ -71,10 +73,22 @@ public class TransformationsTest {
         transformations = new Transformations();
     }
 
+    private Map<Id<Transformation>, Transformation> createTransformations(
+        long aStartId, Transformation... aTransformations) {
+        Map<Id<Transformation>, Transformation> res = new LinkedHashMap<Id<Transformation>, Transformation>();
+
+        long id = aStartId;
+        for (Transformation t : aTransformations) {
+            res.put(new Id<Transformation>(id++), t);
+        }
+        return res;
+    }
+
     @Test
     public void testOneTransformation() {
-        Id<Transformation> seqno = transformations
-            .addTransformation(new MyTransformation("A", "B"));
+        transformations.replaceTransformations(createTransformations(10L,
+            new MyTransformation("A", "B")));
+
         System.out.println(transformations.toString());
         TransformationPath path = transformations.getPath("A", "B");
         assertEquals(1, path.size());
@@ -92,25 +106,23 @@ public class TransformationsTest {
 
     @Test
     public void testMultipleTransformations() {
-        Id<Transformation> seqno1 = transformations
-            .addTransformation(new MyTransformation("A", "B"));
-        Id<Transformation> seqno2 = transformations
-            .addTransformation(new MyTransformation("B", "C"));
-        Id<Transformation> seqno3 = transformations
-            .addTransformation(new MyTransformation("C", "A"));
+        transformations.replaceTransformations(createTransformations(10L,
+            new MyTransformation("A", "B"), new MyTransformation("B", "C"),
+            new MyTransformation("C", "A")));
+
         System.out.println(transformations);
         assertEquals(2, transformations.getPath("C", "B").size());
-        assertFalse(seqno1.equals(seqno2));
-        assertFalse(seqno2.equals(seqno3));
-        assertFalse(seqno1.equals(seqno3));
 
-        transformations.removeTransformation(seqno1);
+        transformations.replaceTransformations(createTransformations(10L,
+            new MyTransformation("B", "C"), new MyTransformation("C", "A")));
+
         assertNull(transformations.getPath("C", "B"));
     }
 
     @Test
     public void testUnknownDestination() {
-        transformations.addTransformation(new MyTransformation("A", "B"));
+        transformations.replaceTransformations(createTransformations(10L,
+            new MyTransformation("A", "B")));
         assertNull(transformations.getPath("A", "D"));
     }