core/src/test/java/flex/messaging/MessageBrokerTest.java

/*
 * 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 flex.messaging;

import flex.messaging.config.ConfigurationConstants;
import flex.messaging.config.ConfigurationException;
import flex.messaging.factories.JavaFactory;
import flex.messaging.services.MessageService;
import flex.messaging.services.Service;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;

import java.util.List;
import java.util.Map;

/**
 * Important: While adding new tests, make sure you don't create dependencies
 * to other modules. (eg. don't use RemotingDestination as that would create
 * dependency on remoting module; instead add the test in remoting module).
 */
public class MessageBrokerTest {

    private MessageBroker broker;

    @Before
    public void setUp() throws Exception {
        broker = new MessageBroker(false);
        broker.initThreadLocals();
    }

    @Test
    public void testAddFactory() {
        String id = "java2";
        JavaFactory expected = new JavaFactory();
        broker.addFactory("java2", expected);

        JavaFactory actual = (JavaFactory) broker.getFactory(id);
        Assert.assertEquals(expected, actual);
    }

    @Test
    public void testAddSameFactoryIdTwice() {
        String id = "java";
        JavaFactory expected = new JavaFactory();
        try {
            broker.addFactory(id, expected);
            Assert.fail("ConfigurationException expected");
        } catch (ConfigurationException ce) {
            int error = ConfigurationConstants.DUPLICATE_COMPONENT_ID;
            Assert.assertTrue(ce.getNumber() == error);
        }
    }

    @Test
    public void testAddSameFactoryInstanceTwice() {
        String id = "java2";
        JavaFactory expected = new JavaFactory();
        broker.addFactory("java2", expected);
        broker.addFactory("java2", expected);

        JavaFactory actual = (JavaFactory) broker.getFactory(id);
        Assert.assertEquals(expected, actual);
    }

    @Test
    public void testAddFactoryNullId() {
        JavaFactory expected = new JavaFactory();
        try {
            broker.addFactory(null, expected);
            Assert.fail("ConfigurationException expected");
        } catch (ConfigurationException ce) {
            int error = ConfigurationConstants.NULL_COMPONENT_ID;
            Assert.assertTrue(ce.getNumber() == error);
        }
    }

    @Test
    public void testGetFactories() {
        String id = "java2";
        JavaFactory expected = new JavaFactory();
        broker.addFactory(id, expected);

        int size = broker.getFactories().size();
        Assert.assertEquals(2, size);
    }

    @Test
    public void testRemoveFactory() {
        String id = "java";
        broker.removeFactory(id);
        FlexFactory actual = broker.getFactory(id);

        Assert.assertEquals(null, actual);
    }

    @Test
    public void testRemoveFactoryNonexistent() {
        String id = "nonexistent";
        FlexFactory actual = broker.removeFactory(id);

        Assert.assertNull(actual);
    }

    @Test
    public void testAddService() {
        String id = "dummy-service";
        MessageService expected = new MessageService();
        expected.setId(id);
        broker.addService(expected);

        Service actual = broker.getService(id);
        Assert.assertEquals(expected, actual);
        Assert.assertEquals(expected.isManaged(), actual.isManaged());
    }

    @Test
    public void testAddServiceNonManaged() {
        String id = "dummy-service";
        MessageService expected = new MessageService(false);
        expected.setId(id);
        broker.addService(expected);

        Service actual = broker.getService(id);
        Assert.assertEquals(expected, actual);
        Assert.assertEquals(expected.isManaged(), actual.isManaged());
    }

    @Test
    public void testAddServiceNull() {
        try {
            broker.addService(null);

            Assert.fail("ConfigurationException expected");
        } catch (ConfigurationException ce) {
            int error = ConfigurationConstants.NULL_COMPONENT;
            Assert.assertTrue(ce.getNumber() == error);
        }
    }

    @Test
    public void testAddServiceNullId() {
        try {
            MessageService svc = new MessageService();
            broker.addService(svc);

            Assert.fail("ConfigurationException expected");
        } catch (ConfigurationException ce) {
            int error = ConfigurationConstants.NULL_COMPONENT_ID;
            Assert.assertTrue(ce.getNumber() == error);
        }
    }

