apps/vtn/vtnrsc/src/test/java/org/onosproject/vtnrsc/flowclassifier/impl/FlowClassifierManagerTest.java

/*
 * Copyright 2015-present Open Networking Foundation
 *
 * Licensed 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.onosproject.vtnrsc.flowclassifier.impl;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.notNullValue;

import org.junit.Test;

import org.onlab.packet.IpPrefix;

import org.onosproject.vtnrsc.TenantId;
import org.onosproject.vtnrsc.DefaultFlowClassifier;
import org.onosproject.vtnrsc.FlowClassifierId;
import org.onosproject.vtnrsc.VirtualPortId;
import org.onosproject.vtnrsc.FlowClassifier;
import org.onosproject.vtnrsc.util.VtnStorageServiceTest;
import org.onosproject.common.event.impl.TestEventDispatcher;

import static org.onosproject.net.NetTestTools.injectEventDispatcher;

/**
 * Unit tests for FlowClassifierManager class.
 */
public class FlowClassifierManagerTest {

    final String name = "FlowClassifier";
    final String description = "FlowClassifier";
    final String ethType = "IPv4";
    final String protocol = "udp";
    final int minSrcPortRange = 1024;
    final int maxSrcPortRange = 5000;
    final int minDstPortRange = 1024;
    final int maxDstPortRange = 5000;
    final FlowClassifierId flowClassifierId = FlowClassifierId.of("71111111-fc23-aeb6-f44b-56dc5e2fb3ae");
    final TenantId tenantId = TenantId.tenantId("8");
    final IpPrefix srcIpPrefix = IpPrefix.valueOf("0.0.0.0/0");
    final IpPrefix dstIpPrefix = IpPrefix.valueOf("100.100.100.100/0");
    final VirtualPortId virtualSrcPort = VirtualPortId.portId("100");
    final VirtualPortId virtualDstPort = VirtualPortId.portId("200");
    DefaultFlowClassifier.Builder flowClassifierBuilder = new DefaultFlowClassifier.Builder();
    FlowClassifierManager flowClassifierMgr = new FlowClassifierManager();
    FlowClassifier flowClassifier = null;
    private final VtnStorageServiceTest storageService = new VtnStorageServiceTest();

    /**
     * Checks the operation of createFlowClassifier() method.
     */
    @Test
    public void testCreateFlowClassifier() {
        // initialize flow classifier manager
        flowClassifierMgr.storageService = storageService;
        injectEventDispatcher(flowClassifierMgr, new TestEventDispatcher());
        flowClassifierMgr.activate();

        // create flow classifier
        flowClassifier = flowClassifierBuilder.setFlowClassifierId(flowClassifierId).setTenantId(tenantId)
                .setName(name).setDescription(description).setEtherType(ethType).setProtocol(protocol)
                .setMinSrcPortRange(minSrcPortRange).setMaxSrcPortRange(maxSrcPortRange)
                .setMinDstPortRange(minDstPortRange).setMaxDstPortRange(maxDstPortRange).setSrcIpPrefix(srcIpPrefix)
                .setDstIpPrefix(dstIpPrefix).setSrcPort(virtualSrcPort).setDstPort(virtualDstPort).build();
        assertThat(flowClassifierMgr.createFlowClassifier(flowClassifier), is(true));
    }

    /**
     * Checks the operation of exists() method.
     */
    @Test
    public void testExists() {
        testCreateFlowClassifier();
        assertThat(flowClassifierMgr.exists(flowClassifierId), is(true));
    }

    /**
     * Checks the operation of getFlowClassifierCount() method.
     */
    @Test
    public void testGetFlowClassifierCount() {
        testCreateFlowClassifier();
        assertThat(flowClassifierMgr.getFlowClassifierCount(), is(1));
    }

    /**
     * Checks the operation of getFlowClassifiers() method.
     */
    @Test
    public void testGetFlowClassifiers() {
        testCreateFlowClassifier();
        final Iterable<FlowClassifier> flowClassifierList = flowClassifierMgr.getFlowClassifiers();
        assertThat(flowClassifierList, is(notNullValue()));
        assertThat(flowClassifierList.iterator().hasNext(), is(true));
    }

    /**
     * Checks the operation of getFlowClassifier() method.
     */
    @Test
    public void testGetFlowClassifier() {
        testCreateFlowClassifier();
        assertThat(flowClassifier, is(notNullValue()));
        assertThat(flowClassifierMgr.getFlowClassifier(flowClassifierId), is(flowClassifier));
    }

    /**
     * Checks the operation of updateFlowClassifier() method.
     */
    @Test
    public void testUpdateFlowClassifier() {
        // create a flow classifier
        testCreateFlowClassifier();

        // new updates
        final String name2 = "FlowClassifier2";
        final String description2 = "FlowClassifier2";
        final String ethType2 = "IPv6";
        final String protocol2 = "tcp";
        final TenantId tenantId2 = TenantId.tenantId("10");
        final VirtualPortId virtualSrcPort2 = VirtualPortId.portId("300");
        final VirtualPortId virtualDstPort2 = VirtualPortId.portId("400");
        flowClassifier = flowClassifierBuilder.setFlowClassifierId(flowClassifierId)
                .setTenantId(tenantId2).setName(name2).setDescription(description2).setEtherType(ethType2)
                .setProtocol(protocol2).setMinSrcPortRange(minSrcPortRange).setMaxSrcPortRange(maxSrcPortRange)
                .setMinDstPortRange(minDstPortRange).setMaxDstPortRange(maxDstPortRange).setSrcIpPrefix(srcIpPrefix)
                .setDstIpPrefix(dstIpPrefix).setSrcPort(virtualSrcPort2).setDstPort(virtualDstPort2).build();
        assertThat(flowClassifierMgr.updateFlowClassifier(flowClassifier), is(true));
    }

    /**
     * Checks the operation of removeFlowClassifier() method.
     */
    @Test
    public void testRemoveFlowClassifier() {
        testCreateFlowClassifier();
        assertThat(flowClassifierMgr.removeFlowClassifier(flowClassifierId), is(true));
    }
}