/*! ****************************************************************************** * * Pentaho Data Integration * * Copyright (C) 2002-2016 by Pentaho : http://www.pentaho.com * ******************************************************************************* * * Licensed 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. * ******************************************************************************/ package org.pentaho.di.trans.steps.filterrows; import java.util.Arrays; import java.util.HashMap; import java.util.List; import java.util.Map; import com.google.common.collect.ImmutableList; import org.junit.Before; import org.junit.Test; import static org.junit.Assert.*; import org.pentaho.di.core.Condition; import org.pentaho.di.core.KettleEnvironment; import org.pentaho.di.core.exception.KettleException; import org.pentaho.di.core.plugins.PluginRegistry; import org.pentaho.di.trans.step.StepMeta; import org.pentaho.di.trans.steps.dummytrans.DummyTransMeta; import org.pentaho.di.trans.steps.loadsave.LoadSaveTester; import org.pentaho.di.trans.steps.loadsave.validator.ConditionLoadSaveValidator; import org.pentaho.di.trans.steps.loadsave.validator.FieldLoadSaveValidator; import org.pentaho.di.trans.steps.loadsave.validator.StringLoadSaveValidator; public class FilterRowsMetaTest { LoadSaveTester loadSaveTester; Class<FilterRowsMeta> testMetaClass = FilterRowsMeta.class; @Before public void setUpLoadSave() throws Exception { KettleEnvironment.init(); PluginRegistry.init( true ); List<String> attributes = Arrays.asList( "condition", "send_true_to", "send_false_to" ); Map<String, String> getterMap = new HashMap<String, String>(); Map<String, String> setterMap = new HashMap<String, String>(); Map<String, FieldLoadSaveValidator<?>> attrValidatorMap = new HashMap<String, FieldLoadSaveValidator<?>>(); attrValidatorMap.put( "condition", new ConditionLoadSaveValidator() ); attrValidatorMap.put( "trueStepName", new StringLoadSaveValidator() ); attrValidatorMap.put( "falseStepname", new StringLoadSaveValidator() ); getterMap.put( "send_true_to", "getTrueStepname" ); setterMap.put( "send_true_to", "setTrueStepname" ); getterMap.put( "send_false_to", "getFalseStepname" ); setterMap.put( "send_false_to", "setFalseStepname" ); Map<String, FieldLoadSaveValidator<?>> typeValidatorMap = new HashMap<String, FieldLoadSaveValidator<?>>(); loadSaveTester = new LoadSaveTester( testMetaClass, attributes, getterMap, setterMap, attrValidatorMap, typeValidatorMap ); } @Test public void testSerialization() throws KettleException { loadSaveTester.testSerialization(); } @Test public void testClone() { FilterRowsMeta filterRowsMeta = new FilterRowsMeta(); filterRowsMeta.setCondition( new Condition() ); filterRowsMeta.setTrueStepname( "true" ); filterRowsMeta.setFalseStepname( "false" ); FilterRowsMeta clone = (FilterRowsMeta) filterRowsMeta.clone(); assertNotNull( clone.getCondition() ); assertEquals( "true", clone.getTrueStepname() ); assertEquals( "false", clone.getFalseStepname() ); } @Test public void modifiedTarget() throws Exception { FilterRowsMeta filterRowsMeta = new FilterRowsMeta(); StepMeta trueOutput = new StepMeta( "true", new DummyTransMeta() ); StepMeta falseOutput = new StepMeta( "false", new DummyTransMeta() ); filterRowsMeta.setCondition( new Condition() ); filterRowsMeta.setTrueStepname( trueOutput.getName() ); filterRowsMeta.setFalseStepname( falseOutput.getName() ); filterRowsMeta.searchInfoAndTargetSteps( ImmutableList.of( trueOutput, falseOutput ) ); trueOutput.setName( "true renamed" ); falseOutput.setName( "false renamed" ); assertEquals( "true renamed", filterRowsMeta.getTrueStepname() ); assertEquals( "false renamed", filterRowsMeta.getFalseStepname() ); } }