Skip to content

Commit

Permalink
Merging TestSessionManager with TestContext.
Browse files Browse the repository at this point in the history
  • Loading branch information
@feasel0
feasel0 committed Apr 26, 2024
1 parent ca19821 commit 20d5280
Showing 1 changed file with 28 additions and 31 deletions.
59 changes: 28 additions & 31 deletions src/transport/tests/TestSessionManager.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -117,14 +117,11 @@ class TestSessMgrCallback : public SessionMessageDelegate
bool LargeMessageSent = false;
};

class TestSessionManager : public ::testing::Test
class TestSessionManager : public ::testing::Test, public TestContext
{
protected:
TestSessionManager() { inContext = new TestContext(); }
~TestSessionManager() { delete inContext; }
void SetUp() { ASSERT_EQ(inContext->Init(), CHIP_NO_ERROR); }
void TearDown() { inContext->Shutdown(); }
TestContext * inContext;
void SetUp() { ASSERT_EQ(this->Init(), CHIP_NO_ERROR); }
void TearDown() { this->Shutdown(); }
};

TEST_F(TestSessionManager, CheckSimpleInitTest)
Expand All @@ -137,7 +134,7 @@ TEST_F(TestSessionManager, CheckSimpleInitTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &gMessageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &gMessageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));
}

Expand Down Expand Up @@ -166,7 +163,7 @@ TEST_F(TestSessionManager, CheckMessageTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &gMessageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &gMessageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));

sessionManager.SetMessageDelegate(&callback);
Expand Down Expand Up @@ -212,7 +209,7 @@ TEST_F(TestSessionManager, CheckMessageTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 1);

// Let's send the max sized message and make sure it is received
Expand All @@ -227,7 +224,7 @@ TEST_F(TestSessionManager, CheckMessageTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 2);

uint16_t large_payload_len = sizeof(LARGE_PAYLOAD);
Expand Down Expand Up @@ -270,7 +267,7 @@ TEST_F(TestSessionManager, SendEncryptedPacketTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &gMessageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &gMessageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));

sessionManager.SetMessageDelegate(&callback);
Expand Down Expand Up @@ -318,7 +315,7 @@ TEST_F(TestSessionManager, SendEncryptedPacketTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 1);

// Reset receive side message counter, or duplicated message will be denied.
Expand All @@ -328,7 +325,7 @@ TEST_F(TestSessionManager, SendEncryptedPacketTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 2);

sessionManager.Shutdown();
Expand Down Expand Up @@ -359,7 +356,7 @@ TEST_F(TestSessionManager, SendBadEncryptedPacketTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &gMessageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &gMessageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));

sessionManager.SetMessageDelegate(&callback);
Expand Down Expand Up @@ -407,7 +404,7 @@ TEST_F(TestSessionManager, SendBadEncryptedPacketTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 1);

/* -------------------------------------------------------------------------------------------*/
Expand All @@ -428,7 +425,7 @@ TEST_F(TestSessionManager, SendBadEncryptedPacketTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), badMessageCounterMsg);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 1);

/* -------------------------------------------------------------------------------------------*/
Expand All @@ -445,7 +442,7 @@ TEST_F(TestSessionManager, SendBadEncryptedPacketTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), badKeyIdMsg);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 1);

/* -------------------------------------------------------------------------------------------*/
Expand All @@ -455,7 +452,7 @@ TEST_F(TestSessionManager, SendBadEncryptedPacketTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 2);

sessionManager.Shutdown();
Expand Down Expand Up @@ -486,7 +483,7 @@ TEST_F(TestSessionManager, SendPacketWithOldCounterTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &gMessageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &gMessageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));

sessionManager.SetMessageDelegate(&callback);
Expand Down Expand Up @@ -533,7 +530,7 @@ TEST_F(TestSessionManager, SendPacketWithOldCounterTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 1);

// Now advance our message counter by 5.
Expand All @@ -550,15 +547,15 @@ TEST_F(TestSessionManager, SendPacketWithOldCounterTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), newMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 2);

// Now resend our original message. It should be rejected as a duplicate.

err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 2);

sessionManager.Shutdown();
Expand Down Expand Up @@ -589,7 +586,7 @@ TEST_F(TestSessionManager, SendPacketWithTooOldCounterTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &gMessageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &gMessageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));
sessionManager.SetMessageDelegate(&callback);

Expand Down Expand Up @@ -635,7 +632,7 @@ TEST_F(TestSessionManager, SendPacketWithTooOldCounterTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 1);

// Now advance our message counter by at least
Expand All @@ -654,15 +651,15 @@ TEST_F(TestSessionManager, SendPacketWithTooOldCounterTest)
err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), newMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 2);

// Now resend our original message. It should be rejected as a duplicate.

err = sessionManager.SendPreparedMessage(aliceToBobSession.Get().Value(), preparedMessage);
EXPECT_EQ(err, CHIP_NO_ERROR);

inContext->DrainAndServiceIO();
this->DrainAndServiceIO();
EXPECT_EQ(callback.ReceiveHandlerCallCount, 2);

sessionManager.Shutdown();
Expand Down Expand Up @@ -694,7 +691,7 @@ TEST_F(TestSessionManager, SessionAllocationTest)
SessionManager sessionManager;

EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &messageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &messageCounterManager,
&deviceStorage1, &fabricTableHolder.GetFabricTable(), sessionKeystore));

// Allocate a session.
Expand Down Expand Up @@ -732,7 +729,7 @@ TEST_F(TestSessionManager, SessionAllocationTest)
sessionManager.~SessionManager();
new (&sessionManager) SessionManager();
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &messageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &messageCounterManager,
&deviceStorage2, &fabricTableHolder.GetFabricTable(), sessionKeystore));

// Allocate a single session so we know what random id we are starting at.
Expand Down Expand Up @@ -830,7 +827,7 @@ TEST_F(TestSessionManager, SessionCounterExhaustedTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &gMessageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &gMessageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));

Transport::PeerAddress peer(Transport::PeerAddress::UDP(addr, CHIP_PORT));
Expand Down Expand Up @@ -906,7 +903,7 @@ TEST_F(TestSessionManager, SessionShiftingTest)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &messageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &messageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));

Transport::PeerAddress peer(Transport::PeerAddress::UDP(addr, CHIP_PORT));
Expand Down Expand Up @@ -983,7 +980,7 @@ TEST_F(TestSessionManager, TestFindSecureSessionForNode)

EXPECT_EQ(CHIP_NO_ERROR, fabricTableHolder.Init());
EXPECT_EQ(CHIP_NO_ERROR,
sessionManager.Init(&inContext->GetSystemLayer(), &inContext->GetTransportMgr(), &messageCounterManager,
sessionManager.Init(&this->GetSystemLayer(), &this->GetTransportMgr(), &messageCounterManager,
&deviceStorage, &fabricTableHolder.GetFabricTable(), sessionKeystore));

Transport::PeerAddress peer(Transport::PeerAddress::UDP(addr, CHIP_PORT));
Expand Down

0 comments on commit 20d5280

Please sign in to comment.