package org.infinispan.server.jgroups.subsystem;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.FAILED;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.OUTCOME;
import static org.jboss.as.controller.descriptions.ModelDescriptionConstants.SUCCESS;
import org.infinispan.server.commons.controller.Operations;
import org.infinispan.server.jgroups.spi.service.ProtocolStackServiceName;
import org.jboss.as.subsystem.test.KernelServices;
import org.jboss.byteman.contrib.bmunit.BMRule;
import org.jboss.byteman.contrib.bmunit.BMUnitRunner;
import org.jboss.dmr.ModelNode;
import org.jboss.msc.service.ServiceName;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
/**
* Test case for testing sequences of management operations.
*
* @author Richard Achmatowicz (c) 2011 Red Hat Inc.
*/
@RunWith(BMUnitRunner.class)
public class OperationSequencesTestCase extends OperationTestCaseBase {
// stack test operations
static final ModelNode addStackOp = getProtocolStackAddOperation("maximal2");
// addStackOpWithParams calls the operation below to check passing optional parameters
// /subsystem=jgroups/stack=maximal2:add(transport={type=UDP},protocols=[{type=MPING},{type=FLUSH}])
static final ModelNode addStackOpWithParams = getProtocolStackAddOperationWithParameters("maximal2");
static final ModelNode removeStackOp = getProtocolStackRemoveOperation("maximal2");
// transport test operations
static final ModelNode addTransportOp = getTransportAddOperation("maximal2", "UDP");
// protocol test operations
static final ModelNode addProtocolOp = getProtocolAddOperation("maximal2", "MPING");
@Test
public void testProtocolStackAddRemoveAddSequence() throws Exception {
KernelServices services = buildKernelServices();
ModelNode operation = Operations.createCompositeOperation(addStackOp, addTransportOp, addProtocolOp);
// add a protocol stack, its transport and a protocol as a batch
ModelNode result = services.executeOperation(operation);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
// remove the stack
result = services.executeOperation(removeStackOp);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
// add the same stack
result = services.executeOperation(operation);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
}
@Test
public void testProtocolStackRemoveRemoveSequence() throws Exception {
KernelServices services = buildKernelServices();
ModelNode operation = Operations.createCompositeOperation(addStackOp, addTransportOp, addProtocolOp);
// add a protocol stack
ModelNode result = services.executeOperation(operation);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
// remove the protocol stack
result = services.executeOperation(removeStackOp);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
// remove the protocol stack again
result = services.executeOperation(removeStackOp);
Assert.assertEquals(FAILED, result.get(OUTCOME).asString());
}
/*
* Tests the ability of the /subsystem=jgroups/stack=X:add() operation
* to correctly process the optional TRANSPORT and PROTOCOLS parameters.
*/
@Test
public void testProtocolStackAddRemoveSequenceWithParameters() throws Exception {
KernelServices services = buildKernelServices();
// add a protocol stack specifying TRANSPORT and PROTOCOLS parameters
ModelNode result = services.executeOperation(addStackOpWithParams);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
// check some random values
// remove the protocol stack
result = services.executeOperation(removeStackOp);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
// remove the protocol stack again
result = services.executeOperation(removeStackOp);
Assert.assertEquals(FAILED, result.get(OUTCOME).asString());
}
@org.junit.Ignore("This fails for some mysterious reason - but this isn't a critical test")
@Test
@BMRule(name="Test remove rollback operation",
targetClass="org.jboss.as.clustering.jgroups.subsystem.StackRemoveHandler",
targetMethod="performRuntime",
targetLocation="AT EXIT",
action="traceln(\"Injecting rollback fault via Byteman\");$1.setRollbackOnly()")
public void testProtocolStackRemoveRollback() throws Exception {
KernelServices services = buildKernelServices();
ModelNode operation = Operations.createCompositeOperation(addStackOp, addTransportOp, addProtocolOp);
// add a protocol stack
ModelNode result = services.executeOperation(operation);
Assert.assertEquals(SUCCESS, result.get(OUTCOME).asString());
// remove the protocol stack
// the remove has OperationContext.setRollbackOnly() injected
// and so is expected to fail
result = services.executeOperation(removeStackOp);
Assert.assertEquals(FAILED, result.get(OUTCOME).asString());
// need to check that all services are correctly re-installed
ServiceName channelFactoryServiceName = ProtocolStackServiceName.CHANNEL_FACTORY.getServiceName("maximal2");
Assert.assertNotNull("channel factory service not installed", services.getContainer().getService(channelFactoryServiceName));
}
}