Both getInputRowSets and findInputRowSet method should behave the same
@@ -32,6 +32,7 @@ import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.runners.MockitoJUnitRunner;
+import java.util.ArrayList;
import java.util.List;
import static org.mockito.Matchers.any;
@@ -99,6 +100,10 @@ public class XmlJoinOmitNullValuesTest {
when(inputStreamTarget.getTransformName()).thenReturn("target");
when(inputStreamSource.getTransformName()).thenReturn("source");
+ List<IRowSet> inputRowSets = new ArrayList<>();
+ inputRowSets.add(sourceRowSet);
+ inputRowSets.add(targetRowSet);
+ spy.setInputRowSets(inputRowSets);
doReturn(sourceRowSet).when(spy).findInputRowSet("source");
doReturn(targetRowSet).when(spy).findInputRowSet("target");
