package org.opendaylight.openflowplugin.openflow.md.core.session;

import java.math.BigInteger;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import java.util.concurrent.Future;

import junit.framework.Assert;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
import org.opendaylight.openflowjava.protocol.api.connection.ConnectionReadyListener;
import org.opendaylight.openflowplugin.openflow.md.core.ConnectionConductor;
import org.opendaylight.openflowplugin.openflow.md.core.SwitchConnectionDistinguisher;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.EchoInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.EchoOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.EchoReplyInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.ExperimenterInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FlowModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetAsyncInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetAsyncOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetConfigInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetConfigOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetQueueConfigInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetQueueConfigOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GroupModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.HelloInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MeterModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OpenflowProtocolListener;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketOutInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketOutInputBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.Port;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PortModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PortStatus;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestOutput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.SetAsyncInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.SetConfigInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.TableModInput;
import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.system.rev130927.SystemNotificationsListener;
import org.opendaylight.yangtools.yang.common.RpcResult;

public class MessageDispatchServiceImplTest {

    MockSessionContext session;

    /**
     * @throws java.lang.Exception
     */
    @Before
    public void setUp() throws Exception {
        session = new MockSessionContext(0);

    }

    /**
     * @throws java.lang.Exception
     */
    @After
    public void tearDown() throws Exception {

    }

    /**
     * Test barrier message for null cookie
     *
     * @throws Exception
     */
    @Test
    public void testBarrierMessageForPrimary() throws Exception {
        session.getMessageDispatchService().barrier(null, null);
        Assert.assertEquals(MessageType.BARRIER, session.getPrimaryConductor().getMessageType());
    }

    /**
     * Test packet out message for primary connection
     *
     * @throws Exception
     */
    @Test
    public void testPacketOutMessageForPrimary() throws Exception {
        session.getMessageDispatchService().packetOut(null, null);
        Assert.assertEquals(MessageType.PACKETOUT, session.getPrimaryConductor().getMessageType());
    }

    /**
     * Test packet out message for auxiliary connection
     *
     * @throws Exception
     */
    @Test
    public void testPacketOutMessageForAuxiliary() throws Exception {
        MockConnectionConductor conductor = new MockConnectionConductor(1);
        SwitchConnectionDistinguisher cookie = conductor.getAuxiliaryKey();
        session.addAuxiliaryConductor(cookie, conductor);
        session.getMessageDispatchService().packetOut(null, cookie);
        Assert.assertEquals(MessageType.NONE, session.getPrimaryConductor().getMessageType());
        conductor = (MockConnectionConductor) session.getAuxiliaryConductor(cookie);
        Assert.assertEquals(MessageType.PACKETOUT, conductor.getMessageType());
    }

    /**
     * Test packet out message when multiple auxiliary connection exist
     *
     * @throws Exception
     */
    @Test
    public void testPacketOutMessageForMultipleAuxiliary() throws Exception {
        MockConnectionConductor conductor1 = new MockConnectionConductor(1);
        SwitchConnectionDistinguisher cookie1 = conductor1.getAuxiliaryKey();
        session.addAuxiliaryConductor(cookie1, conductor1);
        MockConnectionConductor conductor2 = new MockConnectionConductor(2);
        SwitchConnectionDistinguisher cookie2 = conductor2.getAuxiliaryKey();
        session.addAuxiliaryConductor(cookie2, conductor2);
        MockConnectionConductor conductor3 = new MockConnectionConductor(3);
        SwitchConnectionDistinguisher cookie3 = conductor3.getAuxiliaryKey();
        session.addAuxiliaryConductor(cookie3, conductor3);
        PacketOutInputBuilder builder = new PacketOutInputBuilder();
        // send message
        session.getMessageDispatchService().packetOut(builder.build(), cookie2);

        Assert.assertEquals(MessageType.NONE, session.getPrimaryConductor().getMessageType());

        conductor3 = (MockConnectionConductor) session.getAuxiliaryConductor(cookie3);
        Assert.assertEquals(MessageType.NONE, conductor3.getMessageType());

        conductor2 = (MockConnectionConductor) session.getAuxiliaryConductor(cookie2);
        Assert.assertEquals(MessageType.PACKETOUT, conductor2.getMessageType());

        conductor1 = (MockConnectionConductor) session.getAuxiliaryConductor(cookie1);
        Assert.assertEquals(MessageType.NONE, conductor1.getMessageType());

    }

    /**
     * Test for invalid session
     *
     * @throws Exception
     */
    @Test
    public void testInvalidSession() throws Exception {
        session.setValid(false);
        try {
            session.getMessageDispatchService().packetOut(null, null);
            Assert.assertTrue(false);
        } catch (IllegalArgumentException ex) {
            Assert.assertTrue(true);
        }
    }

}

class MockSessionContext implements SessionContext {
    private MockConnectionConductor conductor;
    private Map<SwitchConnectionDistinguisher, ConnectionConductor> map;
    private IMessageDispatchService messageService;
    private boolean isValid = true;

    MockSessionContext(int conductorNum) {
        conductor = new MockConnectionConductor(conductorNum);
        map = new HashMap<SwitchConnectionDistinguisher, ConnectionConductor>();
        messageService = new MessageDispatchServiceImpl(this);
    }

    @Override
    public MockConnectionConductor getPrimaryConductor() {
        // TODO Auto-generated method stub
        return conductor;
    }

    @Override
    public GetFeaturesOutput getFeatures() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public ConnectionConductor getAuxiliaryConductor(SwitchConnectionDistinguisher auxiliaryKey) {

        return map.get(auxiliaryKey);
    }

