@Before
public void setUp() throws Exception {
when(spoon.getActiveTransformation()).thenReturn(transMeta);
when(spoon.getShell()).thenReturn(shell);
when(uiFactory.getMessageBox(any(Shell.class), anyInt())).thenReturn(messageBox);
when(uiFactory.getShell(shell)).thenReturn(shell);
when(uiFactory.getDataServiceTestDialog(
any(Shell.class), any(DataServiceMeta.class), any(DataServiceContext.class)))
.thenReturn(dataServiceTestDialog);
when(uiFactory.getMessageDialog(
any(Shell.class),
anyString(),
any(Image.class),
anyString(),
anyInt(),
any(String[].class),
anyInt()))
.thenReturn(messageDialog);
when(messageDialog.open()).thenReturn(0);
delegate = new DataServiceDelegate(context, spoon);
when(uiFactory.getDataServiceDialogBuilder(transMeta)).thenReturn(dialogBuilder);
when(dialogBuilder.serviceStep(STEP_NAME)).thenReturn(dialogBuilder);
when(dialogBuilder.edit(dataService)).thenReturn(dialogBuilder);
when(dialogBuilder.build(delegate)).thenReturn(dataServiceDialog);
}
@Test
public void testCreateNewDataServiceNoServiceStepProvided() throws KettleException {
delegate = spy(delegate);
when(dialogBuilder.serviceStep(anyString())).thenReturn(dialogBuilder);
when(dialogBuilder.build(delegate)).thenReturn(dataServiceDialog);
delegate.createNewDataService(null);
verify(delegate).getLastStepOfActiveTrans();
}
