org.apache.flume.RabbitMQUtilTest.java Source code

Introduction

Here is the source code for org.apache.flume.RabbitMQUtilTest.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.apache.flume;

import com.rabbitmq.client.ConnectionFactory;
import junit.framework.Assert;
import org.junit.Test;
import org.junit.Before;

/**
 *
 * @author jcustenborder
 */
public class RabbitMQUtilTest {
    @Test
    public void close_null() {
        RabbitMQUtil.close(null, null);
    }

    @Test(expected = java.lang.IllegalArgumentException.class)
    public void getFactory_null() {
        RabbitMQUtil.getFactory(null);
    }

    @Test(expected = java.lang.IllegalArgumentException.class)
    public void getFactory_null_hostname() {
        context.put(RabbitMQConstants.CONFIG_HOSTNAME, null);
        RabbitMQUtil.getFactory(context);
    }

    Context context = null;

    @Before
    public void createContext() {
        context = new Context();
        context.put(RabbitMQConstants.CONFIG_HOSTNAME, "server01.example.com");
        context.put(RabbitMQConstants.CONFIG_PORT, "12345");
        context.put(RabbitMQConstants.CONFIG_CONNECTIONTIMEOUT, "30000");
        context.put(RabbitMQConstants.CONFIG_PASSWORD, "daofoasidnfioand");
        context.put(RabbitMQConstants.CONFIG_USERNAME, "asdfasdfasd");
        context.put(RabbitMQConstants.CONFIG_VIRTUALHOST, "virtualhost1");
    }

    @Test
    public void getFactory() {
        ConnectionFactory factory = RabbitMQUtil.getFactory(context);
        Assert.assertNotNull("factory should not be null", context);

        Assert.assertEquals("Host does not match", context.getString(RabbitMQConstants.CONFIG_HOSTNAME),
                factory.getHost());
        Assert.assertEquals("Port does not match", context.getInteger(RabbitMQConstants.CONFIG_PORT),
                (Integer) factory.getPort());
        Assert.assertEquals("ConnectionTimeout does not match",
                context.getInteger(RabbitMQConstants.CONFIG_CONNECTIONTIMEOUT),
                (Integer) factory.getConnectionTimeout());
        Assert.assertEquals("Password does not match", context.getString(RabbitMQConstants.CONFIG_PASSWORD),
                factory.getPassword());
        Assert.assertEquals("Username does not match", context.getString(RabbitMQConstants.CONFIG_USERNAME),
                factory.getUsername());
        Assert.assertEquals("VirtualHost does not match", context.getString(RabbitMQConstants.CONFIG_VIRTUALHOST),
                factory.getVirtualHost());
    }

    public void getFactory_minimal() {
        context = new Context();
        context.put(RabbitMQConstants.CONFIG_HOSTNAME, "server01.example.com");

        ConnectionFactory factory = RabbitMQUtil.getFactory(context);
        Assert.assertNotNull("factory should not be null", context);

        Assert.assertEquals("Host does not match", context.getString(RabbitMQConstants.CONFIG_HOSTNAME),
                factory.getHost());
    }

}