BasicClusterableRequestHandlerTest.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 org.apache.openejb.server.ejbd;

import com.agical.rmock.core.describe.ExpressionDescriber;
import com.agical.rmock.core.match.operator.AbstractExpression;
import com.agical.rmock.extension.junit.RMockTestCase;
import org.apache.openejb.AppContext;
import org.apache.openejb.BeanContext;
import org.apache.openejb.ClusteredRPCContainer;
import org.apache.openejb.ModuleContext;
import org.apache.openejb.client.ClusterableRequest;
import org.apache.openejb.client.ClusterableResponse;
import org.apache.openejb.client.ServerMetaData;
import org.apache.openejb.loader.SystemInstance;

import java.io.IOException;
import java.net.URI;

public class BasicClusterableRequestHandlerTest extends RMockTestCase {

    private BasicClusterableRequestHandler requestHandler;
    private ClusterableRequest request;
    private ClusterableResponse response;
    private BeanContext beanContext;
    private ClusteredRPCContainer clusteredContainer;

    @Override
    protected void setUp() throws Exception {
        requestHandler = new BasicClusterableRequestHandler();
        request = (ClusterableRequest) mock(ClusterableRequest.class);
        response = (ClusterableResponse) mock(ClusterableResponse.class);
        clusteredContainer = (ClusteredRPCContainer) mock(ClusteredRPCContainer.class);
        beanContext = new BeanContext("aDeploymentId",
            null,
            new ModuleContext("", null, "", new AppContext("", SystemInstance.get(), null, null, null, false), null, null),
            BasicClusterableRequestHandlerTest.class,
            null,
            null,
            null,
            null,
            null,
            null,
            null,
            null,
            null,
            null,
            false, false);
    }

    public void testNoOpWhenNotAClusteredContainer() throws Exception {
        beanContext.getContainer();

        startVerification();

        requestHandler.updateServer(beanContext, request, response);
    }

    public void testUpdateServerWhenRequestHashDiffersFromServerSideHash() throws Exception {
        final int port = SystemInstance.get().getOptions().get("ejbd.port", 4201);
        final URI[] locations = new URI[]{new URI("ejbd://localhost:" + port)};
        ServerMetaData server = new ServerMetaData(locations);

        beanContext.setContainer(clusteredContainer);

        request.getServerHash();
        modify().returnValue(server.buildHash() + 1);

        response.setServer(null);
        modify().args(new AbstractExpression() {
            @Override
            public void describeWith(ExpressionDescriber arg0) throws IOException {
            }

            @Override
            public boolean passes(Object arg0) {
                ServerMetaData actualServer = (ServerMetaData) arg0;
                assertSame(locations, actualServer.getLocations());
                return true;
            }
        });

        clusteredContainer.getLocations(beanContext);
        modify().returnValue(locations);

        startVerification();

        requestHandler.updateServer(beanContext, request, response);
    }

    public void testServerIsNotUpdatedWhenRequestHashEqualsServerSideHash() throws Exception {
        final int port = SystemInstance.get().getOptions().get("ejbd.port", 4201);
        URI[] locations = new URI[]{new URI("ejbd://localhost:" + port)};
        ServerMetaData server = new ServerMetaData(locations);

        beanContext.setContainer(clusteredContainer);

        request.getServerHash();
        modify().returnValue(server.buildHash());

        clusteredContainer.getLocations(beanContext);
        modify().returnValue(locations);

        startVerification();

        requestHandler.updateServer(beanContext, request, response);
    }

}