TSDRNetflowCollectorImplTest.java

/*
 * Copyright (c) 2015 xFlow Research Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.tsdr.netflow;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mockito;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.tsdr.collector.spi.rev150915.TsdrCollectorSpiService;

/**
 * Unit test for Netflow statistics collector
 * @author Muhammad Umair(muhammad.umair@xflowresearch.com)
 * Created: Dec 30, 2015
 * Modified: Aug 02, 2016
**/

public class TSDRNetflowCollectorImplTest {
    private TsdrCollectorSpiService collectorSPIService = Mockito.mock(TsdrCollectorSpiService.class);
    private TSDRNetflowCollectorImpl implObj;
    private DatagramSocket socket;
    @Before
    public void setup() throws Exception{
        implObj = new TSDRNetflowCollectorImpl(collectorSPIService);
        socket = new DatagramSocket();
    }
    public void sendNetflowData(byte[] data) throws IOException{
        DatagramPacket packet = new DatagramPacket(data, data.length, InetAddress.getByName("127.0.0.1"), 2055);
        socket.send(packet);
    }
    public static byte[] hexStringToByteArray(String s) {
        int len = s.length();
        byte[] data = new byte[len / 2];
        for (int i = 0; i < len - 1; i += 2) {
            data[i / 2] = (byte) ((Character.digit(s.charAt(i), 16) << 4) + Character.digit(s.charAt(i + 1), 16));
        }
        return data;
    }
    @Test
    public void sendPacketsTest() throws InterruptedException, IOException{
        byte[] a = new byte[1000];
        //netflow v5 data
        a = hexStringToByteArray("0005000101bc454657a03ec60000000000000046000000000a0000020a0000030000000000030005000000010000004001bb5ae601bc4546109200500000110100020003201f0000");
        sendNetflowData(a);
        //netflow v9 packet
        a = hexStringToByteArray("000900124a3d1d5857961622000168b700000000010003644a3cb4984a3cb498000000490000000100090000c0a801100808080811009062003500000000000000181000000000004a3cb9f84a3cb9f80000003b00000001000a0000ac10047173babc031100cd20003500000000000000161000000000004a3cbcc84a3cba140000130200000007000a0002ac1004716e5dc2261100cd2101bb000073bab0ad00161000000000004a3cbf884a3cbf880000002800000001000a0002ac1004714ad005060600e46103e1000073bab0ad00161000000000004a3cc1684a3cc1680000002900000001000a0002ac1004716e5dc2390600e49101bb000073bab0ad00161000000000004a3cc6844a3cc6840000004500000001000a0000ac10047173babc031100e448003500000000000000161000000000004a3cc8504a3cc8500000004b00000001000a0000ac10047173babc031100fed0003500000000000000161000000000004a3ccc684a3cc684000000ea00000003000a0000ac100471ac1007ff11000089008900000000000016161000000000004a3ccc344a3cc68c00000c7600000009000a0002ac1004716e5dc20f1100fcf001bb000073bab0ad00161000000000004a3ccd204a3ccd200000003c0000000100090000c0a8011f73babc031100ba09003500000000000000181000000000004a3ccd244a3ccd240000003c0000000100090000c0a8011f73babc03110086aa003500000000000000181000000000004a3ccd284a3ccd28000000320000000100090000c0a8011f73babc031100b1fb003500000000000000181000000000004a3ccd284a3ccd28000000320000000100090000c0a8011f73babc031100db8d003500000000000000181000000000004a3d06704a3ccbf40000007900000003000a0002ac1004716e5dc21e0600e48b01bb000073bab0ad00161100000000004a3cd3b04a3cd3b00000002900000001000a0002ac1004716e5dc21e0600e49201bb000073bab0ad00161000000000004a3cdc304a3cd9a40000057600000003000a0002ac1004714137df210600e4509c59000073bab0ad00161800000000004a3ce1c84a3cde900000013200000003000a0002ac10047189744bf40600e0d701bb000073bab0ad00161800000000004a3ce2a44a3ce2a40000005900000001000a0002ac1004719d3890d71100ec390dd8000073bab0ad0016100000000000");
        sendNetflowData(a);
        Thread.sleep(2000);//2 secs delay for packet(s) transmission
        Assert.assertEquals(2, (int)implObj.getPacketCount());
    }
    @Test
    public void getIncomingNetflowSizeTest(){
        long packetCount = implObj.getIncomingNetflowSize();
        Assert.assertEquals((long)implObj.getPacketCount(), packetCount);
    }
    @After
    public void teardown(){
        this.socket.close();
        implObj = null;
        collectorSPIService = null;
    }
}