MockConnector.java

package org.teiid.connector.metadata.runtime;

import java.util.Properties;

import junit.framework.TestCase;

import org.mockito.Mockito;
import org.teiid.language.Call;
import org.teiid.language.ColumnReference;
import org.teiid.language.DerivedColumn;
import org.teiid.language.NamedTable;
import org.teiid.language.QueryExpression;
import org.teiid.metadata.AbstractMetadataRecord;
import org.teiid.metadata.Column;
import org.teiid.metadata.RuntimeMetadata;
import org.teiid.metadata.BaseColumn.NullType;
import org.teiid.metadata.Column.SearchType;
import org.teiid.translator.ExecutionContext;
import org.teiid.translator.ExecutionFactory;
import org.teiid.translator.ProcedureExecution;
import org.teiid.translator.ResultSetExecution;
import org.teiid.translator.TranslatorException;


public class MockConnector extends ExecutionFactory<Object, Object> {
	
	
	@Override
	public ProcedureExecution createProcedureExecution(
			Call procedure, ExecutionContext executionContext,
			RuntimeMetadata metadata, Object connection) throws TranslatorException {
		Properties props = new Properties();
		props.setProperty("customBehaviour", "SkipExecute");//$NON-NLS-1$ //$NON-NLS-2$
	
        AbstractMetadataRecord metaObject = procedure.getMetadataObject();
        
        TestCase.assertEquals("AnyModel.ProcedureB",procedure.getProcedureName()); //$NON-NLS-1$
        TestCase.assertEquals("PROC", metaObject.getNameInSource()); //$NON-NLS-1$
        TestCase.assertEquals(props, metaObject.getProperties());
        ProcedureExecution exec = Mockito.mock(ProcedureExecution.class);
        Mockito.stub(exec.next()).toReturn(null);
        return exec;
	}
	
	@Override
	public ResultSetExecution createResultSetExecution(
			QueryExpression query, ExecutionContext executionContext,
			RuntimeMetadata metadata, Object connection) throws TranslatorException {
		Properties groupProps = new Properties();
		groupProps.setProperty("customName", "CustomTableA");//$NON-NLS-1$ //$NON-NLS-2$
		NamedTable group = (NamedTable)query.getProjectedQuery().getFrom().get(0);			
		AbstractMetadataRecord groupMD = group.getMetadataObject();
		TestCase.assertEquals(groupProps, groupMD.getProperties());
		
		
		DerivedColumn symbl = query.getProjectedQuery().getDerivedColumns().get(0);
		ColumnReference element = (ColumnReference)symbl.getExpression();
		Column elementMD = element.getMetadataObject();

		Properties elementProps = new Properties();
		elementProps.setProperty("customPosition", "11");//$NON-NLS-1$ //$NON-NLS-2$
		
		TestCase.assertEquals(0, elementMD.getLength());
		TestCase.assertEquals("Foo", elementMD.getDefaultValue()); //$NON-NLS-1$
		TestCase.assertEquals("TrimNulls", elementMD.getFormat()); //$NON-NLS-1$
		TestCase.assertEquals(String.class, elementMD.getJavaType()); 
		TestCase.assertEquals(null, elementMD.getMaximumValue());
		TestCase.assertEquals(null, elementMD.getMinimumValue());
		TestCase.assertEquals("COLUMN1", elementMD.getNameInSource()); //$NON-NLS-1$
		TestCase.assertEquals("STR", elementMD.getNativeType()); //$NON-NLS-1$
		TestCase.assertEquals(NullType.Nullable, elementMD.getNullType());
		TestCase.assertEquals(0, elementMD.getPosition());
		TestCase.assertEquals(0, elementMD.getPrecision());
		TestCase.assertEquals(0, elementMD.getScale());
		TestCase.assertEquals(SearchType.Searchable, elementMD.getSearchType());
		TestCase.assertEquals(false, elementMD.isAutoIncremented());
		TestCase.assertEquals(true, elementMD.isCaseSensitive());
		TestCase.assertEquals(elementProps, elementMD.getProperties());
		
		
		DerivedColumn symbl2 = query.getProjectedQuery().getDerivedColumns().get(1);
		ColumnReference element2 = (ColumnReference)symbl2.getExpression();
		Column elementMD2 = element2.getMetadataObject();

		Properties elementProps2 = new Properties();
		elementProps2.setProperty("customPosition", "12");//$NON-NLS-1$ //$NON-NLS-2$
		
		TestCase.assertEquals(10, elementMD2.getLength());
		TestCase.assertEquals("23", elementMD2.getDefaultValue()); //$NON-NLS-1$
		TestCase.assertEquals("YesFormat", elementMD2.getFormat()); //$NON-NLS-1$
		TestCase.assertEquals(Integer.class, elementMD2.getJavaType());
		TestCase.assertEquals("1", elementMD2.getMaximumValue()); //$NON-NLS-1$
		TestCase.assertEquals("100", elementMD2.getMinimumValue()); //$NON-NLS-1$
		TestCase.assertEquals("COLUMN2", elementMD2.getNameInSource()); //$NON-NLS-1$
		TestCase.assertEquals("INT", elementMD2.getNativeType()); //$NON-NLS-1$
		TestCase.assertEquals(NullType.No_Nulls, elementMD2.getNullType());
		TestCase.assertEquals(1, elementMD2.getPosition());
		TestCase.assertEquals(0, elementMD2.getPrecision());
		TestCase.assertEquals(10, elementMD2.getScale());
		TestCase.assertEquals(SearchType.Searchable, elementMD2.getSearchType());
		TestCase.assertEquals(true, elementMD2.isAutoIncremented());
		TestCase.assertEquals(false, elementMD2.isCaseSensitive());
		
		TestCase.assertEquals(elementProps2, elementMD2.getProperties());
		ResultSetExecution exec = Mockito.mock(ResultSetExecution.class);
        Mockito.stub(exec.next()).toReturn(null);
        return exec;
	}	
}