tests: add test for labelmanager plugin

An external label manager plugin may want to use the following
functions:
    - create_label_chunk
    - assign_label_chunk
    - delete_label_chunk
    - release_label_chunk
This test ensures that they are externally visible.

Signed-off-by: Pat Ruddy <pat@voltanet.io>

diff --git a/tests/.gitignore b/tests/.gitignore
index 5e809a81e6014063094746854592c3f5b3fb31e3..b1b8f92a87f1647c529c522aa0b8e7946d529861 100644 (file)
--- a/tests/.gitignore
+++ b/tests/.gitignore
@@ -51,3 +51,4 @@
 /lib/test_zmq
 /ospf6d/test_lsdb
 /ospf6d/test_lsdb_clippy.c
+/zebra/test_lm_plugin
\ No newline at end of file

diff --git a/tests/subdir.am b/tests/subdir.am
index 211814c1c33b4956e9bd5c50d1b86fda5940e4f4..1f173d7f1a5ff2c9d30449a2dfe4ad549881a0ce 100644 (file)
--- a/tests/subdir.am
+++ b/tests/subdir.am
@@ -41,6 +41,16 @@ TESTS_OSPF6D =
 IGNORE_OSPF6D = --ignore=ospf6d/
 endif
 
+if ZEBRA
+TESTS_ZEBRA = \
+       tests/zebra/test_lm_plugin \
+       #end
+IGNORE_ZEBRA =
+else
+TESTS_ZEBRA =
+IGNORE_ZEBRA = --ignore=zebra/
+endif
+
 clippy_scan += \
        tests/lib/cli/test_cli.c \
        tests/ospf6d/test_lsdb.c \
@@ -81,6 +91,7 @@ check_PROGRAMS = \
        $(TESTS_BGPD) \
        $(TESTS_ISISD) \
        $(TESTS_OSPF6D) \
+       $(TESTS_ZEBRA) \
        # end
 
 if ZEROMQ
@@ -135,6 +146,7 @@ ALL_TESTS_LDADD = lib/libfrr.la $(LIBCAP)
 BGP_TEST_LDADD = bgpd/libbgp.a $(RFPLDADD) $(ALL_TESTS_LDADD) -lm
 ISISD_TEST_LDADD = isisd/libisis.a $(ALL_TESTS_LDADD)
 OSPF6_TEST_LDADD = ospf6d/libospf6.a $(ALL_TESTS_LDADD)
+ZEBRA_TEST_LDADD = zebra/label_manager.o $(ALL_TESTS_LDADD)
 
 tests_bgpd_test_aspath_CFLAGS = $(TESTS_CFLAGS)
 tests_bgpd_test_aspath_CPPFLAGS = $(TESTS_CPPFLAGS)
@@ -336,6 +348,11 @@ tests_ospf6d_test_lsdb_CPPFLAGS = $(TESTS_CPPFLAGS)
 tests_ospf6d_test_lsdb_LDADD = $(OSPF6_TEST_LDADD)
 tests_ospf6d_test_lsdb_SOURCES = tests/ospf6d/test_lsdb.c tests/lib/cli/common_cli.c
 
+tests_zebra_test_lm_plugin_CFLAGS = $(TESTS_CFLAGS)
+tests_zebra_test_lm_plugin_CPPFLAGS = $(TESTS_CPPFLAGS)
+tests_zebra_test_lm_plugin_LDADD = $(ZEBRA_TEST_LDADD)
+tests_zebra_test_lm_plugin_SOURCES = tests/zebra/test_lm_plugin.c
+
 EXTRA_DIST += \
        tests/runtests.py \
        tests/bgpd/test_aspath.py \
@@ -383,6 +400,8 @@ EXTRA_DIST += \
        tests/ospf6d/test_lsdb.py \
        tests/ospf6d/test_lsdb.in \
        tests/ospf6d/test_lsdb.refout \
+       tests/zebra/test_lm_plugin.py \
+       tests/zebra/test_lm_plugin.refout \
        # end
 
 .PHONY: tests/tests.xml

