private TableDescriptor createDummyTableDescriptor(String tableId) { BaseTableDescriptor tableDescriptor = mock(BaseTableDescriptor.class); when(tableDescriptor.getTableId()).thenReturn(tableId); return tableDescriptor; }
@Override public TaskApplicationDescriptor withTable(TableDescriptor tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return this; }
@Override public TaskApplicationDescriptor withTable(TableDescriptor tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return this; }
@Override public TaskApplicationDescriptor withTable(TableDescriptor tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return this; }
@Override public TaskApplicationDescriptor withTable(TableDescriptor tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return this; }
@Override public <K, V> Table<KV<K, V>> getTable(TableDescriptor<K, V, ?> tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return new TableImpl(baseTableDescriptor.getTableSpec()); }
@Override public <K, V> Table<KV<K, V>> getTable(TableDescriptor<K, V, ?> tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return new TableImpl(baseTableDescriptor.getTableSpec()); }
@Override public <K, V> Table<KV<K, V>> getTable(TableDescriptor<K, V, ?> tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return new TableImpl(baseTableDescriptor.getTableSpec()); }
@Override public <K, V> Table<KV<K, V>> getTable(TableDescriptor<K, V, ?> tableDescriptor) { addTableDescriptor(tableDescriptor); BaseTableDescriptor baseTableDescriptor = (BaseTableDescriptor) tableDescriptor; getOrCreateTableSerdes(baseTableDescriptor.getTableId(), baseTableDescriptor.getSerde()); return new TableImpl(baseTableDescriptor.getTableSpec()); }
@Test(expected = IllegalStateException.class) public void testGetTableWithBadId() { Config mockConfig = getConfig(); new StreamApplicationDescriptorImpl(appDesc -> { BaseTableDescriptor mockTableDescriptor = mock(BaseTableDescriptor.class); when(mockTableDescriptor.getTableId()).thenReturn("my.table"); appDesc.getTable(mockTableDescriptor); }, mockConfig); }
@Test public void testGetTable() throws Exception { Config mockConfig = getConfig(); String tableId = "t1"; BaseTableDescriptor mockTableDescriptor = mock(BaseTableDescriptor.class); when(mockTableDescriptor.getTableId()).thenReturn(tableId); AtomicReference<TableImpl> table = new AtomicReference<>(); StreamApplicationDescriptorImpl streamAppDesc = new StreamApplicationDescriptorImpl(appDesc -> { table.set((TableImpl) appDesc.getTable(mockTableDescriptor)); }, mockConfig); assertEquals(tableId, table.get().getTableId()); }