@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();
  }