From e95d0100812a1d7461a2458c8bd2c845d1fc7816 Mon Sep 17 00:00:00 2001
From: Erik Brakkee <erik@brakkee.org>
Date: Thu, 11 Aug 2011 22:03:07 +0200
Subject: [PATCH] Verified that filtering of types created by transformations
 also works.

---
 .../org/wamblee/xmlrouter/impl/XMLRouter.java |  4 +-
 .../wamblee/xmlrouter/impl/XMLRouterTest.java | 78 +++++++++++++++++++
 2 files changed, 79 insertions(+), 3 deletions(-)

diff --git a/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java b/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
index 09bbdc1..9c0f2d4 100644
--- a/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
+++ b/impl/src/main/java/org/wamblee/xmlrouter/impl/XMLRouter.java
@@ -41,8 +41,6 @@ import org.wamblee.xmlrouter.publish.Gateway;
 import org.wamblee.xmlrouter.subscribe.Destination;
 import org.wamblee.xmlrouter.subscribe.DestinationRegistry;
 
-// TODO check intermediate types during transformation based on filters. 
-
 /**
  * The XML Router.
  * 
@@ -97,7 +95,7 @@ public class XMLRouter implements Gateway, DestinationRegistry {
                     LOGGER
                         .log(
                             Level.FINE,
-                            "Event ''0}'' from source {1} removed because of filters.",
+                            "Event ''{0}'' from source ''{1}'' removed because of filters.",
                             new Object[] { doc, aSource });
                 }
             }
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 c1776f4..c4375c7 100644
--- a/impl/src/test/java/org/wamblee/xmlrouter/impl/XMLRouterTest.java
+++ b/impl/src/test/java/org/wamblee/xmlrouter/impl/XMLRouterTest.java
@@ -21,7 +21,9 @@ import static org.mockito.Mockito.*;
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.UUID;
+import java.util.logging.Handler;
 import java.util.logging.Level;
+import java.util.logging.Logger;
 
 import javax.xml.transform.dom.DOMSource;
 
@@ -80,6 +82,10 @@ public class XMLRouterTest {
 
     @Before
     public void setUp() {
+
+        Logger logger = Logger.getLogger(XMLRouter.class.getName());
+        setLogLevel(logger, Level.FINEST);
+
         routerConfig = new SingleRouterConfig(new Id<RouterConfig>(
             "routerconfig"));
         config = new XMLRouterConfigurationImpl(routerConfig);
@@ -91,6 +97,17 @@ public class XMLRouterTest {
         source3 = mock(DOMSource.class);
     }
 
+    private void setLogLevel(Logger aLogger, Level aLevel) {
+        aLogger.setLevel(aLevel);
+        for (Handler handler : aLogger.getHandlers()) {
+            handler.setLevel(Level.FINEST);
+        }
+        Logger parent = aLogger.getParent();
+        if (parent != null) {
+            setLogLevel(parent, aLevel);
+        }
+    }
+
     @Test
     public void testNoInputDocumentsRegistered() {
         Destination destination = new MyDestination(true, Arrays.asList("any"));
@@ -314,6 +331,41 @@ public class XMLRouterTest {
         verify(listener).notDelivered(any(EventInfo.class));
     }
 
+    @Test
+    public void testOneTransformationOneDestinationFilterRejectsDestinationDocType() {
+        registerDocumentType("any");
+
+        Transformation transformation = mock(Transformation.class);
+        when(transformation.getId())
+            .thenReturn(new Id<Transformation>("trans"));
+        when(transformation.getFromType()).thenReturn("any");
+        when(transformation.getToType()).thenReturn("bla");
+        when(transformation.transform(same(source1))).thenReturn(source2);
+        routerConfig.transformationConfig().add(transformation);
+
+        Filter filter = mock(Filter.class);
+        when(filter.getId()).thenReturn(new Id<Filter>("f"));
+        when(filter.isAllowed(anyString(), same(source2))).thenReturn(false);
+        when(filter.isAllowed(anyString(), same(source1))).thenReturn(true);
+        routerConfig.filterConfig().add(filter);
+
+        config.setRouterConfig(routerConfig);
+
+        Destination destination = mock(Destination.class);
+        when(
+            destination.chooseFromTargetTypes((Collection<String>) anyObject()))
+            .thenReturn(Arrays.asList("bla"));
+
+        router.registerDestination(destination);
+
+        when(destination.receive(any(DOMSource.class))).thenReturn(true);
+        router.publish("bla", source1);
+        verify(listener).notDelivered(any(EventInfo.class));
+
+        verify(transformation).transform(source1);
+        verify(destination, never()).receive(any(DOMSource.class));
+    }
+
     @Test
     public void testMisbehavingTransformationOneDestination() {
         registerDocumentType("any");
@@ -451,6 +503,32 @@ public class XMLRouterTest {
         verify(dest).receive(same(source3));
     }
 
+    @Test
+    public void testMultipleTransformationsFilterRejectsIntermediateDocument() {
+        Destination dest = registerDestination(true, "other");
+        registerDocumentType("any", source1);
+        registerDocumentType("other", source3);
+
+        Transformation t1 = createTransformation("any", "intermediate",
+            source1, source2);
+        routerConfig.transformationConfig().add(t1);
+        Transformation t2 = createTransformation("intermediate", "other",
+            source2, source3);
+        routerConfig.transformationConfig().add(t2);
+        config.setRouterConfig(routerConfig);
+
+        Filter filter = mock(Filter.class);
+        when(filter.getId()).thenReturn(new Id<Filter>("f"));
+        when(filter.isAllowed(anyString(), same(source1))).thenReturn(true);
+        when(filter.isAllowed(anyString(), same(source2))).thenReturn(false);
+        when(filter.isAllowed(anyString(), same(source3))).thenReturn(true);
+        routerConfig.filterConfig().add(filter);
+
+        router.publish("source", source1);
+        verify(listener).notDelivered(any(EventInfo.class));
+        verify(dest, never()).receive(any(DOMSource.class));
+    }
+
     @Test
     public void testDestinationGivesError() {
         Destination destination = mock(Destination.class);
-- 
2.31.1