MockTaskSideInputStorageManagerBuilder addLoggedStore(String storeName, Set<SystemStreamPartition> ssps) { StorageEngine storageEngine = mock(StorageEngine.class); when(storageEngine.getStoreProperties()).thenReturn( new StoreProperties.StorePropertiesBuilder().setLoggedStore(false).setPersistedToDisk(true).build()); stores.put(storeName, storageEngine); storeToProcessor.put(storeName, mock(SideInputsProcessor.class)); storeToSSps.put(storeName, ssps); return this; }
MockTaskSideInputStorageManagerBuilder addInMemoryStore(String storeName, Set<SystemStreamPartition> ssps) { StorageEngine storageEngine = mock(StorageEngine.class); when(storageEngine.getStoreProperties()).thenReturn( new StoreProperties.StorePropertiesBuilder().setLoggedStore(false).setPersistedToDisk(false).build()); stores.put(storeName, storageEngine); storeToProcessor.put(storeName, mock(SideInputsProcessor.class)); storeToSSps.put(storeName, ssps); return this; }
return new StoreProperties.StorePropertiesBuilder().setLoggedStore(true).build(); }).when(mockStorageEngine).getStoreProperties();
@Override public StorageEngine getStorageEngine(String storeName, File storeDir, Serde<Object> keySerde, Serde<Object> msgSerde, MessageCollector collector, MetricsRegistry registry, SystemStreamPartition changeLogSystemStreamPartition, JobContext jobContext, ContainerContext containerContext, StoreMode storeMode) { StoreProperties storeProperties = new StoreProperties.StorePropertiesBuilder().setLoggedStore(true).build(); return new MockStorageEngine(storeName, storeDir, changeLogSystemStreamPartition, storeProperties); } }