Code clean up
[bgpcep.git] / bgp / mvpn / src / test / java / org / opendaylight / protocol / bgp / mvpn / spi / pojo / nlri / SimpleMvpnNlriRegistryTest.java
1 /*
2  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
3  *
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
7  */
8
9 package org.opendaylight.protocol.bgp.mvpn.spi.pojo.nlri;
10
11 import static org.junit.Assert.assertEquals;
12
13 import io.netty.buffer.ByteBuf;
14 import io.netty.buffer.Unpooled;
15 import java.util.ArrayList;
16 import org.junit.Before;
17 import org.junit.Test;
18 import org.opendaylight.protocol.bgp.mvpn.impl.NlriActivator;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.NlriType;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.mvpn.rev180417.mvpn.MvpnChoice;
21 import org.opendaylight.yangtools.yang.binding.DataContainer;
22
23 public final class SimpleMvpnNlriRegistryTest {
24     @Before
25     public void setUp() {
26         NlriActivator.registerNlriParsers(new ArrayList<>());
27     }
28
29     @Test(expected = IllegalArgumentException.class)
30     public void registryParseTest() {
31         SimpleMvpnNlriRegistry.getInstance().parseMvpn(NlriType.InterAsIPmsiAD, null);
32     }
33
34     @Test
35     public void registryNullTest() {
36         final ByteBuf body = Unpooled.buffer();
37         SimpleMvpnNlriRegistry.getInstance().serializeMvpn(new NotRegistered());
38         assertEquals(0, body.readableBytes());
39     }
40
41     private class NotRegistered implements MvpnChoice {
42         @Override
43         public Class<? extends DataContainer> getImplementedInterface() {
44             return NotRegistered.class;
45         }
46     }
47
48 }