Clean up RibSupport registration

[bgpcep.git] / bgp / rib-spi / src / test / java / org / opendaylight / protocol / bgp / rib / spi / RIBSupportTest.java

diff --git a/bgp/rib-spi/src/test/java/org/opendaylight/protocol/bgp/rib/spi/RIBSupportTest.java b/bgp/rib-spi/src/test/java/org/opendaylight/protocol/bgp/rib/spi/RIBSupportTest.java
index 5ee77cc2a981268b6f780b64dadca73b0adbbd4f..f7c8cf33a1f5585884a7d650742a775a852c8ff5 100644 (file)
--- a/bgp/rib-spi/src/test/java/org/opendaylight/protocol/bgp/rib/spi/RIBSupportTest.java
+++ b/bgp/rib-spi/src/test/java/org/opendaylight/protocol/bgp/rib/spi/RIBSupportTest.java
@@ -179,19 +179,14 @@ public class RIBSupportTest extends AbstractConcurrentDataBrokerTest {
     }
 
     @Test
-    public void getAfi() {
-        assertEquals(Ipv4AddressFamily.VALUE, ribSupportTestImp.getAfi());
-    }
-
-    @Test
-    public void getSafi() {
-        assertEquals(UnicastSubsequentAddressFamily.VALUE, ribSupportTestImp.getSafi());
+    public void getTablesKey() {
+        assertEquals(new TablesKey(Ipv4AddressFamily.VALUE, UnicastSubsequentAddressFamily.VALUE),
+            ribSupportTestImp.getTablesKey());
     }
 
     @Test
     public void routesContainerIdentifier() {
-        assertEquals(IPV4_ROUTES_IDENTIFIER,ribSupportTestImp.routesContainerIdentifier());
-
+        assertEquals(IPV4_ROUTES_IDENTIFIER, ribSupportTestImp.routesContainerIdentifier());
     }
 
     @Test