F5/unison
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

internet: Add ECN test case for SYN retransmission test

Browse Source
This commit is contained in:
Tom Henderson
2020-06-01 08:00:18 -07:00
parent 1b79d59968
commit edd67b60f9
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/internet/test/tcp-syn-connection-failed-test.cc
View File

@@ -39,8 +39,9 @@ public:
/**
* Constructor.
* \param desc Test description.
* \param useEcn Whether to enable ECN.
*/
TcpSynConnectionFailedTest (std::string desc);
TcpSynConnectionFailedTest (std::string desc, bool useEcn);
/**
* \brief Handle a connection failure.
@@ -51,9 +52,10 @@ public:
private:
bool m_connectionFailed{false};
bool m_useEcn {false};
};
TcpSynConnectionFailedTest::TcpSynConnectionFailedTest (std::string desc) : TestCase (desc)
TcpSynConnectionFailedTest::TcpSynConnectionFailedTest (std::string desc, bool useEcn) : TestCase (desc), m_useEcn (useEcn)
{
}
@@ -74,6 +76,11 @@ TcpSynConnectionFailedTest::DoRun ()
TypeId tid = TcpSocketFactory::GetTypeId ();
Ptr<Socket> socket = Socket::CreateSocket (node, tid);
if (m_useEcn)
{
Ptr<TcpSocketBase> tcpSocket = DynamicCast<TcpSocketBase> (socket);
tcpSocket->SetUseEcn (TcpSocketState::On);
}
socket->Bind ();
socket->SetConnectCallback (MakeNullCallback <void, Ptr<Socket>>(),
MakeCallback (&TcpSynConnectionFailedTest::HandleConnectionFailed, this));
@@ -96,7 +103,8 @@ class TcpSynConnectionFailedTestSuite : public TestSuite
public:
TcpSynConnectionFailedTestSuite () : TestSuite ("tcp-syn-connection-failed-test", UNIT)
{
AddTestCase (new TcpSynConnectionFailedTest ("TCP SYN connection failed test"), TestCase::QUICK);
AddTestCase (new TcpSynConnectionFailedTest ("TCP SYN connection failed test no ECN", false), TestCase::QUICK);
AddTestCase (new TcpSynConnectionFailedTest ("TCP SYN connection failed test with ECN", true), TestCase::QUICK);
}
};