    @Override
    public Set<Entry<SwitchConnectionDistinguisher, ConnectionConductor>> getAuxiliaryConductors() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public void addAuxiliaryConductor(SwitchConnectionDistinguisher auxiliaryKey, ConnectionConductor conductor) {
        map.put(auxiliaryKey, conductor);
    }

    @Override
    public ConnectionConductor removeAuxiliaryConductor(SwitchConnectionDistinguisher connectionCookie) {
        return map.remove(connectionCookie);
    }

    @Override
    public boolean isValid() {
        // TODO Auto-generated method stub
        return isValid;
    }

    @Override
    public void setValid(boolean valid) {
        isValid = valid;
    }

    @Override
    public SwitchConnectionDistinguisher getSessionKey() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public IMessageDispatchService getMessageDispatchService() {
        // TODO Auto-generated method stub
        return messageService;
    }

    @Override
    public Long getNextXid() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Map<Long, Port> getPhysicalPorts() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Set<Long> getPorts() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Port getPhysicalPort(Long portNumber) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Boolean getPortBandwidth(Long portNumber) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public boolean isPortEnabled(long portNumber) {
        // TODO Auto-generated method stub
        return false;
    }

    @Override
    public boolean isPortEnabled(Port port) {
        // TODO Auto-generated method stub
        return false;
    }

    @Override
    public List<Port> getEnabledPorts() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public void processPortStatusMsg(PortStatus arg0) {
        // TODO Auto-generated method stub

    }

}

class MockConnectionConductor implements ConnectionConductor {

    private int conductorNum;
    private MockConnectionAdapter adapter;

    public MockConnectionConductor(int conductorNumber) {
        conductorNum = conductorNumber;
        adapter = new MockConnectionAdapter();
    }

    @Override
    public void init() {
        // TODO Auto-generated method stub

    }

    @Override
    public Short getVersion() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public CONDUCTOR_STATE getConductorState() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public void setConductorState(CONDUCTOR_STATE conductorState) {
        // TODO Auto-generated method stub

    }

    @Override
    public Future<Boolean> disconnect() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public void setSessionContext(SessionContext context) {
        // TODO Auto-generated method stub

    }

    @Override
    public void setConnectionCookie(SwitchConnectionDistinguisher auxiliaryKey) {
        // TODO Auto-generated method stub

    }

    @Override
    public SessionContext getSessionContext() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public SwitchConnectionDistinguisher getAuxiliaryKey() {
        if (0 != conductorNum) {
            SwitchConnectionCookieOFImpl key = new SwitchConnectionCookieOFImpl();
            key.setDatapathId(BigInteger.valueOf(10L));
            key.setAuxiliaryId((short) conductorNum);
            key.initId();
            return key;
        }
        return null;
    }

    @Override
    public ConnectionAdapter getConnectionAdapter() {
        // TODO Auto-generated method stub
        return adapter;
    }

    public MessageType getMessageType() {
        return adapter.getMessageType();
    }
}

enum MessageType {
    NONE, BARRIER, FLOWMOD, TABLEMOD, PACKETOUT;
}

class MockConnectionAdapter implements ConnectionAdapter {

    private MessageType messageType;
    private ConnectionReadyListener connectionReadyListener;

    public MockConnectionAdapter() {
        setMessageType(MessageType.NONE);
    }

    @Override
    public Future<RpcResult<BarrierOutput>> barrier(BarrierInput input) {
        setMessageType(MessageType.BARRIER);
        return null;
    }

    @Override
    public Future<RpcResult<EchoOutput>> echo(EchoInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> echoReply(EchoReplyInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> experimenter(ExperimenterInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> flowMod(FlowModInput input) {
        setMessageType(MessageType.FLOWMOD);
        return null;
    }

    @Override
    public Future<RpcResult<GetAsyncOutput>> getAsync(GetAsyncInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<GetConfigOutput>> getConfig(GetConfigInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<GetFeaturesOutput>> getFeatures(GetFeaturesInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<GetQueueConfigOutput>> getQueueConfig(GetQueueConfigInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> groupMod(GroupModInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> hello(HelloInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> meterMod(MeterModInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> packetOut(PacketOutInput input) {
        setMessageType(MessageType.PACKETOUT);
        return null;
    }

    @Override
    public Future<RpcResult<Void>> portMod(PortModInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<RoleRequestOutput>> roleRequest(RoleRequestInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> setAsync(SetAsyncInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> setConfig(SetConfigInput input) {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public Future<RpcResult<Void>> tableMod(TableModInput input) {
        setMessageType(MessageType.TABLEMOD);
        return null;
    }

    @Override
    public Future<Boolean> disconnect() {
        // TODO Auto-generated method stub
        return null;
    }

    @Override
    public boolean isAlive() {
        // TODO Auto-generated method stub
        return false;
    }

    @Override
    public void setMessageListener(OpenflowProtocolListener messageListener) {
        // TODO Auto-generated method stub

    }

    @Override
    public void setSystemListener(SystemNotificationsListener systemListener) {
        // TODO Auto-generated method stub

    }

    @Override
    public void checkListeners() {
        // TODO Auto-generated method stub

    }

    /**
     * @return the messageType
     */
    public MessageType getMessageType() {
        return messageType;
    }

    /**
     * @param messageType
     *            the messageType to set
     */
    public void setMessageType(MessageType messageType) {
        this.messageType = messageType;
    }

    @Override
    public void fireConnectionReadyNotification() {
            connectionReadyListener.onConnectionReady();
    }

    @Override
    public void setConnectionReadyListener(
            ConnectionReadyListener connectionReadyListener) {
        this.connectionReadyListener = connectionReadyListener;
    }

}