tests: add bgp yang module registery to attr test

Signed-off-by: Chirag Shah <chirag@nvidia.com>

diff --git a/tests/bgpd/test_peer_attr.c b/tests/bgpd/test_peer_attr.c
index 52a83fe13c129db633883592d0b9c73de7675087..0a15886c108858faf35d8bf1328c363e2bc59a8b 100644 (file)
--- a/tests/bgpd/test_peer_attr.c
+++ b/tests/bgpd/test_peer_attr.c
@@ -30,6 +30,8 @@
 #include "bgpd/bgp_vty.h"
 #include "bgpd/bgp_zebra.h"
 #include "bgpd/bgp_network.h"
+#include "lib/routing_nb.h"
+#include "bgpd/bgp_nb.h"
 
 #ifdef ENABLE_BGP_VNC
 #include "bgpd/rfapi/rfapi_backend.h"
@@ -1378,6 +1380,15 @@ static void test_peer_attr(struct test *test, struct test_peer_attr *pa)
        test_process(test, pa, p, g->conf, true, false);
 }
 
+static const struct frr_yang_module_info *const bgpd_yang_modules[] = {
+       &frr_bgp_info,
+       &frr_filter_info,
+       &frr_interface_info,
+       &frr_route_map_info,
+       &frr_routing_info,
+       &frr_vrf_info,
+};
+
 static void bgp_startup(void)
 {
        cmd_init(1);
@@ -1387,7 +1398,7 @@ static void bgp_startup(void)
 
        master = thread_master_create(NULL);
        yang_init(true);
-       nb_init(master, NULL, 0, false);
+       nb_init(master, bgpd_yang_modules, array_size(bgpd_yang_modules), false);
        bgp_master_init(master, BGP_SOCKET_SNDBUF_SIZE);
        bgp_option_set(BGP_OPT_NO_LISTEN);
        vrf_init(NULL, NULL, NULL, NULL, NULL);

diff --git a/tests/subdir.am b/tests/subdir.am
index d7318efc721cf04e403077247bfa9ddabb4f7ccf..211814c1c33b4956e9bd5c50d1b86fda5940e4f4 100644 (file)
--- a/tests/subdir.am
+++ b/tests/subdir.am
@@ -168,6 +168,19 @@ tests_bgpd_test_peer_attr_CFLAGS = $(TESTS_CFLAGS)
 tests_bgpd_test_peer_attr_CPPFLAGS = $(TESTS_CPPFLAGS)
 tests_bgpd_test_peer_attr_LDADD = $(BGP_TEST_LDADD)
 tests_bgpd_test_peer_attr_SOURCES = tests/bgpd/test_peer_attr.c
+nodist_tests_bgpd_test_peer_attr_SOURCES = \
+    yang/frr-bgp-types.yang.c \
+    yang/frr-bgp.yang.c \
+    yang/frr-bgp-common-structure.yang.c \
+    yang/frr-bgp-common.yang.c \
+    yang/frr-bgp-common-multiprotocol.yang.c \
+    yang/frr-bgp-neighbor.yang.c \
+    yang/frr-bgp-peer-group.yang.c \
+    yang/frr-bgp-bmp.yang.c \
+    yang/frr-bgp-rpki.yang.c \
+    yang/frr-deviations-bgp-datacenter.yang.c \
+    # end
+
 
 tests_isisd_test_fuzz_isis_tlv_CFLAGS = $(TESTS_CFLAGS) -I$(top_builddir)/tests/isisd
 tests_isisd_test_fuzz_isis_tlv_CPPFLAGS = $(TESTS_CPPFLAGS) -I$(top_builddir)/tests/isisd