package org.opendaylight.controller.statisticsmanager.internal;

import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.osgi.framework.ServiceReference;
import org.osgi.framework.Bundle;
import javax.inject.Inject;

import org.junit.Assert;
import org.junit.Test;
import org.junit.Before;
import org.junit.runner.RunWith;
import org.opendaylight.controller.forwardingrulesmanager.FlowEntry;
import org.opendaylight.controller.sal.action.Action;
import org.opendaylight.controller.sal.action.Drop;
import org.opendaylight.controller.sal.core.Node;
import org.opendaylight.controller.sal.core.NodeConnector;
import org.opendaylight.controller.sal.flowprogrammer.Flow;
import org.opendaylight.controller.sal.match.Match;
import org.opendaylight.controller.sal.match.MatchType;
import org.opendaylight.controller.sal.reader.FlowOnNode;
import org.opendaylight.controller.sal.reader.NodeConnectorStatistics;
import org.opendaylight.controller.sal.reader.NodeDescription;
import org.opendaylight.controller.sal.utils.NodeCreator;
import org.opendaylight.controller.statisticsmanager.*;
import org.ops4j.pax.exam.junit.PaxExam;
import org.osgi.framework.BundleContext;
import static org.junit.Assert.*;
import org.ops4j.pax.exam.junit.Configuration;
import static org.ops4j.pax.exam.CoreOptions.*;

import org.ops4j.pax.exam.Option;
import org.ops4j.pax.exam.util.PathUtils;

@RunWith(PaxExam.class)
public class StatisticsManagerIntegrationTest {
    private Logger log = LoggerFactory
            .getLogger(StatisticsManagerIntegrationTest.class);
    // get the OSGI bundle context
    @Inject
    private BundleContext bc;

    private IStatisticsManager manager = null;

    // Configure the OSGi container
    @Configuration
    public Option[] config() {
        return options(
                //
                systemProperty("logback.configurationFile").value(
                        "file:" + PathUtils.getBaseDir()
                                + "/src/test/resources/logback.xml"),
                // To start OSGi console for inspection remotely
                systemProperty("osgi.console").value("2401"),
                // Set the systemPackages (used by clustering)
                systemPackages("sun.reflect", "sun.reflect.misc", "sun.misc"),
                // List framework bundles
                mavenBundle("equinoxSDK381", "org.eclipse.equinox.console",
                        "1.0.0.v20120522-1841"),
                mavenBundle("equinoxSDK381", "org.eclipse.equinox.util",
                        "1.0.400.v20120522-2049"),
                mavenBundle("equinoxSDK381", "org.eclipse.osgi.services",
                        "3.3.100.v20120522-1822"),
                mavenBundle("equinoxSDK381", "org.eclipse.equinox.ds",
                        "1.4.0.v20120522-1841"),
                mavenBundle("equinoxSDK381", "org.apache.felix.gogo.command",
                        "0.8.0.v201108120515"),
                mavenBundle("equinoxSDK381", "org.apache.felix.gogo.runtime",
                        "0.8.0.v201108120515"),
                mavenBundle("equinoxSDK381", "org.apache.felix.gogo.shell",
                        "0.8.0.v201110170705"),
                // List logger bundles
                mavenBundle("org.slf4j", "slf4j-api", "1.7.2"),
                mavenBundle("org.slf4j", "log4j-over-slf4j", "1.7.2"),
                mavenBundle("ch.qos.logback", "logback-core", "1.0.9"),
                mavenBundle("ch.qos.logback", "logback-classic", "1.0.9"),
                // List all the bundles on which the test case depends
                mavenBundle("org.opendaylight.controller", "sal",
                        "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller",
                        "sal.implementation", "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller", "statisticsmanager",
                        "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller", "statisticsmanager.implementation",
                                "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller",
                        "protocol_plugins.stub", "0.4.0-SNAPSHOT"),
                // needed by statisticsmanager
                mavenBundle("org.opendaylight.controller", "containermanager",
                        "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller",
                        "containermanager.implementation", "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller",
                        "forwardingrulesmanager", "0.4.0-SNAPSHOT"),

                mavenBundle("org.opendaylight.controller",
                        "clustering.services", "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller",
                        "clustering.services-implementation", "0.4.0-SNAPSHOT"),

                // needed by forwardingrulesmanager
                mavenBundle("org.opendaylight.controller", "switchmanager",
                        "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller", "configuration",
                        "0.4.0-SNAPSHOT"),

                mavenBundle("org.opendaylight.controller",
                        "configuration.implementation", "0.4.0-SNAPSHOT"),
                mavenBundle("org.opendaylight.controller", "hosttracker",
                        "0.4.0-SNAPSHOT"),

                // needed by hosttracker
                mavenBundle("org.opendaylight.controller", "topologymanager",
                        "0.4.0-SNAPSHOT"),

                mavenBundle("org.jboss.spec.javax.transaction",
                        "jboss-transaction-api_1.1_spec", "1.0.1.Final"),
                mavenBundle("org.apache.commons", "commons-lang3", "3.1"),
                mavenBundle("org.apache.felix",
                        "org.apache.felix.dependencymanager", "3.1.0"),
                junitBundles());
    }

