Examples with MockDataStore org.eclipse.jetty.memcached.sessions.MemcachedTestHelper.MockDataStore used on opensource projects

Example 1 with MockDataStore

use of org.eclipse.jetty.memcached.sessions.MemcachedTestHelper.MockDataStore in project jetty.project by eclipse.

the class CachingSessionDataStoreTest method testSessionCRUD.

@Test
public void testSessionCRUD() throws Exception {
    String servletMapping = "/server";
    int scavengePeriod = -1;
    int maxInactivePeriod = -1;
    DefaultSessionCacheFactory cacheFactory = new DefaultSessionCacheFactory();
    SessionDataStoreFactory storeFactory = MemcachedTestHelper.newSessionDataStoreFactory();
    //Make sure sessions are evicted on request exit so they will need to be reloaded via cache/persistent store
    TestServer server = new TestServer(0, maxInactivePeriod, scavengePeriod, cacheFactory, storeFactory);
    ServletContextHandler context = server.addContext("/");
    context.addServlet(TestServlet.class, servletMapping);
    String contextPath = "";
    try {
        server.start();
        int port = server.getPort();
        HttpClient client = new HttpClient();
        client.start();
        try {
            //
            //Create a session
            //
            ContentResponse response = client.GET("http://localhost:" + port + contextPath + servletMapping + "?action=create");
            assertEquals(HttpServletResponse.SC_OK, response.getStatus());
            String sessionCookie = response.getHeaders().get("Set-Cookie");
            assertTrue(sessionCookie != null);
            // Mangle the cookie, replacing Path with $Path, etc.
            sessionCookie = sessionCookie.replaceFirst("(\\W)(P|p)ath=", "$1\\$Path=");
            String id = TestServer.extractSessionId(sessionCookie);
            //check that the memcache contains the session, and the session data store contains the session
            CachingSessionDataStore ds = (CachingSessionDataStore) context.getSessionHandler().getSessionCache().getSessionDataStore();
            assertNotNull(ds);
            SessionDataStore persistentStore = ds.getSessionStore();
            SessionDataMap dataMap = ds.getSessionDataMap();
            //the backing persistent store contains the session
            assertNotNull(persistentStore.load(id));
            //the memcache cache contains the session
            assertNotNull(dataMap.load(id));
            //
            //Update a session and check that is is NOT loaded via the persistent store
            //
            ((MockDataStore) persistentStore).zeroLoadCount();
            Request request = client.newRequest("http://localhost:" + port + contextPath + servletMapping + "?action=update");
            request.header("Cookie", sessionCookie);
            response = request.send();
            assertEquals(HttpServletResponse.SC_OK, response.getStatus());
            assertEquals(0, ((MockDataStore) persistentStore).getLoadCount());
            //check it was updated in the persistent store
            SessionData sd = persistentStore.load(id);
            assertNotNull(sd);
            assertEquals("bar", sd.getAttribute("foo"));
            //check it was updated in the cache
            sd = dataMap.load(id);
            assertNotNull(sd);
            assertEquals("bar", sd.getAttribute("foo"));
        } finally {
            client.stop();
        }
    } finally {
        server.stop();
    }
}