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.parser.impl;
11 import com.google.common.collect.Lists;
12 import io.netty.buffer.ByteBuf;
13 import io.netty.buffer.Unpooled;
14 import java.util.Collections;
15 import org.junit.Assert;
16 import org.junit.Before;
17 import org.junit.Test;
18 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
19 import org.opendaylight.protocol.bgp.parser.BGPParsingException;
20 import org.opendaylight.protocol.bgp.parser.impl.message.open.GracefulCapabilityHandler;
21 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
22 import org.opendaylight.protocol.bgp.parser.spi.pojo.ServiceLoaderBGPExtensionProviderContext;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.open.message.bgp.parameters.optional.capabilities.CParametersBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.CParameters1Builder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.GracefulRestartCapability.RestartFlags;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.GracefulRestartCapabilityBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.graceful.restart.capability.Tables.AfiFlags;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev180329.mp.capabilities.graceful.restart.capability.TablesBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.AddressFamily;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.Ipv4AddressFamily;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.SubsequentAddressFamily;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev180329.UnicastSubsequentAddressFamily;
35 public class GracefulCapabilityHandlerTest {
37 private BGPExtensionProviderContext ctx;
41 this.ctx = ServiceLoaderBGPExtensionProviderContext.getSingletonInstance();
45 public void testGracefulCapabilityHandler() throws BGPDocumentedException, BGPParsingException {
46 final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(
47 this.ctx.getAddressFamilyRegistry(),this.ctx.getSubsequentAddressFamilyRegistry());
49 final byte[] capaBytes = {
50 (byte) 0x40, (byte) 0x06, (byte) 0x81, (byte) 0xf4,
51 (byte) 0x00, (byte) 0x01, (byte) 0x01, (byte) 0x80
54 final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
55 capaBuilder.setRestartFlags(new RestartFlags(true));
56 capaBuilder.setRestartTime(500);
57 final TablesBuilder tablesBuilder = new TablesBuilder();
58 tablesBuilder.setAfiFlags(new AfiFlags(true));
59 tablesBuilder.setAfi(Ipv4AddressFamily.class);
60 tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
61 capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
63 final ByteBuf buffer = Unpooled.buffer(capaBytes.length);
64 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
65 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(), buffer);
66 Assert.assertArrayEquals(capaBytes, buffer.array());
68 Assert.assertEquals( new CParametersBuilder().addAugmentation(CParameters1.class,
69 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(),
70 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes).slice(2, capaBytes.length - 2)));
72 final byte[] capaBytes2 = {
73 (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
74 (byte) 0x00, (byte) 0x01, (byte) 0x01, (byte) 0x00
76 capaBuilder.setRestartFlags(new RestartFlags(false));
77 capaBuilder.setRestartTime(0);
78 tablesBuilder.setAfiFlags(new AfiFlags(false));
79 capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
81 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
82 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(), buffer);
83 Assert.assertArrayEquals(capaBytes2, buffer.array());
84 Assert.assertEquals( new CParametersBuilder().addAugmentation(CParameters1.class,
85 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(),
86 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes2).slice(2, capaBytes2.length - 2)));
88 capaBuilder.setRestartFlags(null);
89 tablesBuilder.setAfiFlags(null);
90 capaBuilder.setRestartTime(null);
91 capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
93 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
94 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(), buffer);
95 Assert.assertArrayEquals(capaBytes2, buffer.array());
97 final byte[] capaBytes3 = {
98 (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
99 (byte) 0x00, (byte) 0x00, (byte) 0x00, (byte) 0x00
101 capaBuilder.setRestartFlags(new RestartFlags(false));
102 capaBuilder.setRestartTime(0);
103 capaBuilder.setTables(Collections.emptyList());
104 Assert.assertEquals( new CParametersBuilder().addAugmentation(CParameters1.class,
105 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(),
106 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes3).slice(2, capaBytes3.length - 2)));
108 final byte[] capaBytes4 = {
109 (byte) 0x40, (byte) 0x06, (byte) 0x00, (byte) 0x00,
110 (byte) 0x00, (byte) 0x01, (byte) 0x00, (byte) 0x00
112 Assert.assertEquals( new CParametersBuilder().addAugmentation(CParameters1.class,
113 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(),
114 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes4).slice(2, capaBytes4.length - 2)));
115 final byte[] capaBytes5 = {
116 (byte) 0x40, (byte) 0x02, (byte) 0x00, (byte) 0x00
118 final ByteBuf b = Unpooled.buffer(capaBytes5.length);
120 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
121 new CParameters1Builder().setGracefulRestartCapability(new GracefulRestartCapabilityBuilder().build()).build()).build(), b);
122 Assert.assertArrayEquals(capaBytes5, b.array());
123 Assert.assertEquals(new CParametersBuilder().addAugmentation(CParameters1.class, new CParameters1Builder()
124 .setGracefulRestartCapability(new GracefulRestartCapabilityBuilder().setRestartFlags(new RestartFlags(Boolean.FALSE))
125 .setRestartTime(0).setTables(Collections.emptyList()).build()).build()).build(),
126 handler.parseCapability(Unpooled.wrappedBuffer(capaBytes5).slice(2, capaBytes5.length - 2)));
129 @Test(expected=IllegalArgumentException.class)
130 public void testUnhandledAfi() {
131 final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
132 this.ctx.getSubsequentAddressFamilyRegistry());
134 final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
135 capaBuilder.setRestartFlags(new RestartFlags(true));
136 capaBuilder.setRestartTime(50);
137 final TablesBuilder tablesBuilder = new TablesBuilder();
138 tablesBuilder.setAfiFlags(new AfiFlags(true));
140 tablesBuilder.setAfi(AddressFamily.class);
141 tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
142 capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
144 final ByteBuf buffer = Unpooled.buffer();
145 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
146 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(), buffer);
149 @Test(expected=IllegalArgumentException.class)
150 public void testUnhandledSafi() {
151 final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
152 this.ctx.getSubsequentAddressFamilyRegistry());
154 final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
155 capaBuilder.setRestartFlags(new RestartFlags(true));
156 capaBuilder.setRestartTime(50);
157 final TablesBuilder tablesBuilder = new TablesBuilder();
158 tablesBuilder.setAfiFlags(new AfiFlags(true));
159 tablesBuilder.setAfi(Ipv4AddressFamily.class);
160 tablesBuilder.setSafi(SubsequentAddressFamily.class);
161 capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
163 final ByteBuf buffer = Unpooled.buffer();
164 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
165 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(), buffer);
168 @Test(expected=IllegalArgumentException.class)
169 public void testRestartTimeMinValue() {
170 final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
171 this.ctx.getSubsequentAddressFamilyRegistry());
173 final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
174 capaBuilder.setRestartFlags(new RestartFlags(true));
175 capaBuilder.setRestartTime(-1);
176 final TablesBuilder tablesBuilder = new TablesBuilder();
177 tablesBuilder.setAfiFlags(new AfiFlags(true));
178 tablesBuilder.setAfi(Ipv4AddressFamily.class);
179 tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
180 capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
182 final ByteBuf buffer = Unpooled.buffer();
183 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
184 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(), buffer);
187 @Test(expected=IllegalArgumentException.class)
188 public void testRestartTimeMaxValue() {
189 final GracefulCapabilityHandler handler = new GracefulCapabilityHandler(this.ctx.getAddressFamilyRegistry(),
190 this.ctx.getSubsequentAddressFamilyRegistry());
192 final GracefulRestartCapabilityBuilder capaBuilder = new GracefulRestartCapabilityBuilder();
193 capaBuilder.setRestartFlags(new RestartFlags(true));
194 capaBuilder.setRestartTime(50 * 1000);
195 final TablesBuilder tablesBuilder = new TablesBuilder();
196 tablesBuilder.setAfiFlags(new AfiFlags(true));
197 tablesBuilder.setAfi(Ipv4AddressFamily.class);
198 tablesBuilder.setSafi(UnicastSubsequentAddressFamily.class);
199 capaBuilder.setTables(Lists.newArrayList(tablesBuilder.build()));
201 final ByteBuf buffer = Unpooled.buffer();
202 handler.serializeCapability( new CParametersBuilder().addAugmentation(CParameters1.class,
203 new CParameters1Builder().setGracefulRestartCapability(capaBuilder.build()).build()).build(), buffer);