use of org.pentaho.di.trans.Trans in project pentaho-kettle by pentaho.
the class UpdateMetaTest method setUp.
@Before
public void setUp() throws KettleException {
KettleEnvironment.init();
PluginRegistry.init(false);
TransMeta transMeta = new TransMeta();
transMeta.setName("delete1");
Map<String, String> vars = new HashMap<String, String>();
vars.put("max.sz", "10");
transMeta.injectVariables(vars);
umi = new UpdateMeta();
ud = new UpdateData();
PluginRegistry plugReg = PluginRegistry.getInstance();
String deletePid = plugReg.getPluginId(StepPluginType.class, umi);
stepMeta = new StepMeta(deletePid, "delete", umi);
Trans trans = new Trans(transMeta);
transMeta.addStep(stepMeta);
mockHelper = new StepMockHelper<>("Update", UpdateMeta.class, UpdateData.class);
Mockito.when(mockHelper.logChannelInterfaceFactory.create(Mockito.any(), Mockito.any(LoggingObjectInterface.class))).thenReturn(mockHelper.logChannelInterface);
upd = new Update(stepMeta, ud, 1, transMeta, trans);
upd.copyVariablesFrom(transMeta);
List<String> attributes = Arrays.asList("schemaName", "tableName", "commitSize", "errorIgnored", "ignoreFlagField", "skipLookup", "useBatchUpdate", "keyStream", "keyLookup", "keyCondition", "keyStream2", "updateLookup", "updateStream", "databaseMeta");
Map<String, String> getterMap = new HashMap<String, String>() {
{
put("schemaName", "getSchemaName");
put("tableName", "getTableName");
put("commitSize", "getCommitSizeVar");
put("errorIgnored", "isErrorIgnored");
put("ignoreFlagField", "getIgnoreFlagField");
put("skipLookup", "isSkipLookup");
put("useBatchUpdate", "useBatchUpdate");
put("keyStream", "getKeyStream");
put("keyLookup", "getKeyLookup");
put("keyCondition", "getKeyCondition");
put("keyStream2", "getKeyStream2");
put("updateLookup", "getUpdateLookup");
put("updateStream", "getUpdateStream");
put("databaseMeta", "getDatabaseMeta");
}
};
Map<String, String> setterMap = new HashMap<String, String>() {
{
put("schemaName", "setSchemaName");
put("tableName", "setTableName");
put("commitSize", "setCommitSize");
put("errorIgnored", "setErrorIgnored");
put("ignoreFlagField", "setIgnoreFlagField");
put("skipLookup", "setSkipLookup");
put("useBatchUpdate", "setUseBatchUpdate");
put("keyStream", "setKeyStream");
put("keyLookup", "setKeyLookup");
put("keyCondition", "setKeyCondition");
put("keyStream2", "setKeyStream2");
put("updateLookup", "setUpdateLookup");
put("updateStream", "setUpdateStream");
put("databaseMeta", "setDatabaseMeta");
}
};
FieldLoadSaveValidator<String[]> stringArrayLoadSaveValidator = new ArrayLoadSaveValidator<String>(new StringLoadSaveValidator(), 5);
Map<String, FieldLoadSaveValidator<?>> attrValidatorMap = new HashMap<String, FieldLoadSaveValidator<?>>();
attrValidatorMap.put("keyStream", stringArrayLoadSaveValidator);
attrValidatorMap.put("keyLookup", stringArrayLoadSaveValidator);
attrValidatorMap.put("keyCondition", stringArrayLoadSaveValidator);
attrValidatorMap.put("keyStream2", stringArrayLoadSaveValidator);
attrValidatorMap.put("updateLookup", stringArrayLoadSaveValidator);
attrValidatorMap.put("updateStream", stringArrayLoadSaveValidator);
attrValidatorMap.put("databaseMeta", new DatabaseMetaLoadSaveValidator());
Map<String, FieldLoadSaveValidator<?>> typeValidatorMap = new HashMap<String, FieldLoadSaveValidator<?>>();
loadSaveTester = new LoadSaveTester(testMetaClass, attributes, new ArrayList<String>(), new ArrayList<String>(), getterMap, setterMap, attrValidatorMap, typeValidatorMap, this);
}
use of org.pentaho.di.trans.Trans in project pentaho-kettle by pentaho.
the class ZipFileMetaTest method testGetStep.
@Test
public void testGetStep() throws Exception {
StepMeta stepInfo = mock(StepMeta.class);
when(stepInfo.getName()).thenReturn("Zip Step Name");
StepDataInterface stepData = mock(StepDataInterface.class);
TransMeta transMeta = mock(TransMeta.class);
when(transMeta.findStep("Zip Step Name")).thenReturn(stepInfo);
Trans trans = mock(Trans.class);
ZipFileMeta zipFileMeta = new ZipFileMeta();
ZipFile zipFile = (ZipFile) zipFileMeta.getStep(stepInfo, stepData, 0, transMeta, trans);
assertEquals(stepInfo, zipFile.getStepMeta());
assertEquals(stepData, zipFile.getStepDataInterface());
assertEquals(transMeta, zipFile.getTransMeta());
assertEquals(trans, zipFile.getTrans());
assertEquals(0, zipFile.getCopy());
}
use of org.pentaho.di.trans.Trans in project pentaho-kettle by pentaho.
the class CleanupTransServletTest method testCleanupTransServletEscapesHtmlWhenTransFound.
@Test
@PrepareForTest({ Encode.class })
public void testCleanupTransServletEscapesHtmlWhenTransFound() throws ServletException, IOException {
HttpServletRequest mockHttpServletRequest = mock(HttpServletRequest.class);
HttpServletResponse mockHttpServletResponse = mock(HttpServletResponse.class);
Trans mockTrans = mock(Trans.class);
StringWriter out = new StringWriter();
PrintWriter printWriter = new PrintWriter(out);
PowerMockito.spy(Encode.class);
when(mockHttpServletRequest.getContextPath()).thenReturn(CleanupTransServlet.CONTEXT_PATH);
when(mockHttpServletRequest.getParameter(anyString())).thenReturn(ServletTestUtils.BAD_STRING_TO_TEST);
when(mockHttpServletResponse.getWriter()).thenReturn(printWriter);
when(mockTransformationMap.getTransformation(any(CarteObjectEntry.class))).thenReturn(mockTrans);
cleanupTransServlet.doGet(mockHttpServletRequest, mockHttpServletResponse);
assertFalse(ServletTestUtils.hasBadText(ServletTestUtils.getInsideOfTag("H1", out.toString())));
PowerMockito.verifyStatic(atLeastOnce());
Encode.forHtml(anyString());
}
use of org.pentaho.di.trans.Trans in project pentaho-kettle by pentaho.
the class PrepareExecutionTransServletTest method testPauseTransServletEscapesHtmlWhenTransFound.
@Test
@PrepareForTest({ Encode.class })
public void testPauseTransServletEscapesHtmlWhenTransFound() throws ServletException, IOException {
KettleLogStore.init();
HttpServletRequest mockHttpServletRequest = mock(HttpServletRequest.class);
HttpServletResponse mockHttpServletResponse = mock(HttpServletResponse.class);
Trans mockTrans = mock(Trans.class);
TransConfiguration mockTransConf = mock(TransConfiguration.class);
TransMeta mockTransMeta = mock(TransMeta.class);
TransExecutionConfiguration mockTransExecutionConf = mock(TransExecutionConfiguration.class);
LogChannelInterface mockChannelInterface = mock(LogChannelInterface.class);
StringWriter out = new StringWriter();
PrintWriter printWriter = new PrintWriter(out);
PowerMockito.spy(Encode.class);
when(mockHttpServletRequest.getContextPath()).thenReturn(PrepareExecutionTransServlet.CONTEXT_PATH);
when(mockHttpServletRequest.getParameter(anyString())).thenReturn(ServletTestUtils.BAD_STRING_TO_TEST);
when(mockHttpServletResponse.getWriter()).thenReturn(printWriter);
when(mockTransformationMap.getTransformation(any(CarteObjectEntry.class))).thenReturn(mockTrans);
when(mockTransformationMap.getConfiguration(any(CarteObjectEntry.class))).thenReturn(mockTransConf);
when(mockTransConf.getTransExecutionConfiguration()).thenReturn(mockTransExecutionConf);
when(mockTrans.getLogChannel()).thenReturn(mockChannelInterface);
when(mockTrans.getTransMeta()).thenReturn(mockTransMeta);
when(mockTransMeta.getMaximum()).thenReturn(new Point(10, 10));
prepareExecutionTransServlet.doGet(mockHttpServletRequest, mockHttpServletResponse);
assertFalse(ServletTestUtils.hasBadText(ServletTestUtils.getInsideOfTag("H1", out.toString())));
PowerMockito.verifyStatic(atLeastOnce());
Encode.forHtml(anyString());
}
use of org.pentaho.di.trans.Trans in project pentaho-kettle by pentaho.
the class PanTest method testConfigureParameters.
@Test
public void testConfigureParameters() throws Exception {
TransMeta transMeta = new TransMeta();
transMeta.addParameterDefinition(TEST_PARAM_NAME, DEFAULT_PARAM_VALUE, "This tests a default parameter");
assertEquals("Default parameter was not set correctly on TransMeta", DEFAULT_PARAM_VALUE, transMeta.getParameterDefault(TEST_PARAM_NAME));
assertEquals("Parameter value should be blank in TransMeta", "", transMeta.getParameterValue(TEST_PARAM_NAME));
Trans trans = new Trans(transMeta);
assertEquals("Default parameter was not set correctly on Trans", DEFAULT_PARAM_VALUE, trans.getParameterDefault(TEST_PARAM_NAME));
assertEquals("Parameter value should be blank in Trans", "", trans.getParameterValue(TEST_PARAM_NAME));
NamedParams params = new NamedParamsDefault();
params.addParameterDefinition(TEST_PARAM_NAME, NOT_DEFAULT_PARAM_VALUE, "This tests a non-default parameter");
params.setParameterValue(TEST_PARAM_NAME, NOT_DEFAULT_PARAM_VALUE);
Pan.configureParameters(trans, params, transMeta);
assertEquals("Parameter was not set correctly in Trans", NOT_DEFAULT_PARAM_VALUE, trans.getParameterValue(TEST_PARAM_NAME));
assertEquals("Parameter was not set correctly in TransMeta", NOT_DEFAULT_PARAM_VALUE, transMeta.getParameterValue(TEST_PARAM_NAME));
}
Aggregations