Skip to content

Commit

Permalink
Add SwitchMLNX2700 tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@kcudnik
kcudnik committed Aug 23, 2021
1 parent 2e418ab commit 4d7956b
Show file tree
Hide file tree
Showing 5 changed files with 4,367 additions and 3 deletions.
5 changes: 3 additions & 2 deletions unittest/vslib/Makefile.am
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
AM_CXXFLAGS = $(SAIINC) -I$(top_srcdir)/meta -I$(top_srcdir)/vslib -I/usr/include/libnl3
AM_CXXFLAGS = $(SAIINC) -I$(top_srcdir)/meta -I$(top_srcdir)/vslib -I$(top_srcdir)/lib -I/usr/include/libnl3

bin_PROGRAMS = tests testslibsaivs

Expand Down Expand Up @@ -36,7 +36,8 @@ tests_SOURCES = main.cpp \
TestTrafficFilterPipes.cpp \
TestSwitchConfig.cpp \
TestSwitchContainer.cpp \
TestSwitch.cpp
TestSwitch.cpp \
TestSwitchMLNX2700.cpp

tests_CXXFLAGS = $(DBGFLAGS) $(AM_CXXFLAGS) $(CXXFLAGS_COMMON)
tests_LDADD = $(LDADD_GTEST) $(top_srcdir)/vslib/libSaiVS.a -lhiredis -lswsscommon -lnl-genl-3 -lnl-nf-3 -lnl-route-3 -lnl-3 \
Expand Down
42 changes: 42 additions & 0 deletions unittest/vslib/TestHostInterfaceInfo.cpp
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
#include "HostInterfaceInfo.h"

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include <gtest/gtest.h>

using namespace saivs;
Expand All @@ -24,3 +29,40 @@ TEST(HostInterfaceInfo, sendTo)

EXPECT_EQ(hii.sendTo(s, buf, 0), false);
}

TEST(HostInterfaceInfo, async_process_packet_for_fdb_event)
{
auto eq = std::make_shared<EventQueue>(std::make_shared<Signal>());

int s = socket(AF_INET, SOCK_DGRAM, 0);
int fd = socket(AF_INET, SOCK_DGRAM, 0);

HostInterfaceInfo hii(0, s, fd, "tap", 0, eq);

usleep(100*1000); // give some time to start thread

struct sockaddr_in server;

memset(&server, 0, sizeof(server));

server.sin_family = AF_INET;
server.sin_addr.s_addr = inet_addr("127.0.0.1");
server.sin_port = htons(4000);

unsigned char buffer[200];

ssize_t res = sendto(fd, buffer, 200, 0,
(const struct sockaddr*)&server, sizeof(server));

EXPECT_FALSE(res < 0);

res = sendto(s, buffer, 200, 0,
(const struct sockaddr*)&server, sizeof(server));

EXPECT_FALSE(res < 0);

usleep(100*1000); // give some time to process async

close(s);
close(fd);
}
267 changes: 267 additions & 0 deletions unittest/vslib/TestSwitchMLNX2700.cpp
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,267 @@
#include "SwitchMLNX2700.h"

#include "meta/sai_serialize.h"

#include <gtest/gtest.h>

using namespace saivs;

TEST(SwitchMLNX2700, ctr)
{
auto sc = std::make_shared<SwitchConfig>(0, "");
auto signal = std::make_shared<Signal>();
auto eventQueue = std::make_shared<EventQueue>(signal);

sc->m_saiSwitchType = SAI_SWITCH_TYPE_NPU;
sc->m_switchType = SAI_VS_SWITCH_TYPE_MLNX2700;
sc->m_bootType = SAI_VS_BOOT_TYPE_COLD;
sc->m_useTapDevice = false;
sc->m_laneMap = LaneMap::getDefaultLaneMap(0);
sc->m_eventQueue = eventQueue;

auto scc = std::make_shared<SwitchConfigContainer>();

scc->insert(sc);

SwitchMLNX2700 sw(
0x2100000000,
std::make_shared<RealObjectIdManager>(0, scc),
sc);

SwitchMLNX2700 sw2(
0x2100000000,
std::make_shared<RealObjectIdManager>(0, scc),
sc,
nullptr);

sai_attribute_t attr;

attr.id = SAI_SWITCH_ATTR_INIT_SWITCH;
attr.value.booldata = true;

EXPECT_EQ(sw.initialize_default_objects(1, &attr), SAI_STATUS_SUCCESS);
}

