import static org.mockito.Mockito.doAnswer;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
+
+import com.google.common.collect.Lists;
+
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandler;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcerr.message.pcerr.message.Errors;
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);
- Open remote = (Open) this.receivedMsgs.get(0);
- this.serverSession.handleMessage(this.openmsg);
- assertEquals(2, this.receivedMsgs.size());
- assertTrue(this.receivedMsgs.get(1) instanceof Keepalive);
- this.serverSession.handleMessage(Util.createErrorMessage(PCEPErrors.NON_ACC_NEG_SESSION_CHAR, remote.getOpenMessage().getOpen()));
- assertEquals(3, this.receivedMsgs.size());
- assertTrue(this.receivedMsgs.get(2) instanceof Open);
- this.serverSession.handleMessage(this.kamsg);
- assertEquals(this.serverSession.getState(), DefaultPCEPSessionNegotiator.State.Finished);
- }
-
- /**
- * 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());
- }
- }
- }
-
- /**
- * KeepWaitTimer expired.
- *
- * @throws Exception
- */
- @Test
- public void testErrorOneSeven() throws Exception {
- this.serverSession.channelActive(null);
- assertEquals(1, this.receivedMsgs.size());
- assertTrue(this.receivedMsgs.get(0) instanceof Open);
- this.serverSession.handleMessage(this.openmsg);
- Thread.sleep(1000);
- 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) 7), obj.getErrorObject().getValue());
- }
- }
- }
-
- /************* Tests commented because of their long duration (tested timers) **************/
-
- /**
- * OpenWait timer expired.
- *
- * @throws InterruptedException
- */
- @Test
- @Ignore
- public void testErrorOneTwo() throws InterruptedException {
- this.serverSession.channelActive(null);
- assertEquals(1, this.receivedMsgs.size());
- assertTrue(this.receivedMsgs.get(0) instanceof OpenMessage);
- Thread.sleep(60 * 1000);
- 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) 2), obj.getErrorObject().getValue());
- }
- }
- }
-
- @Test
- @Ignore
- public void testUnknownMessage() throws InterruptedException {
- final SimpleSessionListener client = new SimpleSessionListener();
- final PCEPSessionImpl s = new PCEPSessionImpl(new HashedWheelTimer(), client, 5, this.clientListener, this.openmsg.getOpenMessage().getOpen(), this.openmsg.getOpenMessage().getOpen());
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(1, s.unknownMessagesTimes.size());
- Thread.sleep(10000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(2, s.unknownMessagesTimes.size());
- Thread.sleep(10000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(3, s.unknownMessagesTimes.size());
- Thread.sleep(20000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(4, s.unknownMessagesTimes.size());
- Thread.sleep(30000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(3, s.unknownMessagesTimes.size());
- Thread.sleep(10000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(3, s.unknownMessagesTimes.size());
- Thread.sleep(5000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(4, s.unknownMessagesTimes.size());
- Thread.sleep(1000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- assertEquals(5, s.unknownMessagesTimes.size());
- Thread.sleep(1000);
- s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
- synchronized (client) {
- while (client.up) {
- try {
- client.wait();
- } catch (final InterruptedException e) {
- e.printStackTrace();
- }
- }
- }
- assertTrue(!client.up);
- }
-
- @After
- public void tearDown() {
- }
+ 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);
+ Open remote = (Open) this.receivedMsgs.get(0);
+ this.serverSession.handleMessage(this.openmsg);
+ assertEquals(2, this.receivedMsgs.size());
+ assertTrue(this.receivedMsgs.get(1) instanceof Keepalive);
+ this.serverSession.handleMessage(Util.createErrorMessage(PCEPErrors.NON_ACC_NEG_SESSION_CHAR, remote.getOpenMessage().getOpen()));
+ assertEquals(3, this.receivedMsgs.size());
+ assertTrue(this.receivedMsgs.get(2) instanceof Open);
+ this.serverSession.handleMessage(this.kamsg);
+ assertEquals(this.serverSession.getState(), DefaultPCEPSessionNegotiator.State.Finished);
+ }
+
+ /**
+ * 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());
+ }
+ }
+ }
+
+ /**
+ * KeepWaitTimer expired.
+ *
+ * @throws Exception
+ */
+ @Test
+ public void testErrorOneSeven() throws Exception {
+ this.serverSession.channelActive(null);
+ assertEquals(1, this.receivedMsgs.size());
+ assertTrue(this.receivedMsgs.get(0) instanceof Open);
+ this.serverSession.handleMessage(this.openmsg);
+ Thread.sleep(1000);
+ 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) 7), obj.getErrorObject().getValue());
+ }
+ }
+ }
+
+ /************* Tests commented because of their long duration (tested timers) **************/
+
+ /**
+ * OpenWait timer expired.
+ *
+ * @throws InterruptedException
+ */
+ @Test
+ @Ignore
+ public void testErrorOneTwo() throws InterruptedException {
+ this.serverSession.channelActive(null);
+ assertEquals(1, this.receivedMsgs.size());
+ assertTrue(this.receivedMsgs.get(0) instanceof OpenMessage);
+ Thread.sleep(60 * 1000);
+ 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) 2), obj.getErrorObject().getValue());
+ }
+ }
+ }
+
+ @Test
+ @Ignore
+ public void testUnknownMessage() throws InterruptedException {
+ final SimpleSessionListener client = new SimpleSessionListener();
+ final PCEPSessionImpl s = new PCEPSessionImpl(new HashedWheelTimer(), client, 5, this.clientListener, this.openmsg.getOpenMessage().getOpen(), this.openmsg.getOpenMessage().getOpen());
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(1, s.unknownMessagesTimes.size());
+ Thread.sleep(10000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(2, s.unknownMessagesTimes.size());
+ Thread.sleep(10000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(3, s.unknownMessagesTimes.size());
+ Thread.sleep(20000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(4, s.unknownMessagesTimes.size());
+ Thread.sleep(30000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(3, s.unknownMessagesTimes.size());
+ Thread.sleep(10000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(3, s.unknownMessagesTimes.size());
+ Thread.sleep(5000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(4, s.unknownMessagesTimes.size());
+ Thread.sleep(1000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ assertEquals(5, s.unknownMessagesTimes.size());
+ Thread.sleep(1000);
+ s.handleMalformedMessage(PCEPErrors.CAPABILITY_NOT_SUPPORTED);
+ synchronized (client) {
+ while (client.up) {
+ client.wait();
+ }
+ }
+ assertTrue(!client.up);
+ }
+
+ @After
+ public void tearDown() {
+ }
}
Code Review / bgpcep.git / blobdiff