pcap/org.eclipse.tracecompass.pcap.core.tests/src/org/eclipse/tracecompass/pcap/core/tests/protocol/ProtocolTest.java

/*******************************************************************************
 * Copyright (c) 2014 Ericsson
 *
 * All rights reserved. This program and the accompanying materials are
 * made available under the terms of the Eclipse Public License 2.0 which
 * accompanies this distribution, and is available at
 * https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *   Vincent Perot - Initial API and implementation
 *******************************************************************************/

package org.eclipse.tracecompass.pcap.core.tests.protocol;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.fail;

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

import org.eclipse.tracecompass.internal.pcap.core.protocol.PcapProtocol;
import org.junit.Test;

/**
 * JUnit Class that tests whether protocol operation are happening without
 * error.
 *
 * @author Vincent Perot
 */
public class ProtocolTest {

    /**
     * Test that verify if the protocol attributes are as expected.
     */
    @Test
    public void TestProtocolAttributes() {
        assertEquals(PcapProtocol.PCAP.getName(), "Packet Capture");
        assertEquals(PcapProtocol.PCAP.getShortName(), "pcap");
        assertEquals(PcapProtocol.PCAP.getLayer(), PcapProtocol.Layer.LAYER_0);
    }

    /**
     * Test that verify if the protocol getter methods are working properly.
     */
    @Test
    public void TestgetProtocols() {
        List<PcapProtocol> manualListLayer = new ArrayList<>();
        for (PcapProtocol.Layer layer : PcapProtocol.Layer.values()) {
            assertNotNull(layer);
            Collection<PcapProtocol> listLayer = PcapProtocol.getProtocolsOnLayer(layer);

            manualListLayer.clear();
            switch (layer) {
            case LAYER_0:
                manualListLayer.add(PcapProtocol.PCAP);
                break;
            case LAYER_1:
                break;
            case LAYER_2:
                manualListLayer.add(PcapProtocol.ETHERNET_II);
                break;
            case LAYER_3:
                manualListLayer.add(PcapProtocol.IPV4);
                break;
            case LAYER_4:
                manualListLayer.add(PcapProtocol.TCP);
                manualListLayer.add(PcapProtocol.UDP);
                break;
            case LAYER_5:
                break;
            case LAYER_6:
                break;
            case LAYER_7:
                manualListLayer.add(PcapProtocol.UNKNOWN);
                break;
            default:
                fail("Illegal layer value!");
            }
            assertEquals(manualListLayer, listLayer);
        }
    }

}