    @Test
    public void testAddServiceExists() {
        try {
            String id = "dummy-service";
            MessageService svc1 = new MessageService();
            svc1.setId(id);
            broker.addService(svc1);
            MessageService svc2 = new MessageService();
            svc2.setId(id);
            broker.addService(svc2);

            Assert.fail("ConfigurationException expected");
        } catch (ConfigurationException ce) {
            int error = ConfigurationConstants.DUPLICATE_COMPONENT_ID;
            Assert.assertTrue(ce.getNumber() == error);
        }
    }

    @Test
    public void testCreateService() {

        String serviceId = "message-service";
        String serviceClass = MessageService.class.getName();
        Service expected = broker.createService(serviceId, serviceClass);

        Service actual = broker.getService(serviceId);
        Assert.assertEquals(expected, actual);
    }

    @Test
    public void testCreateServiceNonexistingClass() {
        String serviceId = "remoting-service";
        String serviceClass = "NonexistingClass";

        try {
            broker.createService(serviceId, serviceClass);
            Assert.fail("MessageException expected");
        } catch (MessageException me) {
            int error = 10008;  //ClassUtil.TYPE_NOT_FOUND
            Assert.assertEquals(me.getNumber(), error);
        }

    }

    @Test
    public void testCreateServiceWithExistingId() {
        String id = "message-service";
        String serviceClass = MessageService.class.getName();

        broker.createService(id, serviceClass);
        try {
            broker.createService(id, serviceClass);

            Assert.fail("ConfigurationException expected");
        } catch (ConfigurationException ce) {
            int error = ConfigurationConstants.DUPLICATE_COMPONENT_ID;
            Assert.assertEquals(ce.getNumber(), error);
        }
    }

    @Test
    public void testRemoveService() {
        String id = "dummy-service";
        MessageService svc = new MessageService();
        svc.setId(id);
        broker.addService(svc);
        broker.removeService(id);
        Service actual = broker.getService(id);

        Assert.assertEquals(null, actual);
    }

    @Test
    public void testRemoveServiceNonexistent() {
        Service actual = broker.removeService("id");
        Assert.assertNull(actual);
    }

    @Test
    public void testGetService() {
        String id = "dummy-service";
        MessageService expected = new MessageService();
        expected.setId(id);
        broker.addService(expected);

        Service actual = broker.getService(id);
        Assert.assertEquals(expected, actual);
    }


    //Get services_no AUthentication
    @Test
    public void testGetServices() {
        String id = "remoting-service";
        MessageService expected1 = new MessageService();
        expected1.setId(id);
        id = "message-service";
        MessageService expected2 = new MessageService();
        expected2.setId(id);

        broker.addService(expected1);
        broker.addService(expected2);

        Map<String, Service> actual = broker.getServices();
        Assert.assertEquals(expected1, actual.get("remoting-service"));
        Assert.assertEquals(expected2, actual.get("message-service"));
    }

    //GetServices should not return an AuthenticationService to the client
//    public void testGetServicesNoAuthenticationSvc()
//    {
//        String id = "remoting-service";
//        RemotingService expected = new RemotingService();
//        expected.setId(id);
//        id = "auth-service";
//        AuthenticationService notExpected = new AuthenticationService();
//        notExpected.setId(id);
//        
//        broker.addService(expected);
//        broker.addService(notExpected);
//        
//        List actual = broker.getServices();
//        Assert.assertEquals(expected, actual.get(0));
//        Assert.assertNotSame(notExpected, actual.get(1));
//        Assert.assertEquals(1, actual.size());
//    }

    @Test
    public void testGetServiceNonExistent() {
        Service actual = broker.getService("non-existent-id");
        Assert.assertNull(actual);
    }

    @Test
    public void testGetChannelIdsNull() {
        List<String> channelIds = broker.getChannelIds();
        Assert.assertNull(channelIds);
    }
}