/*
 * Copyright (c) 2013 Cisco Systems, 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.controller.switchmanager;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.Set;

import org.junit.Assert;
import org.junit.Test;
import org.opendaylight.controller.sal.core.Node;
import org.opendaylight.controller.sal.core.NodeConnector;
import org.opendaylight.controller.sal.utils.NodeConnectorCreator;
import org.opendaylight.controller.sal.utils.NodeCreator;

public class SwitchTest {

	@Test
	public void testSwitchCreation() {
		
		Node node = NodeCreator.createOFNode(((long)10));
		NodeConnector nc0 = NodeConnectorCreator.createOFNodeConnector((short)20, node);
		NodeConnector nc1 = NodeConnectorCreator.createOFNodeConnector((short)30, node);
		NodeConnector nc2 = NodeConnectorCreator.createOFNodeConnector((short)40, node);
		NodeConnector nc3 = NodeConnectorCreator.createOFNodeConnector((short)50, node);
		NodeConnector nc4 = NodeConnectorCreator.createOFNodeConnector((short)60, node);
		NodeConnector nc5 = NodeConnectorCreator.createOFNodeConnector((short)70, node);

		Set<NodeConnector> ncSet = new HashSet<NodeConnector>();
		ArrayList<NodeConnector> portList  = new ArrayList<NodeConnector>();
		
		Switch sw = new Switch(node);
		ncSet.add(nc0);
		ncSet.add(nc1);
		ncSet.add(nc2);
		sw.addNodeConnector(nc3);
		
		portList.add(nc4);
		portList.add(nc5);

		sw.setNodeConnectors(ncSet);
		sw.addSpanPorts(portList);
		
		byte[] dlAddress = {(byte)0x01, (byte)0x02, (byte)0x03, (byte)0x04, (byte)0x05, (byte)0x06};
		sw.setDataLayerAddress(dlAddress);
		
		Node resultNode = sw.getNode();
		Set<NodeConnector> resultncSet = sw.getNodeConnectors();
		byte[] resultdlAddress = sw.getDataLayerAddress();		
		ArrayList<NodeConnector> resultSpanPort = (ArrayList<NodeConnector>) sw.getSpanPorts();
		
		Assert.assertEquals(node, resultNode);
		for (int i = 0; i < dlAddress.length; i++)
			Assert.assertEquals(dlAddress[i], resultdlAddress[i]);	
		
		Assert.assertTrue(ncSet.equals(resultncSet));
		
		for (int i = 0; i < portList.size(); i++)
			Assert.assertEquals(portList.get(i), resultSpanPort.get(i));
	}
	
	@Test
	public void testSwitchAddRemovePort() {
		Node node = NodeCreator.createOFNode(((long)10));

		NodeConnector nc0 = NodeConnectorCreator.createOFNodeConnector((short)20, node);
		NodeConnector nc1 = NodeConnectorCreator.createOFNodeConnector((short)30, node);
		NodeConnector nc4 = NodeConnectorCreator.createOFNodeConnector((short)60, node);
		NodeConnector nc5 = NodeConnectorCreator.createOFNodeConnector((short)70, node);
		ArrayList<NodeConnector> portList  = new ArrayList<NodeConnector>();
		
		portList.add(nc4);
		portList.add(nc5);

		Set<NodeConnector> ncSet = new HashSet<NodeConnector>();
		ncSet.add(nc0);
		ncSet.add(nc1);
		
		Switch sw = new Switch(node);
		sw.setNodeConnectors(ncSet);
		sw.removeNodeConnector(nc0);
		Assert.assertFalse(ncSet.contains(nc0));
		
		sw.removeSpanPorts(portList);
		Assert.assertTrue(sw.getSpanPorts().isEmpty());

	}
}