<commit_before>/*
 * 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.protocol.pcep.impl;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.DefaultChannelPromise;
import io.netty.util.HashedWheelTimer;
import io.netty.util.concurrent.DefaultPromise;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.net.SocketAddress;
import java.util.Arrays;
import java.util.List;

import org.junit.After;
import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.opendaylight.protocol.pcep.spi.PCEPErrorMapping;
import org.opendaylight.protocol.pcep.spi.PCEPErrors;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Keepalive;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.KeepaliveBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Open;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.OpenBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Pcerr;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.OpenMessage;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.keepalive.message.KeepaliveMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.message.OpenMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObjectBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.Errors;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.ErrorsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.SessionCaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.session._case.SessionBuilder;
import org.opendaylight.yangtools.yang.binding.Notification;

import com.google.common.collect.Lists;

public class FiniteStateMachineTest {

	private DefaultPCEPSessionNegotiator serverSession;

	@Mock
	private Channel clientListener;

	@Mock
	private ChannelPipeline pipeline;

	@Mock
	private SocketAddress address;

	private final List<Notification> receivedMsgs = Lists.newArrayList();

	private Open openmsg;

	private Keepalive kamsg;

	@Before
	public void setUp() {
		MockitoAnnotations.initMocks(this);
		final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open localPrefs = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder().setKeepalive(
				(short) 1).build();
		this.serverSession = new DefaultPCEPSessionNegotiator(new HashedWheelTimer(), new DefaultPromise<PCEPSessionImpl>(GlobalEventExecutor.INSTANCE), this.clientListener, new SimpleSessionListener(), (short) 1, 20, localPrefs);
		final ChannelFuture future = new DefaultChannelPromise(this.clientListener);
		doAnswer(new Answer<Object>() {
			@Override
			public Object answer(final InvocationOnMock invocation) {
				final Object[] args = invocation.getArguments();
				FiniteStateMachineTest.this.receivedMsgs.add((Notification) args[0]);
				return future;
			}
		}).when(this.clientListener).writeAndFlush(any(Notification.class));
		doReturn("TestingChannel").when(this.clientListener).toString();
		doReturn(this.pipeline).when(this.clientListener).pipeline();
		doReturn(this.address).when(this.clientListener).localAddress();
		doReturn(this.address).when(this.clientListener).remoteAddress();
		doReturn(this.pipeline).when(this.pipeline).replace(any(ChannelHandler.class), any(String.class), any(ChannelHandler.class));
		doReturn(true).when(this.clientListener).isActive();
		doReturn(mock(ChannelFuture.class)).when(this.clientListener).close();
		this.openmsg = new OpenBuilder().setOpenMessage(
				new OpenMessageBuilder().setOpen(
						new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder().setDeadTimer(
								(short) 45).setKeepalive((short) 15).build()).build()).build();
		this.kamsg = new KeepaliveBuilder().setKeepaliveMessage(new KeepaliveMessageBuilder().build()).build();
	}

	/**
	 * Both PCEs accept session characteristics. Also tests KeepAliveTimer and error message and when pce attempts to
	 * establish pce session for the 2nd time.
	 * 
	 * @throws Exception
	 */
	@Test
	public void testSessionCharsAccBoth() throws Exception {
		this.serverSession.channelActive(null);
		assertEquals(1, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(0) instanceof Open);
		this.serverSession.handleMessage(this.openmsg);
		assertEquals(2, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(1) instanceof Keepalive);
		this.serverSession.handleMessage(this.kamsg);
		assertEquals(this.serverSession.getState(), DefaultPCEPSessionNegotiator.State.Finished);
	}

	/**
	 * Mock PCE does not accept session characteristics the first time.
	 * 
	 * @throws Exception
	 */
	@Test
	public void testSessionCharsAccMe() throws Exception {
		this.serverSession.channelActive(null);
		assertEquals(1, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(0) instanceof Open);
		this.serverSession.handleMessage(this.openmsg);
		assertEquals(2, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(1) instanceof Keepalive);
		this.serverSession.handleMessage(createErrorMessageWOpen(PCEPErrors.NON_ACC_NEG_SESSION_CHAR));
		assertEquals(3, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(2) instanceof Open);
		this.serverSession.handleMessage(this.kamsg);
		assertEquals(this.serverSession.getState(), DefaultPCEPSessionNegotiator.State.Finished);
	}

	private Pcerr createErrorMessageWOpen(final PCEPErrors e) {
		final PCEPErrorMapping maping = PCEPErrorMapping.getInstance();
		return new PcerrBuilder().setPcerrMessage(
				new PcerrMessageBuilder().setErrorType(
						new SessionCaseBuilder().setSession(
								new SessionBuilder().setOpen(
										new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder().setKeepalive(
												(short) 1).build()).build()).build()).setErrors(
														Arrays.asList(new ErrorsBuilder().setErrorObject(
																new ErrorObjectBuilder().setType(maping.getFromErrorsEnum(e).type).setValue(
																		maping.getFromErrorsEnum(e).value).build()).build())).build()).build();
	}

	/**
	 * Sending different PCEP Message than Open in session establishment phase.
	 * 
	 * @throws Exception
	 */
	@Test
	public void testErrorOneOne() throws Exception {
		this.serverSession.channelActive(null);
		assertEquals(1, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(0) instanceof Open);
		this.serverSession.handleMessage(this.kamsg);
		for (final Notification m : this.receivedMsgs) {
			if (m instanceof Pcerr) {
				final Errors obj = ((Pcerr) m).getPcerrMessage().getErrors().get(0);
				assertEquals(new Short((short) 1), obj.getErrorObject().getType());
				assertEquals(new Short((short) 1), obj.getErrorObject().getValue());
			}
		}
	}<commit_msg>Get rid of mockito exceptions in tests.<commit_after>/*
 * 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.protocol.pcep.impl;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.DefaultChannelPromise;
import io.netty.util.HashedWheelTimer;
import io.netty.util.concurrent.DefaultPromise;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.util.Arrays;
import java.util.List;

import org.junit.After;
import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.opendaylight.protocol.pcep.spi.PCEPErrorMapping;
import org.opendaylight.protocol.pcep.spi.PCEPErrors;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Keepalive;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.KeepaliveBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Open;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.OpenBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Pcerr;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.PcerrBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.OpenMessage;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.keepalive.message.KeepaliveMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.message.OpenMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcep.error.object.ErrorObjectBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.PcerrMessageBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.Errors;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.ErrorsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.SessionCaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.error.type.session._case.SessionBuilder;
import org.opendaylight.yangtools.yang.binding.Notification;

import com.google.common.collect.Lists;

public class FiniteStateMachineTest {

	private DefaultPCEPSessionNegotiator serverSession;

	@Mock
	private Channel clientListener;

	@Mock
	private ChannelPipeline pipeline;

	@Mock
	private SocketAddress address;

	private final List<Notification> receivedMsgs = Lists.newArrayList();

	private Open openmsg;

	private Keepalive kamsg;

	@Before
	public void setUp() {
		MockitoAnnotations.initMocks(this);
		final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open localPrefs = new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder().setKeepalive(
				(short) 1).build();
		this.serverSession = new DefaultPCEPSessionNegotiator(new HashedWheelTimer(), new DefaultPromise<PCEPSessionImpl>(GlobalEventExecutor.INSTANCE), this.clientListener, new SimpleSessionListener(), (short) 1, 20, localPrefs);
		final ChannelFuture future = new DefaultChannelPromise(this.clientListener);
		doAnswer(new Answer<Object>() {
			@Override
			public Object answer(final InvocationOnMock invocation) {
				final Object[] args = invocation.getArguments();
				FiniteStateMachineTest.this.receivedMsgs.add((Notification) args[0]);
				return future;
			}
		}).when(this.clientListener).writeAndFlush(any(Notification.class));
		doReturn("TestingChannel").when(this.clientListener).toString();
		doReturn(this.pipeline).when(this.clientListener).pipeline();
		doReturn(this.address).when(this.clientListener).localAddress();
		doReturn(this.address).when(this.clientListener).remoteAddress();
		doReturn(this.pipeline).when(this.pipeline).replace(any(ChannelHandler.class), any(String.class), any(ChannelHandler.class));
		doReturn(true).when(this.clientListener).isActive();
		doReturn(mock(ChannelFuture.class)).when(this.clientListener).close();
		doReturn(InetSocketAddress.createUnresolved("127.0.0.1", 4189)).when(this.clientListener).remoteAddress();
		doReturn(InetSocketAddress.createUnresolved("127.0.0.1", 4189)).when(this.clientListener).localAddress();
		this.openmsg = new OpenBuilder().setOpenMessage(
				new OpenMessageBuilder().setOpen(
						new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder().setDeadTimer(
								(short) 45).setKeepalive((short) 15).build()).build()).build();
		this.kamsg = new KeepaliveBuilder().setKeepaliveMessage(new KeepaliveMessageBuilder().build()).build();
	}

	/**
	 * Both PCEs accept session characteristics. Also tests KeepAliveTimer and error message and when pce attempts to
	 * establish pce session for the 2nd time.
	 * 
	 * @throws Exception
	 */
	@Test
	public void testSessionCharsAccBoth() throws Exception {
		this.serverSession.channelActive(null);
		assertEquals(1, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(0) instanceof Open);
		this.serverSession.handleMessage(this.openmsg);
		assertEquals(2, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(1) instanceof Keepalive);
		this.serverSession.handleMessage(this.kamsg);
		assertEquals(this.serverSession.getState(), DefaultPCEPSessionNegotiator.State.Finished);
	}

	/**
	 * Mock PCE does not accept session characteristics the first time.
	 * 
	 * @throws Exception
	 */
	@Test
	public void testSessionCharsAccMe() throws Exception {
		this.serverSession.channelActive(null);
		assertEquals(1, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(0) instanceof Open);
		this.serverSession.handleMessage(this.openmsg);
		assertEquals(2, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(1) instanceof Keepalive);
		this.serverSession.handleMessage(createErrorMessageWOpen(PCEPErrors.NON_ACC_NEG_SESSION_CHAR));
		assertEquals(3, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(2) instanceof Open);
		this.serverSession.handleMessage(this.kamsg);
		assertEquals(this.serverSession.getState(), DefaultPCEPSessionNegotiator.State.Finished);
	}

	private Pcerr createErrorMessageWOpen(final PCEPErrors e) {
		final PCEPErrorMapping maping = PCEPErrorMapping.getInstance();
		return new PcerrBuilder().setPcerrMessage(
				new PcerrMessageBuilder().setErrorType(
						new SessionCaseBuilder().setSession(
								new SessionBuilder().setOpen(
										new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.OpenBuilder().setKeepalive(
												(short) 1).build()).build()).build()).setErrors(
						Arrays.asList(new ErrorsBuilder().setErrorObject(
								new ErrorObjectBuilder().setType(maping.getFromErrorsEnum(e).type).setValue(
										maping.getFromErrorsEnum(e).value).build()).build())).build()).build();
	}

	/**
	 * Sending different PCEP Message than Open in session establishment phase.
	 * 
	 * @throws Exception
	 */
	@Test
	public void testErrorOneOne() throws Exception {
		this.serverSession.channelActive(null);
		assertEquals(1, this.receivedMsgs.size());
		assertTrue(this.receivedMsgs.get(0) instanceof Open);
		this.serverSession.handleMessage(this.kamsg);
		for (final Notification m : this.receivedMsgs) {
			if (m instanceof Pcerr) {
				final Errors obj = ((Pcerr) m).getPcerrMessage().getErrors().get(0);
				assertEquals(new Short((short) 1), obj.getErrorObject().getType());
				assertEquals(new Short((short) 1), obj.getErrorObject().getValue());
			}
		}
	}
{'fraction_non_alphanumeric': 0.12525927633095182, 'fraction_numerical': 0.020396404701544135, 'mean_word_length': 5.559712773998489, 'pattern_counts': {'":': 0, '<': 9, '<?xml version=': 0, '>': 9, 'https://': 0, 'lorem ipsum': 0, 'www.': 2, 'xml': 38}, 'pii_count': 2, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '3527601', 'n_tokens_mistral': 6273, 'n_tokens_neox': 5620, 'n_words': 798}