1 files changed, 121 insertions, 0 deletions
diff --git a/velocity-engine-core/src/test/java/org/apache/velocity/test/util/introspection/ChainedUberspectorsTestCase.java b/velocity-engine-core/src/test/java/org/apache/velocity/test/util/introspection/ChainedUberspectorsTestCase.java
new file mode 100644
index 00000000..2687d662
--- /dev/null
+++ b/velocity-engine-core/src/test/java/org/apache/velocity/test/util/introspection/ChainedUberspectorsTestCase.java
@@ -0,0 +1,121 @@
+package org.apache.velocity.test.util.introspection;
+
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+import junit.framework.Test;
+import junit.framework.TestSuite;
+import org.apache.velocity.VelocityContext;
+import org.apache.velocity.app.Velocity;
+import org.apache.velocity.test.BaseTestCase;
+import org.apache.velocity.test.misc.TestLogger;
+import org.apache.velocity.util.introspection.AbstractChainableUberspector;
+import org.apache.velocity.util.introspection.Info;
+import org.apache.velocity.util.introspection.UberspectImpl;
+import org.apache.velocity.util.introspection.VelPropertyGet;
+import org.apache.velocity.util.introspection.VelPropertySet;
+
+import java.io.StringWriter;
+
+/**
+ * Tests uberspectors chaining
+ */
+public class ChainedUberspectorsTestCase extends BaseTestCase {
+
+    public ChainedUberspectorsTestCase(String name)
+    	throws Exception
+    {
+        super(name);
+    }
+
+    public static Test suite()
+    {
+        return new TestSuite(ChainedUberspectorsTestCase.class);
+    }
+
+    @Override
+    public void setUp()
+            throws Exception
+    {
+        Velocity.reset();
+        Velocity.setProperty(Velocity.RUNTIME_LOG_INSTANCE, new TestLogger());
+        Velocity.addProperty(Velocity.UBERSPECT_CLASSNAME,"org.apache.velocity.util.introspection.UberspectImpl");
+        Velocity.addProperty(Velocity.UBERSPECT_CLASSNAME,"org.apache.velocity.test.util.introspection.ChainedUberspectorsTestCase$ChainedUberspector");
+        Velocity.addProperty(Velocity.UBERSPECT_CLASSNAME,"org.apache.velocity.test.util.introspection.ChainedUberspectorsTestCase$LinkedUberspector");
+	    Velocity.init();
+    }
+
+    @Override
+    public void tearDown()
+    {
+    }
+
+    public void testChaining()
+    	throws Exception
+    {
+        VelocityContext context = new VelocityContext();
+        context.put("foo",new Foo());
+        StringWriter writer = new StringWriter();
+
+        Velocity.evaluate(context,writer,"test","$foo.zeMethod()");
+        assertEquals(writer.toString(),"ok");
+
+        Velocity.evaluate(context,writer,"test","#set($foo.foo = 'someValue')");
+
+        writer = new StringWriter();
+        Velocity.evaluate(context,writer,"test","$foo.bar");
+        assertEquals(writer.toString(),"someValue");
+
+        writer = new StringWriter();
+        Velocity.evaluate(context,writer,"test","$foo.foo");
+        assertEquals(writer.toString(),"someValue");
+    }
+
+    // replaces getFoo by getBar
+    public static class ChainedUberspector extends AbstractChainableUberspector
+    {
+        @Override
+        public VelPropertySet getPropertySet(Object obj, String identifier, Object arg, Info info)
+        {
+            identifier = identifier.replaceAll("foo","bar");
+            return inner.getPropertySet(obj,identifier,arg,info);
+        }
+    }
+
+    // replaces setFoo by setBar
+    public static class LinkedUberspector extends UberspectImpl
+    {
+        @Override
+        public VelPropertyGet getPropertyGet(Object obj, String identifier, Info info)
+        {
+            identifier = identifier.replaceAll("foo","bar");
+            return super.getPropertyGet(obj,identifier,info);
+        }
+    }
+
+    public static class Foo
+    {
+        private String bar;
+
+        public String zeMethod() { return "ok"; }
+        public String getBar() { return bar; }
+        public void setBar(String s) { bar = s; }
+    }
+
+}