BeanCommandTest.java

package org.cyclopsgroup.jmxterm.cmd;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;

import java.io.IOException;
import java.io.StringWriter;

import javax.management.JMException;
import javax.management.MBeanServerConnection;
import javax.management.ObjectName;

import org.cyclopsgroup.jmxterm.MockSession;
import org.jmock.Expectations;
import org.jmock.Mockery;
import org.junit.Test;

/**
 * Test case for {@link BeanCommand}
 *
 * @author <a href="mailto:jiaqi.guo@gmail.com">Jiaqi Guo</a>
 */
public class BeanCommandTest
{
    private BeanCommand command = new BeanCommand();

    private StringWriter output = new StringWriter();

    private void setBeanAndVerify( String beanName, String domainName, final String expectedBean )
        throws IOException, JMException
    {
        Mockery context = new Mockery();
        final MBeanServerConnection con = context.mock( MBeanServerConnection.class );
        command.setBean( beanName );
        MockSession s = new MockSession( output, con );
        if ( domainName != null )
        {
            s.setDomain( domainName );
        }
        context.checking( new Expectations()
        {
            {
                atLeast( 1 ).of( con ).getMBeanInfo( new ObjectName( expectedBean ) );
            }
        } );
        command.setSession( s );
        command.execute();
        assertEquals( expectedBean, s.getBean() );
    }

    /**
     * Test execution with NULL result
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test
    public void testExecuteWithGettingNull()
        throws IOException, JMException
    {
        command.setSession( new MockSession( output, null ) );
        command.execute();
        assertEquals( "null", output.toString().trim() );
    }

    /**
     * Test execution with some result
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test
    public void testExecuteWithGettingSomething()
        throws IOException, JMException
    {
        MockSession s = new MockSession( output, null );
        s.setBean( "something" );
        command.setSession( s );
        command.execute();
        assertEquals( "something", output.toString().trim() );
    }

    /**
     * Test the case where an illegal bean is requested
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test( expected = IllegalArgumentException.class )
    public void testExecuteWithInvalidBean()
        throws IOException, JMException
    {
        command.setBean( "blablabla" );
        command.setSession( new MockSession( output, null ) );
        command.execute();
    }

    /**
     * Test the case where NULL is get
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test
    public void testExecuteWithSettingNull()
        throws IOException, JMException
    {
        command.setBean( "null" );
        MockSession s = new MockSession( output, null );
        s.setBean( "something" );
        command.setSession( s );
        command.execute();
        assertNull( s.getBean() );
    }

    /**
     * Test setting names with special character such as dot, dash and underline, without setting a domain first
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test
    public void testSettingSpecialCharactersWithoutDomain()
        throws IOException, JMException
    {
        setBeanAndVerify( "domain_name.with-dash:attr.name_1-1=a.b", null, "domain_name.with-dash:attr.name_1-1=a.b" );
    }

    /**
     * Test the case where a domain is set
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test
    public void testSettingWithDomain()
        throws IOException, JMException
    {
        setBeanAndVerify( "type=x", "something", "something:type=x" );
    }

    /**
     * Test the case where domain is set
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test
    public void testSettingWithoutDomain()
        throws IOException, JMException
    {
        setBeanAndVerify( "something:type=x", null, "something:type=x" );
    }

    /**
     * Test setting names with special character such as dot, dash and underline
     *
     * @throws IOException Allows network IO errors
     * @throws JMException Allows JMX exceptions
     */
    @Test
    public void testSettingWithSpecialCharacters()
        throws IOException, JMException
    {
        setBeanAndVerify( "attr.name_1-1=a.b", "domain_name.with-dash", "domain_name.with-dash:attr.name_1-1=a.b" );
    }
}