diff --git a/tests/zebra/test_lm_plugin.c b/tests/zebra/test_lm_plugin.c
new file mode 100644 (file)
index 0000000..4a9344f
--- /dev/null
+++ b/tests/zebra/test_lm_plugin.c
@@ -0,0 +1,134 @@
+/*
+ * Label Manager tests.
+ * Copyright (C) 2020 Volta Networks
+ *                    Patrick Ruddy
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License as published by the Free
+ * Software Foundation; either version 2 of the License, or (at your option)
+ * any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; see the file COPYING; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+
+#include <zebra.h>
+#include "zebra/zapi_msg.h"
+#include "zebra/label_manager.h"
+
+/* shim out unused functions/variables to allow the lablemanager to compile*/
+DEFINE_KOOH(zserv_client_close, (struct zserv * client), (client));
+unsigned long zebra_debug_packet = 0;
+struct zserv *zserv_find_client_session(uint8_t proto, unsigned short instance,
+                                       uint32_t session_id)
+{
+       return NULL;
+}
+
+int zsend_label_manager_connect_response(struct zserv *client, vrf_id_t vrf_id,
+                                        unsigned short result)
+{
+       return 0;
+}
+
+int zsend_assign_label_chunk_response(struct zserv *client, vrf_id_t vrf_id,
+                                     struct label_manager_chunk *lmc)
+{
+       return 0;
+}
+
+
+static int test_client_connect(struct zserv *client, vrf_id_t vrf_id)
+{
+       return 0;
+}
+
+static int test_client_disconnect(struct zserv *client)
+{
+       return 0;
+}
+
+/* external test hook functions */
+static int lm_get_chunk_pi(struct label_manager_chunk **lmc,
+                          struct zserv *client, uint8_t keep, uint32_t size,
+                          uint32_t base, vrf_id_t vrf_id)
+{
+       if (base == 0)
+               *lmc = create_label_chunk(10, 55, 0, 1, 50, 50 + size);
+       else
+               *lmc = assign_label_chunk(10, 55, 0, 1, size, base);
+
+       return 0;
+}
+
+static int lm_release_chunk_pi(struct zserv *client, uint32_t start,
+                              uint32_t end)
+{
+       return release_label_chunk(client->proto, client->instance,
+                                  client->session_id, start, end);
+}
+
+
+/* use external allocations */
+static void lp_plugin_init()
+{
+       /* register our own hooks */
+       hook_register(lm_client_connect, test_client_connect);
+       hook_register(lm_client_disconnect, test_client_disconnect);
+       hook_register(lm_get_chunk, lm_get_chunk_pi);
+       hook_register(lm_release_chunk, lm_release_chunk_pi);
+}
+
+static void lp_plugin_cleanup()
+{
+       /* register our own hooks */
+       hook_unregister(lm_client_connect, test_client_connect);
+       hook_unregister(lm_client_disconnect, test_client_disconnect);
+       hook_unregister(lm_get_chunk, lm_get_chunk_pi);
+       hook_unregister(lm_release_chunk, lm_release_chunk_pi);
+}
+
+
+/* tests */
+
+static void test_lp_plugin()
+{
+       struct label_manager_chunk *lmc;
+
+       lmc = assign_label_chunk(10, 55, 0, 1, 50, 0);
+       fprintf(stdout,
+               "chunk: start %u end %u proto %u instance %u session %u keep %s\n",
+               lmc->start, lmc->end, lmc->proto, lmc->instance,
+               lmc->session_id, lmc->keep ? "yes" : "no");
+       delete_label_chunk(lmc);
+
+       lmc = assign_label_chunk(10, 55, 0, 1, 50, 100);
+       fprintf(stdout,
+               "chunk: start %u end %u proto %u instance %u session %u keep %s\n",
+               lmc->start, lmc->end, lmc->proto, lmc->instance,
+               lmc->session_id, lmc->keep ? "yes" : "no");
+       release_label_chunk(10, 55, 0, lmc->start, lmc->end);
+}
+
+int main(int argc, char **argv)
+{
+       /* set up label manager and release it's hooks */
+       label_manager_init();
+       lm_hooks_unregister();
+
+       /* test plugin */
+       lp_plugin_init();
+       test_lp_plugin();
+       lp_plugin_cleanup();
+
+       /* this keeps the compiler happy */
+       hook_call(zserv_client_close, NULL);
+       return 0;
+}

diff --git a/tests/zebra/test_lm_plugin.py b/tests/zebra/test_lm_plugin.py
new file mode 100644 (file)
index 0000000..bf4f3ce
--- /dev/null
+++ b/tests/zebra/test_lm_plugin.py
@@ -0,0 +1,5 @@
+import frrtest
+
+
+class TestLmplugin(frrtest.TestRefOut):
+    program = "./test_lm_plugin"

diff --git a/tests/zebra/test_lm_plugin.refout b/tests/zebra/test_lm_plugin.refout
new file mode 100644 (file)
index 0000000..35824f1
--- /dev/null
+++ b/tests/zebra/test_lm_plugin.refout
@@ -0,0 +1,2 @@
+chunk: start 16 end 65 proto 10 instance 55 session 0 keep yes
+chunk: start 100 end 149 proto 10 instance 55 session 0 keep yes