    private String stateToString(int state) {
        switch (state) {
        case Bundle.ACTIVE:
            return "ACTIVE";
        case Bundle.INSTALLED:
            return "INSTALLED";
        case Bundle.RESOLVED:
            return "RESOLVED";
        case Bundle.UNINSTALLED:
            return "UNINSTALLED";
        default:
            return "Not CONVERTED";
        }
    }

    @Before
    public void areWeReady() {
        assertNotNull(bc);
        boolean debugit = false;
        Bundle b[] = bc.getBundles();
        for (int i = 0; i < b.length; i++) {
            int state = b[i].getState();
            if (state != Bundle.ACTIVE && state != Bundle.RESOLVED) {
                log.debug("Bundle:" + b[i].getSymbolicName() + " state:"
                        + stateToString(state));
                debugit = true;
            }
        }
        if (debugit) {
            log.debug("Do some debugging because some bundle is "
                    + "unresolved");
        }

        // Assert if true, if false we are good to go!
        assertFalse(debugit);

        ServiceReference r = bc.getServiceReference(IStatisticsManager.class
                .getName());
        if (r != null) {
            this.manager = (IStatisticsManager) bc.getService(r);
        }
        // If StatisticsManager is null, cannot run tests.
        assertNotNull(this.manager);

    }

    @Test
    public void testGetFlows() {
        Node node = NodeCreator.createOFNode(1L);
        List<FlowOnNode> flows = this.manager.getFlows(node);
        FlowOnNode fn = flows.get(0);
        Assert.assertTrue(fn.getByteCount() == 100);
        Assert.assertTrue(fn.getDurationNanoseconds() == 400);
        Assert.assertTrue(fn.getDurationSeconds() == 40);
        Assert.assertTrue(fn.getTableId() == (byte) 0x1);
        Assert.assertTrue(fn.getPacketCount() == 200);

        Match match = new Match();
        try {
            match.setField(MatchType.NW_DST, InetAddress.getByName("1.1.1.1"));
        } catch (UnknownHostException e) {
            fail("Couldn't create match");
        }
        Assert.assertTrue(match.equals(fn.getFlow().getMatch()));
        Assert.assertTrue(fn.getFlow().getActions().get(0).equals(new Drop()));
    }

