apps/openstackvtap/app/src/test/java/org/onosproject/openstackvtap/impl/DefaultOpenstackVtapTest.java

/*
 * Copyright 2018-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.openstackvtap.impl;

import com.google.common.collect.Sets;
import com.google.common.testing.EqualsTester;
import org.junit.Before;
import org.junit.Test;
import org.onlab.packet.IpAddress;
import org.onlab.packet.IpPrefix;
import org.onlab.packet.TpPort;
import org.onosproject.openstackvtap.api.OpenstackVtap;
import org.onosproject.openstackvtap.api.OpenstackVtapCriterion;
import org.onosproject.openstackvtap.api.OpenstackVtapId;

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

/**
 * Unit tests for DefaultOpenstackVtap.
 */
public class DefaultOpenstackVtapTest {

    private static final OpenstackVtapId VTAP_ID_1 = OpenstackVtapId.vTapId();
    private static final OpenstackVtapId VTAP_ID_2 = OpenstackVtapId.vTapId();

    private static final OpenstackVtap.Type VTAP_TYPE_1 = OpenstackVtap.Type.VTAP_TX;
    private static final OpenstackVtap.Type VTAP_TYPE_2 = OpenstackVtap.Type.VTAP_RX;

    private static final IpPrefix SRC_IP_PREFIX_1 =
            IpPrefix.valueOf(IpAddress.valueOf("10.10.10.1"), 32);
    private static final IpPrefix SRC_IP_PREFIX_2 =
            IpPrefix.valueOf(IpAddress.valueOf("20.20.20.1"), 32);
    private static final IpPrefix DST_IP_PREFIX_1 =
            IpPrefix.valueOf(IpAddress.valueOf("10.10.10.2"), 32);
    private static final IpPrefix DST_IP_PREFIX_2 =
            IpPrefix.valueOf(IpAddress.valueOf("20.20.20.2"), 32);

    private static final TpPort SRC_PORT_1 = TpPort.tpPort(10);
    private static final TpPort SRC_PORT_2 = TpPort.tpPort(20);
    private static final TpPort DST_PORT_1 = TpPort.tpPort(30);
    private static final TpPort DST_PORT_2 = TpPort.tpPort(40);

    private static final byte IP_PROTOCOL_1 = (byte) 1;
    private static final byte IP_PROTOCOL_2 = (byte) 2;

    private static final OpenstackVtapCriterion CRITERION_1 =
            createVtapCriterion(SRC_IP_PREFIX_1, DST_IP_PREFIX_1, SRC_PORT_1, DST_PORT_1, IP_PROTOCOL_1);
    private static final OpenstackVtapCriterion CRITERION_2 =
            createVtapCriterion(SRC_IP_PREFIX_2, DST_IP_PREFIX_2, SRC_PORT_2, DST_PORT_2, IP_PROTOCOL_2);

    private OpenstackVtap vtap1;
    private OpenstackVtap sameAsVtap1;
    private OpenstackVtap vtap2;

    @Before
    public void setup() {

        OpenstackVtap.Builder builder1 = DefaultOpenstackVtap.builder();

        vtap1 = builder1
                    .id(VTAP_ID_1)
                    .type(VTAP_TYPE_1)
                    .vTapCriterion(CRITERION_1)
                    .txDeviceIds(Sets.newHashSet())
                    .rxDeviceIds(Sets.newHashSet())
                    .annotations()
                    .build();

        OpenstackVtap.Builder builder2 = DefaultOpenstackVtap.builder();

        sameAsVtap1 = builder2
                        .id(VTAP_ID_1)
                        .type(VTAP_TYPE_1)
                        .vTapCriterion(CRITERION_1)
                        .txDeviceIds(Sets.newHashSet())
                        .rxDeviceIds(Sets.newHashSet())
                        .annotations()
                        .build();

        OpenstackVtap.Builder builder3 = DefaultOpenstackVtap.builder();

        vtap2 = builder3
                    .id(VTAP_ID_2)
                    .type(VTAP_TYPE_2)
                    .vTapCriterion(CRITERION_2)
                    .txDeviceIds(Sets.newHashSet())
                    .rxDeviceIds(Sets.newHashSet())
                    .annotations()
                    .build();

    }

    @Test
    public void testEquality() {
        new EqualsTester()
                .addEqualityGroup(vtap1, sameAsVtap1)
                .addEqualityGroup(vtap2).testEquals();
    }

    @Test
    public void testConstruction() {
        DefaultOpenstackVtap vtap = (DefaultOpenstackVtap) vtap1;

        assertThat(vtap.id(), is(VTAP_ID_1));
        assertThat(vtap.vTapCriterion(), is(CRITERION_1));
        assertThat(vtap.type(), is(VTAP_TYPE_1));
    }

    private static OpenstackVtapCriterion createVtapCriterion(IpPrefix srcIp, IpPrefix dstIp,
                                                              TpPort srcPort, TpPort dstPort,
                                                              byte ipProtocol) {
        return DefaultOpenstackVtapCriterion.builder()
                                            .srcIpPrefix(srcIp)
                                            .dstIpPrefix(dstIp)
                                            .srcTpPort(srcPort)
                                            .dstTpPort(dstPort)
                                            .ipProtocol(ipProtocol)
                                            .build();
    }
}