001: /*
002: $Id: TupleListTest.java 3790 2006-05-30 18:27:48Z blackdrag $
003:
004: Copyright 2003 (C) James Strachan and Bob Mcwhirter. All Rights Reserved.
005:
006: Redistribution and use of this software and associated documentation
007: ("Software"), with or without modification, are permitted provided
008: that the following conditions are met:
009:
010: 1. Redistributions of source code must retain copyright
011: statements and notices. Redistributions must also contain a
012: copy of this document.
013:
014: 2. Redistributions in binary form must reproduce the
015: above copyright notice, this list of conditions and the
016: following disclaimer in the documentation and/or other
017: materials provided with the distribution.
018:
019: 3. The name "groovy" must not be used to endorse or promote
020: products derived from this Software without prior written
021: permission of The Codehaus. For written permission,
022: please contact info@codehaus.org.
023:
024: 4. Products derived from this Software may not be called "groovy"
025: nor may "groovy" appear in their names without prior written
026: permission of The Codehaus. "groovy" is a registered
027: trademark of The Codehaus.
028:
029: 5. Due credit should be given to The Codehaus -
030: http://groovy.codehaus.org/
031:
032: THIS SOFTWARE IS PROVIDED BY THE CODEHAUS AND CONTRIBUTORS
033: ``AS IS'' AND ANY EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT
034: NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
035: FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
036: THE CODEHAUS OR ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
037: INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
038: (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
039: SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
040: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
041: STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
042: ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
043: OF THE POSSIBILITY OF SUCH DAMAGE.
044:
045: */
046:
047: package org.codehaus.groovy.classgen;
048:
049: import org.codehaus.groovy.ast.ClassHelper;
050: import org.codehaus.groovy.ast.ClassNode;
051: import org.codehaus.groovy.ast.ConstructorNode;
052: import org.codehaus.groovy.ast.MethodNode;
053: import org.codehaus.groovy.ast.Parameter;
054: import org.codehaus.groovy.ast.PropertyNode;
055: import org.codehaus.groovy.ast.expr.DeclarationExpression;
056: import org.codehaus.groovy.ast.expr.ConstantExpression;
057: import org.codehaus.groovy.ast.expr.Expression;
058: import org.codehaus.groovy.ast.expr.ListExpression;
059: import org.codehaus.groovy.ast.expr.MapExpression;
060: import org.codehaus.groovy.ast.expr.TupleExpression;
061: import org.codehaus.groovy.ast.expr.VariableExpression;
062: import org.codehaus.groovy.ast.stmt.ExpressionStatement;
063: import org.codehaus.groovy.ast.stmt.ForStatement;
064: import org.codehaus.groovy.ast.stmt.Statement;
065: import org.codehaus.groovy.ast.stmt.BlockStatement;
066: import org.codehaus.groovy.runtime.InvokerHelper;
067: import org.codehaus.groovy.runtime.InvokerInvocationException;
068: import org.codehaus.groovy.syntax.Token;
069:
070: /**
071: *
072: * @author <a href="mailto:james@coredevelopers.net">James Strachan</a>
073: * @version $Revision: 3790 $
074: */
075: public class TupleListTest extends TestSupport {
076:
077: public void testIterateOverTuple() throws Exception {
078: TupleExpression listExpression = new TupleExpression();
079: listExpression.addExpression(new ConstantExpression("a"));
080: listExpression.addExpression(new ConstantExpression("b"));
081: listExpression.addExpression(new ConstantExpression("c"));
082: assertIterate("iterateOverTuple", listExpression);
083: }
084:
085: public void testIterateOverList() throws Exception {
086: ListExpression listExpression = new ListExpression();
087: listExpression.addExpression(new ConstantExpression("a"));
088: listExpression.addExpression(new ConstantExpression("b"));
089: listExpression.addExpression(new ConstantExpression("c"));
090: listExpression.addExpression(new ConstantExpression("a"));
091: listExpression.addExpression(new ConstantExpression("b"));
092: listExpression.addExpression(new ConstantExpression("c"));
093: assertIterate("iterateOverList", listExpression);
094: }
095:
096: public void testIterateOverMap() throws Exception {
097: MapExpression mapExpression = new MapExpression();
098: mapExpression.addMapEntryExpression(
099: new ConstantExpression("a"),
100: new ConstantExpression("x"));
101: mapExpression.addMapEntryExpression(
102: new ConstantExpression("b"),
103: new ConstantExpression("y"));
104: mapExpression.addMapEntryExpression(
105: new ConstantExpression("c"),
106: new ConstantExpression("z"));
107: assertIterate("iterateOverMap", mapExpression);
108: }
109:
110: protected void assertIterate(String methodName,
111: Expression listExpression) throws Exception {
112: ClassNode classNode = new ClassNode("Foo", ACC_PUBLIC,
113: ClassHelper.OBJECT_TYPE);
114: classNode.addConstructor(new ConstructorNode(ACC_PUBLIC, null));
115: classNode.addProperty(new PropertyNode("bar", ACC_PUBLIC,
116: ClassHelper.STRING_TYPE, classNode, null, null, null));
117:
118: Statement loopStatement = createPrintlnStatement(new VariableExpression(
119: "i"));
120:
121: BlockStatement block = new BlockStatement();
122: block
123: .addStatement(new ExpressionStatement(
124: new DeclarationExpression(
125: new VariableExpression("list"), Token
126: .newSymbol("=", 0, 0),
127: listExpression)));
128: block.addStatement(new ForStatement(new Parameter(
129: ClassHelper.DYNAMIC_TYPE, "i"), new VariableExpression(
130: "list"), loopStatement));
131: classNode.addMethod(new MethodNode(methodName, ACC_PUBLIC,
132: ClassHelper.VOID_TYPE, Parameter.EMPTY_ARRAY,
133: ClassNode.EMPTY_ARRAY, block));
134:
135: Class fooClass = loadClass(classNode);
136: assertTrue("Loaded a new class", fooClass != null);
137:
138: Object bean = fooClass.newInstance();
139: assertTrue("Managed to create bean", bean != null);
140:
141: System.out
142: .println("################ Now about to invoke method");
143:
144: try {
145: InvokerHelper.invokeMethod(bean, methodName, null);
146: } catch (InvokerInvocationException e) {
147: System.out.println("Caught: " + e.getCause());
148: e.getCause().printStackTrace();
149: fail("Should not have thrown an exception");
150: }
151: System.out.println("################ Done");
152: }
153:
154: }
|