Examples with CacheID org.teiid.dqp.internal.process.SessionAwareCache.CacheID used on opensource projects

Example 16 with CacheID

use of org.teiid.dqp.internal.process.SessionAwareCache.CacheID in project teiid by teiid.

the class TestSessionAwareCache method testRemove.

@Test
public void testRemove() {
    SessionAwareCache<Cachable> cache = new SessionAwareCache<Cachable>("resultset", DefaultCacheFactory.INSTANCE, SessionAwareCache.Type.RESULTSET, 0);
    CacheID id = new CacheID(buildWorkContext(), new ParseInfo(), "SELECT * FROM FOO");
    Cachable result = Mockito.mock(Cachable.class);
    Mockito.stub(result.prepare((BufferManager) anyObject())).toReturn(true);
    Mockito.stub(result.restore((BufferManager) anyObject())).toReturn(true);
    id = new CacheID(buildWorkContext(), new ParseInfo(), "SELECT * FROM FOO");
    cache.put(id, Determinism.VDB_DETERMINISTIC, result, null);
    Object c = cache.get(id);
    assertTrue(result == c);
    assertTrue(cache.remove(id, Determinism.VDB_DETERMINISTIC) != null);
    assertNull(cache.get(id));
    // session scope
    cache.put(id, Determinism.SESSION_DETERMINISTIC, result, null);
    assertTrue(cache.get(id) != null);
    assertTrue(cache.remove(id, Determinism.SESSION_DETERMINISTIC) != null);
    assertNull(cache.get(id));
}