private DataSetLookup getDataSetLookup(final DisplayerSettings settings) { return ConsoleDataSetLookup.fromInstance(settings.getDataSetLookup(), serverTemplateSelectorMenuBuilder.getSelectedServerTemplateId()); }
public static DataSetLookup fromInstance(DataSetLookup orig, String serverTemplateId) { ConsoleDataSetLookup clone = new ConsoleDataSetLookup(); clone.setDataSetUUID(orig.getDataSetUUID()); clone.setRowOffset(orig.getRowOffset()); clone.setNumberOfRows(orig.getNumberOfRows()); for (DataSetOp dataSetOp : orig.getOperationList()) { clone.getOperationList().add(dataSetOp.cloneInstance()); } clone.setServerTemplateId(serverTemplateId); return clone; }
@Override public DataSetLookup cloneInstance() { return fromInstance(super.cloneInstance(), getServerTemplateId()); } }
protected ConsoleDataSetLookup adoptLookup(DataSetDef def, DataSetLookup lookup) { ConsoleDataSetLookup dataSetLookup = null; if (!(lookup instanceof ConsoleDataSetLookup)) { if (def instanceof RemoteDataSetDef) { dataSetLookup = (ConsoleDataSetLookup) ConsoleDataSetLookup.fromInstance(lookup, ((RemoteDataSetDef) def).getServerTemplateId()); DataSetFilter filter = def.getDataSetFilter(); if (filter != null) { dataSetLookup.addOperation(filter); } } else { throw new IllegalArgumentException("DataSetLookup is of incorrect type " + lookup.getClass().getName()); } } else { dataSetLookup = (ConsoleDataSetLookup) lookup; } return dataSetLookup; }
@Test public void testPerformQueryTestMode() { QueryFilterSpec filterSpec = new QueryFilterSpec(); ConsoleDataSetLookup dataSetLookup = Mockito.mock(ConsoleDataSetLookup.class); when(dataSetLookup.testMode()).thenReturn(true); when(dataSetLookup.getNumberOfRows()).thenReturn(10); when(dataSetLookup.getRowOffset()).thenReturn(1); when(dataSetLookup.getDataSetUUID()).thenReturn(""); kieServerDataSetProvider.performQuery(dataSetDef, dataSetLookup, queryServicesClient, filterSpec); verify(kieServerIntegration).broadcastToKieServers(anyString(), any()); verify(dataSetLookup, times(1)).testMode(); verify(queryServicesClient).query(anyString(), anyString(), any(QueryFilterSpec.class), anyInt(), anyInt(), any()); }
final ConsoleDataSetLookup cdsl = (ConsoleDataSetLookup) captor.getValue(); assertEquals(serverTemplateId, cdsl.getServerTemplateId()); assertEquals(dataUUID, cdsl.getDataSetUUID()); verify(select, never()).addOption(anyString(),
dataSetLookup.getServerTemplateId()); if (dataSetLookup.getServerTemplateId() == null || dataSetLookup.getServerTemplateId().isEmpty()) { return buildDataSet(def, new ArrayList<>(), final QueryServicesClient queryClient = getClient(dataSetLookup.getServerTemplateId(), QueryServicesClient.class); for (DataSetFilter filter : dataSetLookup.getOperationList(DataSetFilter.class)) { if (filter != null) { DataSetGroup dataSetGroup = dataSetLookup.getLastGroupOp(); handleDataSetGroup(def, dataSetGroup, DataSetSort sort = dataSetLookup.getFirstSortOp(); if (sort != null) { SortOrder sortOrder = SortOrder.UNSPECIFIED;
@Test public void testAdoptLookup() throws Exception { DataSetLookup dataSetLookup = Mockito.mock(DataSetLookup.class); when(dataSetDef.getDataSetFilter()).thenReturn(Mockito.mock(DataSetFilter.class)); when(dataSetDef.getServerTemplateId()).thenReturn("servereTemplateId"); when(dataSetLookup.cloneInstance()).thenReturn(dataSetLookup); ConsoleDataSetLookup adopted = kieServerDataSetProvider.adoptLookup(dataSetDef, dataSetLookup); verify(dataSetDef, times(1)).getServerTemplateId(); assertNotNull(adopted.getOperationList()); assertEquals(1, adopted.getOperationList().size()); }
QueryFilterSpec filterSpec) { if (dataSetLookup.testMode()) { QueryDefinition queryDefinition = QueryDefinition.builder() .name(dataSetLookup.getDataSetUUID()) .source(def.getDataSource()) .target(def.getQueryTarget()) dataSetLookup.getDataSetUUID(), QueryServicesClient.QUERY_MAP_RAW, filterSpec, dataSetLookup.getRowOffset() / dataSetLookup.getNumberOfRows(), dataSetLookup.getNumberOfRows(), List.class ); } catch (Exception e) { queryClient.unregisterQuery(dataSetLookup.getDataSetUUID()); throw new RuntimeException(e); dataSetLookup.getDataSetUUID(), QueryServicesClient.QUERY_MAP_RAW, filterSpec, dataSetLookup.getRowOffset() / dataSetLookup.getNumberOfRows(), dataSetLookup.getNumberOfRows(), List.class );
final ConsoleDataSetLookup cdsl = (ConsoleDataSetLookup) captor.getValue(); assertEquals(serverTemplateId, cdsl.getServerTemplateId()); assertEquals(dataUUID, cdsl.getDataSetUUID()); verify(select, never()).addOption(anyString(),
@Test public void testAdoptLookup() throws Exception { DataSetLookup dataSetLookup = Mockito.mock(DataSetLookup.class); when(dataSetDef.getDataSetFilter()).thenReturn(Mockito.mock(DataSetFilter.class)); when(dataSetDef.getServerTemplateId()).thenReturn("servereTemplateId"); when(dataSetLookup.cloneInstance()).thenReturn(dataSetLookup); ConsoleDataSetLookup adopted = kieServerDataSetProvider.adoptLookup(dataSetDef, dataSetLookup); verify(dataSetDef, times(1)).getServerTemplateId(); assertNotNull(adopted.getOperationList()); assertEquals(1, adopted.getOperationList().size()); }
@Test public void testPerformQueryRegularMode() { QueryFilterSpec filterSpec = new QueryFilterSpec(); ConsoleDataSetLookup dataSetLookup = Mockito.mock(ConsoleDataSetLookup.class); when(dataSetLookup.testMode()).thenReturn(false); when(dataSetLookup.getNumberOfRows()).thenReturn(10); when(dataSetLookup.getRowOffset()).thenReturn(1); when(dataSetLookup.getDataSetUUID()).thenReturn(""); kieServerDataSetProvider.performQuery(dataSetDef, dataSetLookup, queryServicesClient, filterSpec); verify(kieServerIntegration, times(0)).broadcastToKieServers(anyString(), any()); verify(dataSetLookup, times(1)).testMode(); verify(queryServicesClient).query(anyString(), anyString(), any(QueryFilterSpec.class), anyInt(), anyInt(), any()); }
public void setDataSetHandler(FilterSettings tableSettings) { this.dataSetHandler = new DataSetHandlerImpl(dataSetClientServices, ConsoleDataSetLookup.fromInstance(tableSettings.getDataSetLookup(), tableSettings.getServerTemplateId())); }
final ConsoleDataSetLookup cdsl = (ConsoleDataSetLookup) captor.getValue(); assertEquals(serverTemplateId, cdsl.getServerTemplateId()); assertEquals(dataUUID, cdsl.getDataSetUUID()); verify(select).addOption(columnText, columnValue);
@Override public DataSetLookup cloneInstance() { return fromInstance(super.cloneInstance(), getServerTemplateId()); } }
public static DataSetLookup fromInstance(DataSetLookup orig, String serverTemplateId) { ConsoleDataSetLookup clone = new ConsoleDataSetLookup(); clone.setDataSetUUID(orig.getDataSetUUID()); clone.setRowOffset(orig.getRowOffset()); clone.setNumberOfRows(orig.getNumberOfRows()); for (DataSetOp dataSetOp : orig.getOperationList()) { clone.getOperationList().add(dataSetOp.cloneInstance()); } clone.setServerTemplateId(serverTemplateId); return clone; }
@Test public void testPerformQueryRegularMode() { QueryFilterSpec filterSpec = new QueryFilterSpec(); ConsoleDataSetLookup dataSetLookup = Mockito.mock(ConsoleDataSetLookup.class); when(dataSetLookup.testMode()).thenReturn(false); when(dataSetLookup.getNumberOfRows()).thenReturn(10); when(dataSetLookup.getRowOffset()).thenReturn(1); when(dataSetLookup.getDataSetUUID()).thenReturn(""); kieServerDataSetProvider.performQuery(dataSetDef, dataSetLookup, queryServicesClient, filterSpec); verify(kieServerIntegration, times(0)).broadcastToKieServers(anyString(), any()); verify(dataSetLookup, times(1)).testMode(); verify(queryServicesClient).query(anyString(), anyString(), any(QueryFilterSpec.class), anyInt(), anyInt(), any()); }
public void setDataSetHandler(FilterSettings tableSettings) { this.dataSetHandler = new DataSetHandlerImpl(dataSetClientServices, ConsoleDataSetLookup.fromInstance(tableSettings.getDataSetLookup(), tableSettings.getServerTemplateId())); }
final ConsoleDataSetLookup cdsl = (ConsoleDataSetLookup) captor.getValue(); assertEquals(serverTemplateId, cdsl.getServerTemplateId()); assertEquals(dataUUID, cdsl.getDataSetUUID()); verify(select).addOption(columnText, columnValue);
@Test public void testPerformQueryTestMode() { QueryFilterSpec filterSpec = new QueryFilterSpec(); ConsoleDataSetLookup dataSetLookup = Mockito.mock(ConsoleDataSetLookup.class); when(dataSetLookup.testMode()).thenReturn(true); when(dataSetLookup.getNumberOfRows()).thenReturn(10); when(dataSetLookup.getRowOffset()).thenReturn(1); when(dataSetLookup.getDataSetUUID()).thenReturn(""); kieServerDataSetProvider.performQuery(dataSetDef, dataSetLookup, queryServicesClient, filterSpec); verify(kieServerIntegration).broadcastToKieServers(anyString(), any()); verify(dataSetLookup, times(1)).testMode(); verify(queryServicesClient).query(anyString(), anyString(), any(QueryFilterSpec.class), anyInt(), anyInt(), any()); }