How to use

getDefinition

method

in

org.dashbuilder.dataset.DataSet

protected boolean isRemoteProvider(DataSet dataSet) {
  final DataSetDef def = dataSet.getDefinition();
  return def == null || def.getProvider() == null ? false : def.getProvider().getName().equals("REMOTE");
}

protected boolean isRemoteProvider(DataSet dataSet) {
  final DataSetDef def = dataSet.getDefinition();
  return def == null || def.getProvider() == null ? false : def.getProvider().getName().equals("REMOTE");
}

  public static void assertDataSetDefinition(final DataSet dataSet, final String uuid) {
    assertThat(dataSet.getUUID()).isEqualTo(uuid);
    assertThat(dataSet.getDefinition()).isNotNull();
    assertThat(dataSet.getDefinition().getUUID()).isEqualTo(uuid);
  }
}

@Override
public void callback(final DataSet dataSet) {
  getDataSetDef().setColumns(dataSet.getDefinition().getColumns());
  testDataSetCallback.onSuccess(dataSet);
}

  public static void assertDataSetDefinition(final DataSet dataSet, final String uuid) {
    assertThat(dataSet.getUUID()).isEqualTo(uuid);
    assertThat(dataSet.getDefinition()).isNotNull();
    assertThat(dataSet.getDefinition().getUUID()).isEqualTo(uuid);
  }
}

@Test
public void testIsPagerEnabledRemote() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(() -> "REMOTE");
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertFalse(table.isPagerEnabled(dataSet,
                   5));
  assertTrue(table.isPagerEnabled(dataSet,
                  10));
  verify(dataSet,
      never()).getRowCountNonTrimmed();
}

@Test
public void testIsPagerEnabledRemote() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(() -> "REMOTE");
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertFalse(table.isPagerEnabled(dataSet,
                   5));
  assertTrue(table.isPagerEnabled(dataSet,
                  10));
  verify(dataSet,
      never()).getRowCountNonTrimmed();
}

@Test
public void testIsPagerEnabledSQL() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(DataSetProviderType.SQL);
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  when(dataSet.getRowCountNonTrimmed()).thenReturn(5,
                           10,
                           15);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertFalse(table.isPagerEnabled(dataSet,
                   5));
  assertFalse(table.isPagerEnabled(dataSet,
                   10));
  assertTrue(table.isPagerEnabled(dataSet,
                  15));
}

@Test
public void testIsPagerEnabledSQL() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(DataSetProviderType.SQL);
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  when(dataSet.getRowCountNonTrimmed()).thenReturn(5,
                           10,
                           15);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertFalse(table.isPagerEnabled(dataSet,
                   5));
  assertFalse(table.isPagerEnabled(dataSet,
                   10));
  assertTrue(table.isPagerEnabled(dataSet,
                  15));
}

@Test
public void testIsTotalRowsExactSQL() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(DataSetProviderType.SQL);
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertTrue(table.isTotalRowsExact(dataSet,
                   5));
  assertTrue(table.isTotalRowsExact(dataSet,
                   10));
  assertTrue(table.isTotalRowsExact(dataSet,
                   15));
}

@Test
public void testIsTotalRowsExactRemote() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(() -> "REMOTE");
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertTrue(table.isTotalRowsExact(dataSet,
                   5));
  assertFalse(table.isTotalRowsExact(dataSet,
                    10));
  assertFalse(table.isTotalRowsExact(dataSet,
                    15));
}

@Test
public void testIsTotalRowsExactRemote() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(() -> "REMOTE");
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertTrue(table.isTotalRowsExact(dataSet,
                   5));
  assertFalse(table.isTotalRowsExact(dataSet,
                    10));
  assertFalse(table.isTotalRowsExact(dataSet,
                    15));
}

@Test
public void testIsTotalRowsExactSQL() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDef = new DataSetDef();
  dataSetDef.setProvider(DataSetProviderType.SQL);
  when(dataSet.getDefinition()).thenReturn(dataSetDef);
  TableDisplayer table = createTableDisplayer(DisplayerSettingsFactory.newTableSettings().tablePageSize(10).buildSettings());
  assertTrue(table.isTotalRowsExact(dataSet,
                   5));
  assertTrue(table.isTotalRowsExact(dataSet,
                   10));
  assertTrue(table.isTotalRowsExact(dataSet,
                   15));
}

@Test
public void testIsRemoteProvider() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDefSQL = new DataSetDef();
  dataSetDefSQL.setProvider(DataSetProviderType.SQL);
  final DataSetDef dataSetDefRemote = new DataSetDef();
  dataSetDefRemote.setProvider(() -> "REMOTE");
  when(dataSet.getDefinition()).thenReturn(new DataSetDef(),
                       dataSetDefSQL,
                       dataSetDefRemote);
  TableDisplayer table = createTableDisplayer(null);
  //No provider set
  assertFalse(table.isRemoteProvider(dataSet));
  //Provider SQL
  assertFalse(table.isRemoteProvider(dataSet));
  //Provider REMOTE
  assertTrue(table.isRemoteProvider(dataSet));
}

@Test
public void testIsRemoteProvider() {
  final DataSet dataSet = mock(DataSet.class);
  final DataSetDef dataSetDefSQL = new DataSetDef();
  dataSetDefSQL.setProvider(DataSetProviderType.SQL);
  final DataSetDef dataSetDefRemote = new DataSetDef();
  dataSetDefRemote.setProvider(() -> "REMOTE");
  when(dataSet.getDefinition()).thenReturn(new DataSetDef(),
                       dataSetDefSQL,
                       dataSetDefRemote);
  TableDisplayer table = createTableDisplayer(null);
  //No provider set
  assertFalse(table.isRemoteProvider(dataSet));
  //Provider SQL
  assertFalse(table.isRemoteProvider(dataSet));
  //Provider REMOTE
  assertTrue(table.isRemoteProvider(dataSet));
}