OpenDaylightActionsTest.java

package org.opennaas.extensions.openflowswitch.driver.opendaylight.test;

import java.util.ArrayList;
import java.util.List;

import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.opennaas.core.resources.action.ActionException;
import org.opennaas.extensions.openflowswitch.capability.OpenflowForwardingActionSet;
import org.opennaas.extensions.openflowswitch.driver.opendaylight.actionssets.OpenDaylightConstants;
import org.opennaas.extensions.openflowswitch.driver.opendaylight.actionssets.actions.CreateOFForwardingAction;
import org.opennaas.extensions.openflowswitch.model.FloodlightOFAction;
import org.opennaas.extensions.openflowswitch.model.OpenDaylightOFFlow;
import org.opennaas.extensions.openflowswitch.model.FloodlightOFMatch;

public class OpenDaylightActionsTest {

    private CreateOFForwardingAction createOFForwardingAction;

    @Before
    public void initActions() {
        createOFForwardingAction = new CreateOFForwardingAction();
        createOFForwardingAction.setActionID(OpenflowForwardingActionSet.CREATEOFFORWARDINGRULE);
    }

    @Test
    public void createOFForwardingActionValidCreationTest() throws ActionException {
        OpenDaylightOFFlow forwardingRule = generateValidFloodlightOFFlow();
        boolean isOk = createOFForwardingAction.checkParams(forwardingRule);
        Assert.assertTrue("Params are correct", isOk);
    }

    @Test(expected = ActionException.class)
    public void createOFForwardingActionValidInvalidGreaterPriorityCreationTest() throws ActionException {

        int priority = Integer.parseInt(OpenDaylightConstants.MAX_PRIORITY) + 1;

        OpenDaylightOFFlow forwardingRule = generateValidFloodlightOFFlow();
        forwardingRule.setPriority(String.valueOf(priority));
        boolean isOk = createOFForwardingAction.checkParams(forwardingRule);
        Assert.assertFalse("Check params failed", isOk);
    }

    @Test(expected = ActionException.class)
    public void createOFForwardingActionValidInvalidLowerPriorityCreationTest() throws ActionException {

        int priority = Integer.parseInt(OpenDaylightConstants.MIN_PRIORITY) - 1;

        OpenDaylightOFFlow forwardingRule = generateValidFloodlightOFFlow();
        forwardingRule.setPriority(String.valueOf(priority));
        boolean isOk = createOFForwardingAction.checkParams(forwardingRule);
        Assert.assertFalse("Check params failed", isOk);
    }

    private static OpenDaylightOFFlow generateValidFloodlightOFFlow() {
        return generateSampleFloodlightOFFlow("testingFlow", "1", "2");
    }

    private static OpenDaylightOFFlow generateSampleFloodlightOFFlow(String name, String inputPort, String outputPort) {

        OpenDaylightOFFlow forwardingRule = new OpenDaylightOFFlow();
        forwardingRule.setName(name);
        forwardingRule.setPriority(OpenDaylightConstants.DEFAULT_PRIORITY);
        forwardingRule.setActive(true);

        FloodlightOFMatch match = new FloodlightOFMatch();
        match.setIngressPort(inputPort);
        forwardingRule.setMatch(match);

        FloodlightOFAction floodlightAction = new FloodlightOFAction();
        floodlightAction.setType("output");
        floodlightAction.setValue(outputPort);

        List<FloodlightOFAction> actions = new ArrayList<FloodlightOFAction>();
        actions.add(floodlightAction);

        forwardingRule.setActions(actions);

        return forwardingRule;
    }
}