/**
* 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);
}
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);
}
}