@Test public void testCallsForwardedToNonPartitionedBackend() throws Exception { AbstractStateBackend nonPartBackend = mock(AbstractStateBackend.class); RocksDBStateBackend rocksDbBackend = new RocksDBStateBackend(TEMP_URI, nonPartBackend); rocksDbBackend.initializeForJob(getMockEnvironment(), "foo", IntSerializer.INSTANCE); verify(nonPartBackend, times(1)) .initializeForJob(any(Environment.class), anyString(), any(TypeSerializer.class)); rocksDbBackend.disposeAllStateForCurrentJob(); verify(nonPartBackend, times(1)).disposeAllStateForCurrentJob(); rocksDbBackend.close(); verify(nonPartBackend, times(1)).close(); }