From 219204053555f703c002ccef3adb9283e21f18e9 Mon Sep 17 00:00:00 2001
From: kcudnik <kcudnik@gmail.com>
Date: Mon, 16 Aug 2021 22:41:17 +0200
Subject: [PATCH] Add segment route tests

---
 unittest/vslib/Makefile.am                  |  2 ++
 unittest/vslib/test_sai_vs_scheduler.cpp    | 23 ++++++++++++++++++
 unittest/vslib/test_sai_vs_segmentroute.cpp | 26 +++++++++++++++++++++
 3 files changed, 51 insertions(+)
 create mode 100644 unittest/vslib/test_sai_vs_scheduler.cpp
 create mode 100644 unittest/vslib/test_sai_vs_segmentroute.cpp

diff --git a/unittest/vslib/Makefile.am b/unittest/vslib/Makefile.am
index 7e1e06601d8d..ce05624f09bc 100644
--- a/unittest/vslib/Makefile.am
+++ b/unittest/vslib/Makefile.am
@@ -44,6 +44,8 @@ testslibsaivs_SOURCES =	main_libsaivs.cpp \
 				test_sai_vs_router_rpfgroup.cpp \
 				test_sai_vs_router_samplepacket.cpp \
 				test_sai_vs_schedulergroup.cpp \
+				test_sai_vs_scheduler.cpp \
+				test_sai_vs_segmentroute.cpp \
 				test_sai_vs_vlan.cpp \
 				test_sai_vs_hostif.cpp \
 				test_sai_vs_wred.cpp
diff --git a/unittest/vslib/test_sai_vs_scheduler.cpp b/unittest/vslib/test_sai_vs_scheduler.cpp
new file mode 100644
index 000000000000..abbda8905d61
--- /dev/null
+++ b/unittest/vslib/test_sai_vs_scheduler.cpp
@@ -0,0 +1,23 @@
+#include <gtest/gtest.h>
+
+extern "C" {
+#include "sai.h"
+}
+
+#include "swss/logger.h"
+
+TEST(libsaivs, scheduler)
+{
+    sai_scheduler_api_t *api = nullptr;
+
+    sai_api_query(SAI_API_SCHEDULER, (void**)&api);
+
+    EXPECT_NE(api, nullptr);
+
+    sai_object_id_t id;
+
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->create_scheduler(&id,0,0,0));
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->remove_scheduler(0));
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->set_scheduler_attribute(0,0));
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->get_scheduler_attribute(0,0,0));
+}
diff --git a/unittest/vslib/test_sai_vs_segmentroute.cpp b/unittest/vslib/test_sai_vs_segmentroute.cpp
new file mode 100644
index 000000000000..d4021b9c6875
--- /dev/null
+++ b/unittest/vslib/test_sai_vs_segmentroute.cpp
@@ -0,0 +1,26 @@
+#include <gtest/gtest.h>
+
+extern "C" {
+#include "sai.h"
+}
+
+#include "swss/logger.h"
+
+TEST(libsaivs, segmentroute_sidlist)
+{
+    sai_segmentroute_api_t *api = nullptr;
+
+    sai_api_query(SAI_API_SEGMENTROUTE, (void**)&api);
+
+    EXPECT_NE(api, nullptr);
+
+    sai_object_id_t id;
+
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->create_segmentroute_sidlist(&id,0,0,0));
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->remove_segmentroute_sidlist(0));
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->set_segmentroute_sidlist_attribute(0,0));
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->get_segmentroute_sidlist_attribute(0,0,0));
+
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->create_segmentroute_sidlists(0,0,0,0,SAI_BULK_OP_ERROR_MODE_IGNORE_ERROR,0,0));
+    EXPECT_NE(SAI_STATUS_SUCCESS, api->remove_segmentroute_sidlists(0,0,SAI_BULK_OP_ERROR_MODE_IGNORE_ERROR,0));
+}