2 * Copyright (c) 2016 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
8 package org.opendaylight.protocol.bgp.evpn.impl;
10 import static org.junit.Assert.assertEquals;
11 import static org.junit.Assert.assertNull;
12 import static org.opendaylight.protocol.bgp.evpn.impl.BGPActivator.EVPN_SAFI;
13 import static org.opendaylight.protocol.bgp.evpn.impl.BGPActivator.L2VPN_AFI;
15 import org.junit.Test;
16 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
17 import org.opendaylight.protocol.bgp.parser.spi.pojo.SimpleBGPExtensionProviderContext;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.EvpnSubsequentAddressFamily;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.evpn.rev180329.L2vpnAddressFamily;
21 public class BGPActivatorTest {
23 public void testActivator() throws Exception {
24 final BGPActivator act = new BGPActivator();
25 final BGPExtensionProviderContext context = new SimpleBGPExtensionProviderContext();
26 assertNull(context.getAddressFamilyRegistry().classForFamily(L2VPN_AFI));
27 assertNull(context.getSubsequentAddressFamilyRegistry().classForFamily(EVPN_SAFI));
29 assertEquals(L2vpnAddressFamily.class, context.getAddressFamilyRegistry().classForFamily(L2VPN_AFI));
30 assertEquals(EvpnSubsequentAddressFamily.class, context.getSubsequentAddressFamilyRegistry()
31 .classForFamily(EVPN_SAFI));