    @Test
    public void testGetFlowStatistics() {
        Flow flow = new Flow();

        Match match = new Match();
        try {
            match.setField(MatchType.NW_DST, InetAddress.getByName("1.1.1.1"));
        } catch (UnknownHostException e) {
        }
        flow.setMatch(match);
        Action action = new Drop();

        List<Action> actions = new ArrayList<Action>();
        actions.add(action);
        flow.setActions(actions);

        Node node = NodeCreator.createOFNode(1L);
        FlowEntry fe = new FlowEntry("g1", "f1", flow, node);
        List<FlowEntry> list = new ArrayList<FlowEntry>();
        list.add(fe);
        FlowEntry fe2 = new FlowEntry("g1", "f2", flow, node);
        list.add(fe2);

        Map<Node, List<FlowOnNode>> result = this.manager
                .getFlowStatisticsForFlowList(null);
        Assert.assertTrue(result.isEmpty());
        result = this.manager.getFlowStatisticsForFlowList(list);
        List<FlowOnNode> results = result.get(node);
        FlowOnNode fn = results.get(0);
        Assert.assertTrue(fn.getByteCount() == 100);
        Assert.assertTrue(fn.getDurationNanoseconds() == 400);
        Assert.assertTrue(fn.getDurationSeconds() == 40);
        Assert.assertTrue(fn.getTableId() == (byte) 0x1);
        Assert.assertTrue(fn.getPacketCount() == 200);
        Assert.assertTrue(fn.getFlow().equals(flow));

    }

    @Test
    public void testGetFlowsNumber() {
        Node node = NodeCreator.createOFNode(1L);
        Assert.assertTrue(this.manager.getFlowsNumber(node) == 1);
    }

    @Test
    public void testGetNodeDescription() {
        Node node = NodeCreator.createOFNode(1L);
        NodeDescription desc = this.manager.getNodeDescription(node);
        Assert.assertTrue(desc.getDescription().equals(
                "This is a sample node description"));
        Assert.assertTrue(desc.getHardware().equals("stub hardware"));
        Assert.assertTrue(desc.getSoftware().equals("stub software"));
        Assert.assertTrue(desc.getSerialNumber().equals("123"));
        Assert.assertTrue(desc.getManufacturer().equals("opendaylight"));

    }

    @Test
    public void testGetNodeConnectorStatistics() {
        Node node = NodeCreator.createOFNode(1L);
        List<NodeConnectorStatistics> stats = this.manager
                .getNodeConnectorStatistics(node);
        NodeConnectorStatistics ns = stats.get(0);
        Assert.assertTrue(ns.getCollisionCount() == 4);
        Assert.assertTrue(ns.getReceiveByteCount() == 1000);
        Assert.assertTrue(ns.getReceiveCRCErrorCount() == 1);
        Assert.assertTrue(ns.getReceiveDropCount() == 2);
        Assert.assertTrue(ns.getReceiveErrorCount() == 3);
        Assert.assertTrue(ns.getReceiveFrameErrorCount() == 5);
        Assert.assertTrue(ns.getReceiveOverRunErrorCount() == 6);
        Assert.assertTrue(ns.getReceivePacketCount() == 250);
        Assert.assertTrue(ns.getTransmitByteCount() == 5000);
        Assert.assertTrue(ns.getTransmitDropCount() == 50);
        Assert.assertTrue(ns.getTransmitErrorCount() == 10);
        Assert.assertTrue(ns.getTransmitPacketCount() == 500);

        NodeConnector nc = ns.getNodeConnector();
        NodeConnectorStatistics ns2 = this.manager
                .getNodeConnectorStatistics(nc);
        Assert.assertTrue(ns2.getCollisionCount() == 4);
        Assert.assertTrue(ns2.getReceiveByteCount() == 1000);
        Assert.assertTrue(ns2.getReceiveCRCErrorCount() == 1);
        Assert.assertTrue(ns2.getReceiveDropCount() == 2);
        Assert.assertTrue(ns2.getReceiveErrorCount() == 3);
        Assert.assertTrue(ns2.getReceiveFrameErrorCount() == 5);
        Assert.assertTrue(ns2.getReceiveOverRunErrorCount() == 6);
        Assert.assertTrue(ns2.getReceivePacketCount() == 250);
        Assert.assertTrue(ns2.getTransmitByteCount() == 5000);
        Assert.assertTrue(ns2.getTransmitDropCount() == 50);
        Assert.assertTrue(ns2.getTransmitErrorCount() == 10);
        Assert.assertTrue(ns2.getTransmitPacketCount() == 500);
    }

}