TEST(SwitchMLNX2700, refresh_bridge_port_list)
{
auto sc = std::make_shared<SwitchConfig>(0, "");
auto signal = std::make_shared<Signal>();
auto eventQueue = std::make_shared<EventQueue>(signal);

sc->m_saiSwitchType = SAI_SWITCH_TYPE_NPU;
sc->m_switchType = SAI_VS_SWITCH_TYPE_MLNX2700;
sc->m_bootType = SAI_VS_BOOT_TYPE_COLD;
sc->m_useTapDevice = false;
sc->m_laneMap = LaneMap::getDefaultLaneMap(0);
sc->m_eventQueue = eventQueue;

auto scc = std::make_shared<SwitchConfigContainer>();

scc->insert(sc);

SwitchMLNX2700 sw(
0x2100000000,
std::make_shared<RealObjectIdManager>(0, scc),
sc);

sai_attribute_t attr;

attr.id = SAI_SWITCH_ATTR_INIT_SWITCH;
attr.value.booldata = true;

EXPECT_EQ(sw.initialize_default_objects(1, &attr), SAI_STATUS_SUCCESS);

attr.id = SAI_SWITCH_ATTR_DEFAULT_1Q_BRIDGE_ID;
attr.value.oid = SAI_NULL_OBJECT_ID;

EXPECT_EQ(sw.get(SAI_OBJECT_TYPE_SWITCH, "oid:0x2100000000", 1, &attr), SAI_STATUS_SUCCESS);

EXPECT_NE(attr.value.oid, SAI_NULL_OBJECT_ID);

auto boid = attr.value.oid;

auto sboid = sai_serialize_object_id(boid);

sai_object_id_t list[128];

attr.id = SAI_BRIDGE_ATTR_PORT_LIST;

attr.value.objlist.count = 128;
attr.value.objlist.list = list;

EXPECT_EQ(sw.get(SAI_OBJECT_TYPE_BRIDGE, sboid, 1, &attr), SAI_STATUS_SUCCESS);
}

static std::map<sai_object_id_t, WarmBootState> g_warmBootState;

