Skip to content

Commit f6b35da

Browse files
committed
Add ResourceLimiterContainer tests
1 parent 99f04e9 commit f6b35da

File tree

2 files changed

+56
-1
lines changed

2 files changed

+56
-1
lines changed

unittest/vslib/Makefile.am

+4-1
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,10 @@ tests_SOURCES = main.cpp \
2424
TestMACsecIngressFilter.cpp \
2525
TestNetMsgRegistrar.cpp \
2626
TestRealObjectIdManager.cpp \
27-
TestResourceLimiter.cpp
27+
TestResourceLimiter.cpp \
28+
TestResourceLimiterContainer.cpp \
29+
TestResourceLimiterParser.cpp \
30+
TestSaiUnittests.cpp
2831

2932
tests_CXXFLAGS = $(DBGFLAGS) $(AM_CXXFLAGS) $(CXXFLAGS_COMMON)
3033
tests_LDADD = $(LDADD_GTEST) $(top_srcdir)/vslib/libSaiVS.a -lhiredis -lswsscommon -lnl-genl-3 -lnl-nf-3 -lnl-route-3 -lnl-3 \
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,52 @@
1+
#include "ResourceLimiterContainer.h"
2+
3+
#include <gtest/gtest.h>
4+
5+
using namespace saivs;
6+
7+
TEST(ResourceLimiterContainer, insert)
8+
{
9+
auto rl = std::make_shared<ResourceLimiter>(0);
10+
11+
ResourceLimiterContainer rlc;
12+
13+
EXPECT_THROW(rlc.insert(0, nullptr), std::runtime_error);
14+
15+
rlc.insert(0, rl);
16+
17+
EXPECT_NE(rlc.getResourceLimiter(0), nullptr);
18+
}
19+
20+
TEST(ResourceLimiterContainer, remove)
21+
{
22+
auto rl = std::make_shared<ResourceLimiter>(0);
23+
24+
ResourceLimiterContainer rlc;
25+
26+
EXPECT_THROW(rlc.insert(0, nullptr), std::runtime_error);
27+
28+
rlc.insert(0, rl);
29+
30+
EXPECT_NE(rlc.getResourceLimiter(0), nullptr);
31+
32+
rlc.remove(0);
33+
34+
EXPECT_EQ(rlc.getResourceLimiter(0), nullptr);
35+
}
36+
37+
TEST(ResourceLimiterContainer, getResourceLimiter)
38+
{
39+
auto rl = std::make_shared<ResourceLimiter>(0);
40+
41+
ResourceLimiterContainer rlc;
42+
43+
EXPECT_THROW(rlc.insert(0, nullptr), std::runtime_error);
44+
45+
rlc.insert(0, rl);
46+
47+
EXPECT_NE(rlc.getResourceLimiter(0), nullptr);
48+
49+
rlc.clear();
50+
51+
EXPECT_EQ(rlc.getResourceLimiter(0), nullptr);
52+
}

0 commit comments

Comments
 (0)