/*
* JBoss, Home of Professional Open Source.
* See the COPYRIGHT.txt file distributed with this work for information
* regarding copyright ownership. Some portions may be licensed
* to Red Hat, Inc. under one or more contributor license agreements.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
* 02110-1301 USA.
*/
package org.teiid.connector.visitor.util;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import junit.framework.TestCase;
import org.teiid.language.ColumnReference;
import org.teiid.language.Comparison;
import org.teiid.language.Expression;
import org.teiid.language.Function;
import org.teiid.language.LanguageObject;
import org.teiid.language.NamedTable;
import org.teiid.language.Select;
import org.teiid.language.Comparison.Operator;
import org.teiid.language.visitor.CollectorVisitor;
/**
*/
public class TestCollectorVisitor extends TestCase {
/**
* Constructor for TestElementCollectorVisitor.
* @param name
*/
public TestCollectorVisitor(String name) {
super(name);
}
public Set getStringSet(Collection objs) {
Set strings = new HashSet();
Iterator iter = objs.iterator();
while(iter.hasNext()) {
Object obj = iter.next();
if(obj == null) {
strings.add(null);
} else {
strings.add(obj.toString());
}
}
return strings;
}
public void helpTestCollection(LanguageObject obj, Class type, String[] objects) {
Set actualObjects = getStringSet(CollectorVisitor.collectObjects(type, obj));
Set expectedObjects = new HashSet(Arrays.asList(objects));
assertEquals("Did not get expected objects", expectedObjects, actualObjects); //$NON-NLS-1$
}
public LanguageObject example1() {
NamedTable g = new NamedTable("g1", null, null); //$NON-NLS-1$
List symbols = new ArrayList();
symbols.add(new ColumnReference(g, "e1", null, String.class)); //$NON-NLS-1$
Function function = new Function("length", Arrays.asList(new ColumnReference(g, "e2", null, String.class)), Integer.class); //$NON-NLS-1$ //$NON-NLS-2$
symbols.add(function);
List groups = new ArrayList();
groups.add(g);
Select q = new Select(symbols, false, groups, null, null, null, null);
return q;
}
public void testCollection1() {
helpTestCollection(example1(), ColumnReference.class, new String[] {"g1.e1", "g1.e2" }); //$NON-NLS-1$ //$NON-NLS-2$
}
public void testCollection2() {
helpTestCollection(example1(), Function.class, new String[] {"length(g1.e2)" }); //$NON-NLS-1$
}
public void testCollection3() {
helpTestCollection(example1(), Expression.class, new String[] {"g1.e1", "g1.e2", "length(g1.e2)" }); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
}
public void helpTestElementsUsedByGroups(LanguageObject obj, String[] elements, String[] groups) {
Set actualElements = getStringSet(CollectorVisitor.collectElements(obj));
Set actualGroups = getStringSet(CollectorVisitor.collectGroupsUsedByElements(obj));
Set expectedElements = new HashSet(Arrays.asList(elements));
Set expectedGroups = new HashSet(Arrays.asList(groups));
assertEquals("Did not get expected elements", expectedElements, actualElements); //$NON-NLS-1$
assertEquals("Did not get expected groups", expectedGroups, actualGroups); //$NON-NLS-1$
}
public void test1() {
NamedTable g1 = new NamedTable("g1", null, null); //$NON-NLS-1$
ColumnReference e1 = new ColumnReference(g1, "e1", null, String.class); //$NON-NLS-1$
helpTestElementsUsedByGroups(e1, new String[] {"g1.e1"}, new String[] {"g1"}); //$NON-NLS-1$ //$NON-NLS-2$
}
public void test2() {
NamedTable g1 = new NamedTable("g1", null, null); //$NON-NLS-1$
ColumnReference e1 = new ColumnReference(g1, "e1", null, String.class); //$NON-NLS-1$
ColumnReference e2 = new ColumnReference(g1, "e2", null, String.class); //$NON-NLS-1$
Comparison cc = new Comparison(e1, e2, Operator.EQ);
helpTestElementsUsedByGroups(cc, new String[] {"g1.e1", "g1.e2"}, new String[] {"g1"}); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
}
}