// TODO move to utils
static bool getWarmBootState(
_In_ const char* warmBootFile,
_In_ std::shared_ptr<RealObjectIdManager> roidm)
{
SWSS_LOG_ENTER();

std::ifstream ifs;

ifs.open(warmBootFile);

if (!ifs.is_open())
{
SWSS_LOG_ERROR("failed to open: %s", warmBootFile);

return false;
}

std::string line;

while (std::getline(ifs, line))
{
SWSS_LOG_DEBUG("line: %s", line.c_str());

// line format: OBJECT_TYPE OBJECT_ID ATTR_ID ATTR_VALUE
std::istringstream iss(line);

std::string strObjectType;
std::string strObjectId;
std::string strAttrId;
std::string strAttrValue;

iss >> strObjectType >> strObjectId;

if (strObjectType == SAI_VS_FDB_INFO)
{
/*
* If we read line from fdb info set and use tap device is enabled
* just parse line and repopulate fdb info set.
*/

FdbInfo fi = FdbInfo::deserialize(strObjectId);

auto switchId = roidm->switchIdQuery(fi.m_portId);

if (switchId == SAI_NULL_OBJECT_ID)
{
SWSS_LOG_ERROR("switchIdQuery returned NULL on fi.m_port = %s",
sai_serialize_object_id(fi.m_portId).c_str());

g_warmBootState.clear();
return false;
}

g_warmBootState[switchId].m_switchId = switchId;

g_warmBootState[switchId].m_fdbInfoSet.insert(fi);

continue;
}

iss >> strAttrId >> strAttrValue;

sai_object_meta_key_t metaKey;
sai_deserialize_object_meta_key(strObjectType + ":" + strObjectId, metaKey);

/*
* Since all objects we are creating, then during warm boot we need to
* get the biggest object index, so after warm boot we can start
* generating new objects with index value not colliding with objects
* loaded from warm boot scenario. We only need to consider OID
* objects.
*/

roidm->updateWarmBootObjectIndex(metaKey.objectkey.key.object_id);

// query each object for switch id

auto switchId = roidm->switchIdQuery(metaKey.objectkey.key.object_id);

if (switchId == SAI_NULL_OBJECT_ID)
{
SWSS_LOG_ERROR("switchIdQuery returned NULL on oid = %s",
sai_serialize_object_id(metaKey.objectkey.key.object_id).c_str());

g_warmBootState.clear();
return false;
}

g_warmBootState[switchId].m_switchId = switchId;

auto &objectHash = g_warmBootState[switchId].m_objectHash[metaKey.objecttype]; // will create if not exist

if (objectHash.find(strObjectId) == objectHash.end())
{
objectHash[strObjectId] = {};
}

if (strAttrId == "NULL")
{
// skip empty attributes
continue;
}

objectHash[strObjectId][strAttrId] =
std::make_shared<SaiAttrWrap>(strAttrId, strAttrValue);
}

// NOTE notification pointers should be restored by attr_list when creating switch

ifs.close();

return true;
}

TEST(SwitchMLNX2700, warm_update_queues)
{
auto sc = std::make_shared<SwitchConfig>(0, "");
auto signal = std::make_shared<Signal>();
auto eventQueue = std::make_shared<EventQueue>(signal);

sc->m_saiSwitchType = SAI_SWITCH_TYPE_NPU;
sc->m_switchType = SAI_VS_SWITCH_TYPE_MLNX2700;
sc->m_bootType = SAI_VS_BOOT_TYPE_COLD;
sc->m_useTapDevice = false;
sc->m_laneMap = LaneMap::getDefaultLaneMap(0);
sc->m_eventQueue = eventQueue;

auto scc = std::make_shared<SwitchConfigContainer>();

scc->insert(sc);

auto roidm = std::make_shared<RealObjectIdManager>(0, scc);

EXPECT_TRUE(getWarmBootState("files/mlnx2700.warm.bin", roidm));

auto warmBootState = std::make_shared<WarmBootState>(g_warmBootState.at(0x2100000000)); // copy ctr

SwitchMLNX2700 sw(
0x2100000000,
roidm,
sc,
warmBootState);

sai_attribute_t attr;

attr.id = SAI_SWITCH_ATTR_INIT_SWITCH;
attr.value.booldata = true;

EXPECT_EQ(sw.warm_boot_initialize_objects(), SAI_STATUS_SUCCESS);

attr.id = SAI_SWITCH_ATTR_DEFAULT_1Q_BRIDGE_ID;
attr.value.oid = SAI_NULL_OBJECT_ID;

EXPECT_EQ(sw.get(SAI_OBJECT_TYPE_SWITCH, "oid:0x2100000000", 1, &attr), SAI_STATUS_SUCCESS);

EXPECT_NE(attr.value.oid, SAI_NULL_OBJECT_ID);

auto boid = attr.value.oid;

auto sboid = sai_serialize_object_id(boid);

sai_object_id_t list[128];

attr.id = SAI_BRIDGE_ATTR_PORT_LIST;

attr.value.objlist.count = 128;
attr.value.objlist.list = list;

EXPECT_EQ(sw.get(SAI_OBJECT_TYPE_BRIDGE, sboid, 1, &attr), SAI_STATUS_SUCCESS);
}
Loading

0 comments on commit 4d7956b

Please sign in to comment.