org.springframework.data.redis.connection.jredis.JredisPoolTests.java Source code

Introduction

Here is the source code for org.springframework.data.redis.connection.jredis.JredisPoolTests.java

Source

/*
 * Copyright 2013-2014 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.springframework.data.redis.connection.jredis;

import static org.junit.Assert.*;

import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.jredis.JRedis;
import org.jredis.RedisException;
import org.jredis.connector.ConnectionSpec;
import org.jredis.connector.NotConnectedException;
import org.jredis.ri.alphazero.connection.DefaultConnectionSpec;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.springframework.data.redis.SettingsUtils;
import org.springframework.data.redis.connection.PoolException;

/**
 * Integration test of {@link JredisPool}.
 * 
 * @author Jennifer Hickey
 * @author Thomas Darimont
 * @author Christoph Strobl
 */
public class JredisPoolTests {

    private ConnectionSpec connectionSpec;

    private JredisPool pool;

    @Before
    public void setUp() {
        this.connectionSpec = DefaultConnectionSpec.newSpec(SettingsUtils.getHost(), SettingsUtils.getPort(), 0,
                null);
    }

    @After
    public void tearDown() {
        if (this.pool != null) {
            this.pool.destroy();
        }
    }

    @Test
    public void testGetResource() throws RedisException {
        this.pool = new JredisPool(connectionSpec);
        JRedis client = pool.getResource();
        assertNotNull(client);
        client.ping();
    }

    @Test
    public void testGetResourcePoolExhausted() {
        GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();
        poolConfig.setMaxTotal(1);
        poolConfig.setMaxWaitMillis(1);
        this.pool = new JredisPool(connectionSpec, poolConfig);
        JRedis client = pool.getResource();
        assertNotNull(client);
        try {
            pool.getResource();
            fail("PoolException should be thrown when pool exhausted");
        } catch (PoolException e) {

        }
    }

    @Test
    public void testGetResourceValidate() {
        GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();
        poolConfig.setTestOnBorrow(true);
        this.pool = new JredisPool(connectionSpec, poolConfig);
        JRedis client = pool.getResource();
        assertNotNull(client);
    }

    @Test(expected = PoolException.class)
    public void testGetResourceCreationUnsuccessful() {
        // Config poolConfig = new Config();
        // poolConfig.testOnBorrow = true;
        this.pool = new JredisPool(DefaultConnectionSpec.newSpec(SettingsUtils.getHost(), 3333, 0, null));
        pool.getResource();
    }

    @Test
    public void testReturnResource() throws RedisException {

        GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();
        poolConfig.setMaxTotal(1);
        poolConfig.setMaxWaitMillis(1);
        this.pool = new JredisPool(connectionSpec);
        JRedis client = pool.getResource();
        assertNotNull(client);
        pool.returnResource(client);
        assertNotNull(pool.getResource());
    }

    @Test
    public void testReturnBrokenResource() throws RedisException {

        GenericObjectPoolConfig poolConfig = new GenericObjectPoolConfig();
        poolConfig.setMaxTotal(1);
        poolConfig.setMaxWaitMillis(1);
        this.pool = new JredisPool(connectionSpec, poolConfig);
        JRedis client = pool.getResource();
        assertNotNull(client);
        pool.returnBrokenResource(client);
        JRedis client2 = pool.getResource();
        assertNotSame(client, client2);
        try {
            client.ping();
            fail("Broken resouce connection should be closed");
        } catch (NotConnectedException e) {
        }
    }

    @Test
    public void testCreateWithHostAndPort() {
        this.pool = new JredisPool(SettingsUtils.getHost(), SettingsUtils.getPort());
        assertNotNull(pool.getResource());
    }

    @Test
    public void testCreateWithHostPortAndDbIndex() {
        this.pool = new JredisPool(SettingsUtils.getHost(), SettingsUtils.getPort(), 1, null, 0);
        assertNotNull(pool.getResource());
    }

}