AddInterfaceToLRTest.java

package org.opennaas.extensions.router.junos.actionssets.actions.test.logicalrouters;

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

import java.util.HashMap;

import org.junit.BeforeClass;
import org.junit.Test;
import org.opennaas.core.protocols.sessionmanager.ProtocolManager;
import org.opennaas.core.protocols.sessionmanager.ProtocolSessionManager;
import org.opennaas.core.resources.action.ActionException;
import org.opennaas.core.resources.action.ActionResponse;
import org.opennaas.core.resources.action.ActionResponse.STATUS;
import org.opennaas.core.resources.mock.MockEventManager;
import org.opennaas.core.resources.protocol.ProtocolException;
import org.opennaas.core.resources.protocol.ProtocolSessionContext;
import org.opennaas.extensions.protocols.netconf.NetconfProtocolSessionFactory;
import org.opennaas.extensions.router.junos.actionssets.actions.logicalrouters.AddInterfaceToLogicalRouterAction;
import org.opennaas.extensions.router.junos.actionssets.actions.logicalrouters.RemoveInterfaceFromLogicalRouterAction;
import org.opennaas.extensions.router.model.ComputerSystem;
import org.opennaas.extensions.router.model.EthernetPort;

public class AddInterfaceToLRTest {

	private static ProtocolManager							protocolManager;
	private static ProtocolSessionManager					protocolSessionManager;
	private static ProtocolSessionContext					netconfContext;
	private static String									resourceId	= "testResource";
	private static AddInterfaceToLogicalRouterAction		addAction;
	private static RemoveInterfaceFromLogicalRouterAction	removeAction;

	@BeforeClass
	public static void init() throws ProtocolException {
		addAction = new AddInterfaceToLogicalRouterAction();
		addAction.setModelToUpdate(new ComputerSystem());

		removeAction = new RemoveInterfaceFromLogicalRouterAction();
		removeAction.setModelToUpdate(new ComputerSystem());

		protocolManager = new ProtocolManager();
		protocolSessionManager = (ProtocolSessionManager) protocolManager.getProtocolSessionManager(resourceId);
		protocolSessionManager.setEventManager(new MockEventManager());
		netconfContext = newSessionContextNetconf();
		protocolManager.sessionFactoryAdded(new NetconfProtocolSessionFactory(), new HashMap<String, String>() {
			{
				put(ProtocolSessionContext.PROTOCOL, "netconf");
				put(ProtocolSessionContext.AUTH_TYPE, "password");
			}
		});
		protocolSessionManager.registerContext(netconfContext);
	}

	@Test
	public void testExecuteAdd() throws ActionException {
		addAction.setParams(createLRModelWithAnInterface());
		ActionResponse response = addAction.execute(protocolSessionManager);
		assertNotNull(response);
		assertEquals(STATUS.OK, response.getStatus());
	}

	@Test
	public void testExecuteRemove() throws ActionException {
		removeAction.setParams(createLRModelWithAnInterface());
		ActionResponse response = removeAction.execute(protocolSessionManager);
		assertNotNull(response);
		assertEquals(STATUS.OK, response.getStatus());
	}

	private ComputerSystem createLRModelWithAnInterface() {
		ComputerSystem model = new ComputerSystem();
		model.setElementName("cpe2");

		EthernetPort port = new EthernetPort();
		port.setName("fe-0/0/3");
		port.setPortNumber(0);
		model.addLogicalDevice(port);

		return model;
	}

	private static ProtocolSessionContext newSessionContextNetconf() {
		String uri = System.getProperty("protocol.uri");
		if (uri == null || uri.equals("${protocol.uri}")) {
			uri = "mock://user:pass@host.net:2212/mocksubsystem";
		}

		// FIXME CAUTION REMOVE BEFORE COMMIT!!!!
		// NOT COMMENT IT, REMOVE IT! IT CONTAINS SENSITIVE DATA!!!!
		// uri =

		ProtocolSessionContext protocolSessionContext = new ProtocolSessionContext();

		protocolSessionContext.addParameter(
				ProtocolSessionContext.PROTOCOL_URI, uri);
		protocolSessionContext.addParameter(ProtocolSessionContext.PROTOCOL,
				"netconf");
		protocolSessionContext.addParameter(ProtocolSessionContext.AUTH_TYPE,
				"password");

		return protocolSessionContext;
	}
}