Examples with TestStats voldemort.socketpool.AbstractSocketPoolTest.TestStats used on opensource projects

Example 1 with TestStats

use of voldemort.socketpool.AbstractSocketPoolTest.TestStats in project voldemort by voldemort.

the class SimpleSocketPoolTest method testClientRequestExecutorLimitSomeTimeout.

@Test
public void testClientRequestExecutorLimitSomeTimeout() throws Exception {
    // start a dummy server
    AbstractSocketService server = ServerTestUtils.getSocketService(useNio, new ClientRequestHandlerFactory(null), 7666, 50, 50, 1000);
    server.start();
    final ResourcePoolConfig config = new ResourcePoolConfig().setTimeout(50, TimeUnit.MILLISECONDS).setMaxPoolSize(20);
    ClientRequestExecutorPool clientRequestExecutorPool = new ClientRequestExecutorPool(config.getMaxPoolSize(), (int) config.getTimeout(TimeUnit.MILLISECONDS), 100, 1000);
    try {
        ResourceFactory<SocketDestination, ClientRequestExecutor> factory = clientRequestExecutorPool.getFactory();
        final AbstractSocketPoolTest<SocketDestination, ClientRequestExecutor> test = new AbstractSocketPoolTest<SocketDestination, ClientRequestExecutor>() {

            @Override
            protected void doSomethingWithResource(SocketDestination key, ClientRequestExecutor resource) throws Exception {
                Thread.sleep(100);
                int random = (int) (Math.random() * 10);
                if (random >= 5)
                    resource.close();
            }

            @Override
            protected SocketDestination getRequestKey() throws Exception {
                return new SocketDestination("localhost", 7666, RequestFormatType.VOLDEMORT_V1);
            }
        };
        // borrow timeout >> doSomething() no timeout expected
        TestStats testStats = test.startTest(factory, config, 50, 200);
        assertEquals("We should see some timeoutRequests", true, testStats.timeoutRequests > 0);
        server.stop();
    } finally {
        clientRequestExecutorPool.close();
    }
}

Example 2 with TestStats

use of voldemort.socketpool.AbstractSocketPoolTest.TestStats in project voldemort by voldemort.

the class SimpleSocketPoolTest method testPoolLimitNoTimeout.

@Test
public void testPoolLimitNoTimeout() throws Exception {
    final ResourcePoolConfig config = new ResourcePoolConfig().setTimeout(1000, TimeUnit.MILLISECONDS).setMaxPoolSize(20);
    ResourceFactory<String, String> factory = ResourcePoolTestUtils.getBasicPoolFactory();
    final AbstractSocketPoolTest<String, String> test = new AbstractSocketPoolTest<String, String>() {

        @Override
        protected void doSomethingWithResource(String key, String resource) throws Exception {
            Thread.sleep(100);
        }

        @Override
        protected String getRequestKey() throws Exception {
            return "test-key";
        }
    };
    // borrow timeout >> doSomething() no timeout expected
    TestStats testStats = test.startTest(factory, config, 50, 200);
    assertEquals("We should see Zero timeoutRequests", 0, testStats.timeoutRequests);
}

Example 3 with TestStats

use of voldemort.socketpool.AbstractSocketPoolTest.TestStats in project voldemort by voldemort.

the class SimpleSocketPoolTest method testPoolLimitSomeTimeout.

@Test
public void testPoolLimitSomeTimeout() throws Exception {
    final ResourcePoolConfig config = new ResourcePoolConfig().setTimeout(50, TimeUnit.MILLISECONDS).setMaxPoolSize(20);
    ResourceFactory<String, String> factory = ResourcePoolTestUtils.getBasicPoolFactory();
    final AbstractSocketPoolTest<String, String> test = new AbstractSocketPoolTest<String, String>() {

        @Override
        protected void doSomethingWithResource(String key, String resource) throws Exception {
            Thread.sleep(100);
        }

        @Override
        protected String getRequestKey() throws Exception {
            return "test-key";
        }
    };
    // borrow timeout >> doSomething() no timeout expected
    TestStats testStats = test.startTest(factory, config, 50, 200);
    assertEquals("We should see some timeoutRequests", true, testStats.timeoutRequests > 0);
}

Example 4 with TestStats

use of voldemort.socketpool.AbstractSocketPoolTest.TestStats in project voldemort by voldemort.

the class SimpleSocketPoolTest method testSocketPoolLimitSomeTimeout.

@Test
public void testSocketPoolLimitSomeTimeout() throws Exception {
    // start a dummy server
    AbstractSocketService server = ServerTestUtils.getSocketService(useNio, new ClientRequestHandlerFactory(null), 7666, 50, 50, 1000);
    server.start();
    final ResourcePoolConfig config = new ResourcePoolConfig().setTimeout(50, TimeUnit.MILLISECONDS).setMaxPoolSize(20);
    ResourceFactory<SocketDestination, SocketAndStreams> factory = ResourcePoolTestUtils.getSocketPoolFactory();
    final AbstractSocketPoolTest<SocketDestination, SocketAndStreams> test = new AbstractSocketPoolTest<SocketDestination, SocketAndStreams>() {

        @Override
        protected void doSomethingWithResource(SocketDestination key, SocketAndStreams resource) throws Exception {
            Thread.sleep(100);
            int random = (int) (Math.random() * 10);
            if (random >= 5)
                resource.getSocket().close();
        }

        @Override
        protected SocketDestination getRequestKey() throws Exception {
            return new SocketDestination("localhost", 7666, RequestFormatType.VOLDEMORT_V1);
        }
    };
    // borrow timeout >> doSomething() no timeout expected
    TestStats testStats = test.startTest(factory, config, 50, 200);
    assertEquals("We should see some timeoutRequests", true, testStats.timeoutRequests > 0);
    server.stop();
}

Example 5 with TestStats

use of voldemort.socketpool.AbstractSocketPoolTest.TestStats in project voldemort by voldemort.

the class SimpleSocketPoolTest method testNoTimeout.

@Test
public void testNoTimeout() throws Exception {
    final ResourcePoolConfig config = new ResourcePoolConfig().setTimeout(100, TimeUnit.MILLISECONDS).setMaxPoolSize(20);
    ResourceFactory<String, String> factory = ResourcePoolTestUtils.getBasicPoolFactory();
    final AbstractSocketPoolTest<String, String> test = new AbstractSocketPoolTest<String, String>() {

        @Override
        protected void doSomethingWithResource(String key, String resource) throws Exception {
            Thread.sleep(200);
        }

        @Override
        protected String getRequestKey() throws Exception {
            return "test-key";
        }
    };
    // threads same as pool size no timeout expected
    TestStats testStats = test.startTest(factory, config, 20, 200);
    assertEquals("We should see Zero timeoutRequests", 0, testStats.timeoutRequests);
}