2 * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.protocol.bgp.rib.impl;
11 import static org.mockito.Matchers.any;
12 import static org.mockito.Mockito.doAnswer;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
16 import com.google.common.collect.Lists;
17 import io.netty.channel.Channel;
18 import io.netty.channel.ChannelFuture;
19 import io.netty.channel.ChannelHandler;
20 import io.netty.channel.ChannelPipeline;
21 import io.netty.channel.EventLoop;
22 import io.netty.util.concurrent.Future;
23 import io.netty.util.concurrent.GenericFutureListener;
24 import io.netty.util.concurrent.GlobalEventExecutor;
25 import java.net.InetAddress;
26 import java.net.InetSocketAddress;
27 import java.net.UnknownHostException;
28 import java.util.List;
29 import java.util.concurrent.TimeUnit;
30 import org.junit.Assert;
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.mockito.Mock;
34 import org.mockito.Mockito;
35 import org.mockito.MockitoAnnotations;
36 import org.mockito.invocation.InvocationOnMock;
37 import org.mockito.stubbing.Answer;
38 import org.opendaylight.protocol.bgp.parser.BGPError;
39 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.KeepaliveBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Notify;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.NotifyBuilder;
45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Open;
46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.OpenBuilder;
47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.ProtocolVersion;
48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Update;
49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.UpdateBuilder;
50 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParameters;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParametersBuilder;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.c.parameters.As4BytesCaseBuilder;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.bgp.parameters.c.parameters.as4.bytes._case.As4BytesCapabilityBuilder;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.c.parameters.MultiprotocolCaseBuilder;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.c.parameters.multiprotocol._case.MultiprotocolCapabilityBuilder;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Ipv4AddressFamily;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.UnicastSubsequentAddressFamily;
59 import org.opendaylight.yangtools.yang.binding.Notification;
61 public class BGPSessionImplTest {
63 private static final int HOLD_TIMER = 3;
64 private static final AsNumber AS_NUMBER = new AsNumber(30L);
65 private static final Ipv4Address BGP_ID = new Ipv4Address("1.1.1.2");
68 private EventLoop eventLoop;
71 private Channel speakerListener;
74 private ChannelPipeline pipeline;
76 private final BgpTableType ipv4tt = new BgpTableTypeImpl(Ipv4AddressFamily.class, UnicastSubsequentAddressFamily.class);
78 private final List<Notification> receivedMsgs = Lists.newArrayList();
80 private Open classicOpen;
82 private BGPSessionImpl bgpSession;
84 private SimpleSessionListener listener;
87 public void setUp() throws UnknownHostException {
88 MockitoAnnotations.initMocks(this);
89 final List<BgpParameters> tlvs = Lists.newArrayList();
90 this.classicOpen = new OpenBuilder().setMyAsNumber(AS_NUMBER.getValue().intValue()).setHoldTimer(HOLD_TIMER).setVersion(new ProtocolVersion((short) 4)).setBgpParameters(
91 tlvs).setBgpIdentifier(BGP_ID).build();
93 tlvs.add(new BgpParametersBuilder().setCParameters(
94 new MultiprotocolCaseBuilder().setMultiprotocolCapability(
95 new MultiprotocolCapabilityBuilder().setAfi(this.ipv4tt.getAfi()).setSafi(this.ipv4tt.getSafi()).build()).build()).build());
96 tlvs.add(new BgpParametersBuilder().setCParameters(new As4BytesCaseBuilder().setAs4BytesCapability(new As4BytesCapabilityBuilder().setAsNumber(
97 AS_NUMBER).build()).build()).build());
99 final ChannelFuture f = mock(ChannelFuture.class);
100 doReturn(null).when(f).addListener(Mockito.<GenericFutureListener<? extends Future<? super Void>>>any());
102 doAnswer(new Answer<Object>() {
104 public Object answer(final InvocationOnMock invocation) {
105 final Object[] args = invocation.getArguments();
106 BGPSessionImplTest.this.receivedMsgs.add((Notification) args[0]);
109 }).when(this.speakerListener).writeAndFlush(any(Notification.class));
110 doReturn(this.eventLoop).when(this.speakerListener).eventLoop();
111 doAnswer(new Answer<Void>() {
113 public Void answer(InvocationOnMock invocation) throws Throwable {
114 final Runnable command = (Runnable) invocation.getArguments()[0];
115 final long delay = (long) invocation.getArguments()[1];
116 final TimeUnit unit = (TimeUnit) invocation.getArguments()[2];
117 GlobalEventExecutor.INSTANCE.schedule(command, delay, unit);
120 }).when(this.eventLoop).schedule(any(Runnable.class), any(long.class), any(TimeUnit.class));
121 doReturn("TestingChannel").when(this.speakerListener).toString();
122 doReturn(new InetSocketAddress(InetAddress.getByName(BGP_ID.getValue()), 179)).when(this.speakerListener).remoteAddress();
123 doReturn(this.pipeline).when(this.speakerListener).pipeline();
124 doReturn(this.pipeline).when(this.pipeline).replace(any(ChannelHandler.class), any(String.class), any(ChannelHandler.class));
125 doReturn(mock(ChannelFuture.class)).when(this.speakerListener).close();
126 this.listener = new SimpleSessionListener();
127 this.bgpSession = new BGPSessionImpl(this.listener, this.speakerListener, this.classicOpen, this.classicOpen.getHoldTimer());
131 public void testBGPSession() {
132 this.bgpSession.sessionUp();
133 Assert.assertEquals(BGPSessionImpl.State.Up, this.bgpSession.getState());
134 Assert.assertEquals(AS_NUMBER, this.bgpSession.getAsNumber());
135 Assert.assertEquals(BGP_ID, this.bgpSession.getBgpId());
136 Assert.assertEquals(1, this.bgpSession.getAdvertisedTableTypes().size());
137 Assert.assertTrue(this.listener.up);
139 this.bgpSession.handleMessage(new UpdateBuilder().build());
140 Assert.assertEquals(1, this.listener.getListMsg().size());
141 Assert.assertTrue(this.listener.getListMsg().get(0) instanceof Update);
143 this.bgpSession.handleMessage(new KeepaliveBuilder().build());
144 this.bgpSession.handleMessage(new KeepaliveBuilder().build());
146 this.bgpSession.close();
147 Assert.assertEquals(BGPSessionImpl.State.Idle, this.bgpSession.getState());
148 Assert.assertEquals(1, this.receivedMsgs.size());
149 Assert.assertTrue(this.receivedMsgs.get(0) instanceof Notify);
150 final Notify error = (Notify) this.receivedMsgs.get(0);
151 Assert.assertEquals(BGPError.CEASE.getCode(), error.getErrorCode().shortValue());
152 Mockito.verify(this.speakerListener).close();
156 public void testHandleOpenMsg() {
157 this.bgpSession.handleMessage(this.classicOpen);
158 Assert.assertEquals(BGPSessionImpl.State.Idle, this.bgpSession.getState());
159 Assert.assertEquals(1, this.receivedMsgs.size());
160 Assert.assertTrue(this.receivedMsgs.get(0) instanceof Notify);
161 final Notify error = (Notify) this.receivedMsgs.get(0);
162 Assert.assertEquals(BGPError.FSM_ERROR.getCode(), error.getErrorCode().shortValue());
163 Assert.assertEquals(BGPError.FSM_ERROR.getSubcode(), error.getErrorSubcode().shortValue());
164 Mockito.verify(this.speakerListener).close();
168 public void testHandleNotifyMsg() {
169 this.bgpSession.handleMessage(new NotifyBuilder().setErrorCode(BGPError.BAD_BGP_ID.getCode()).setErrorSubcode(BGPError.BAD_BGP_ID.getSubcode()).build());
170 Assert.assertEquals(BGPSessionImpl.State.Idle, this.bgpSession.getState());
171 Mockito.verify(this.speakerListener).close();
175 public void testEndOfInput() {
176 this.bgpSession.sessionUp();
177 Assert.assertFalse(this.listener.down);
178 this.bgpSession.endOfInput();
179 Assert.assertTrue(this.listener.down);
183 public void testHoldTimerExpire() throws InterruptedException {
184 this.bgpSession.sessionUp();
186 Assert.assertEquals(BGPSessionImpl.State.Idle, this.bgpSession.getState());
187 Assert.assertEquals(3, this.receivedMsgs.size());
188 Assert.assertTrue(this.receivedMsgs.get(2) instanceof Notify);
189 final Notify error = (Notify) this.receivedMsgs.get(2);
190 Assert.assertEquals(BGPError.HOLD_TIMER_EXPIRED.getCode(), error.getErrorCode().shortValue());
191 Assert.assertEquals(BGPError.HOLD_TIMER_EXPIRED.getSubcode(), error.getErrorSubcode().shortValue());
192 Mockito.verify(this.speakerListener).close();