dadada/ns-3-leo
dadada/ns-3-leo
1
0
Fork 0
mirror of https://gitlab.ibr.cs.tu-bs.de/tschuber/ns-3-leo.git synced 2025-07-31 03:00:07 +02:00
Code Issues Releases Wiki Activity

Refactor tests for brevity

Browse source
This commit is contained in:
Tim Schubert 2020-08-29 18:21:09 +02:00
parent 37631d0758
commit 4c008e32f0
5 changed files with 112 additions and 139 deletions
Download patch file Download diff file Expand all files Collapse all files

194
test/leo-mock-channel-test-suite.cc
View file

@ -14,137 +14,99 @@ using namespace ns3;
class LeoMockChannelTransmitUnknownTestCase : public TestCase
{
public:
LeoMockChannelTransmitUnknownTestCase ();
virtual ~LeoMockChannelTransmitUnknownTestCase ();
LeoMockChannelTransmitUnknownTestCase () : TestCase ("transmission from unknown source fails") {}
virtual ~LeoMockChannelTransmitUnknownTestCase () {}
private:
virtual void DoRun (void);
virtual void DoRun (void)
{
Ptr<LeoMockChannel> channel = CreateObject<LeoMockChannel> ();
Packet *packet = new Packet ();
Ptr<Packet> p = Ptr<Packet>(packet);
Address destAddr;
Time txTime;
channel->SetAttribute ("PropagationDelay", StringValue ("ns3::ConstantSpeedPropagationDelayModel"));
channel->SetAttribute ("PropagationLoss", StringValue ("ns3::LeoPropagationLossModel"));
bool result = channel->TransmitStart (p, 10000, destAddr, txTime);
NS_TEST_ASSERT_MSG_EQ (result, false, "unknown source fails to deliver");
}
};
LeoMockChannelTransmitUnknownTestCase::LeoMockChannelTransmitUnknownTestCase ()
: TestCase ("Test transmission to unkown destination")
{
}
LeoMockChannelTransmitUnknownTestCase::~LeoMockChannelTransmitUnknownTestCase ()
{
}
void
LeoMockChannelTransmitUnknownTestCase::DoRun (void)
{
Ptr<LeoMockChannel> channel = CreateObject<LeoMockChannel> ();
Packet *packet = new Packet ();
Ptr<Packet> p = Ptr<Packet>(packet);
Address destAddr;
Time txTime;
channel->SetAttribute ("PropagationDelay", StringValue ("ns3::ConstantSpeedPropagationDelayModel"));
channel->SetAttribute ("PropagationLoss", StringValue ("ns3::LeoPropagationLossModel"));
bool result = channel->TransmitStart (p, 10000, destAddr, txTime);
NS_TEST_ASSERT_MSG_EQ (result, false, "Unknown source fails to deliver");
}
class LeoMockChannelTransmitKnownTestCase : public TestCase
{
public:
LeoMockChannelTransmitKnownTestCase ();
virtual ~LeoMockChannelTransmitKnownTestCase ();
LeoMockChannelTransmitKnownTestCase () : TestCase ("transmission from known source succeeds") {}
virtual ~LeoMockChannelTransmitKnownTestCase () {}
private:
virtual void DoRun (void);
virtual void DoRun (void)
{
Ptr<LeoMockChannel> channel = CreateObject<LeoMockChannel> ();
channel->SetAttribute ("PropagationDelay", StringValue ("ns3::ConstantSpeedPropagationDelayModel"));
channel->SetAttribute ("PropagationLoss", StringValue ("ns3::LeoPropagationLossModel"));
Packet *packet = new Packet ();
Ptr<Packet> p = Ptr<Packet>(packet);
Ptr<Node> srcNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> srcDev = CreateObject<LeoMockNetDevice> ();
srcDev->SetNode (srcNode);
srcDev->SetDeviceType (LeoMockNetDevice::GND);
srcDev->SetAddress (Mac48Address::Allocate ());
int32_t srcId = channel->Attach (srcDev);
Ptr<Node> dstNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> dstDev = CreateObject<LeoMockNetDevice> ();
dstDev->SetNode (dstNode);
dstDev->SetDeviceType (LeoMockNetDevice::SAT);
dstDev->SetAddress (Mac48Address::Allocate ());
channel->Attach (dstDev);
Address destAddr = dstDev->GetAddress ();
Time txTime;
bool result = channel->TransmitStart (p, srcId, destAddr, txTime);
NS_TEST_ASSERT_MSG_EQ (result, true, "known source does not deliver");
}
};
LeoMockChannelTransmitKnownTestCase::LeoMockChannelTransmitKnownTestCase ()
: TestCase ("Test transmission to known destination")
{
}
LeoMockChannelTransmitKnownTestCase::~LeoMockChannelTransmitKnownTestCase ()
{
}
void
LeoMockChannelTransmitKnownTestCase::DoRun (void)
{
Ptr<LeoMockChannel> channel = CreateObject<LeoMockChannel> ();
channel->SetAttribute ("PropagationDelay", StringValue ("ns3::ConstantSpeedPropagationDelayModel"));
channel->SetAttribute ("PropagationLoss", StringValue ("ns3::LeoPropagationLossModel"));
Packet *packet = new Packet ();
Ptr<Packet> p = Ptr<Packet>(packet);
Ptr<Node> srcNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> srcDev = CreateObject<LeoMockNetDevice> ();
srcDev->SetNode (srcNode);
srcDev->SetDeviceType (LeoMockNetDevice::GND);
srcDev->SetAddress (Mac48Address::Allocate ());
int32_t srcId = channel->Attach (srcDev);
Ptr<Node> dstNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> dstDev = CreateObject<LeoMockNetDevice> ();
dstDev->SetNode (dstNode);
dstDev->SetDeviceType (LeoMockNetDevice::SAT);
dstDev->SetAddress (Mac48Address::Allocate ());
channel->Attach (dstDev);
Address destAddr = dstDev->GetAddress ();
Time txTime;
bool result = channel->TransmitStart (p, srcId, destAddr, txTime);
NS_TEST_ASSERT_MSG_EQ (result, true, "Known source delivers");
}
class LeoMockChannelTransmitSpaceGroundTestCase : public TestCase
{
public:
LeoMockChannelTransmitSpaceGroundTestCase ();
virtual ~LeoMockChannelTransmitSpaceGroundTestCase ();
LeoMockChannelTransmitSpaceGroundTestCase () : TestCase ("space to ground transmission succeeds") {}
virtual ~LeoMockChannelTransmitSpaceGroundTestCase () {}
private:
virtual void DoRun (void);
virtual void DoRun (void)
{
Ptr<LeoMockChannel> channel = CreateObject<LeoMockChannel> ();
channel->SetAttribute ("PropagationDelay", StringValue ("ns3::ConstantSpeedPropagationDelayModel"));
channel->SetAttribute ("PropagationLoss", StringValue ("ns3::LeoPropagationLossModel"));
Packet *packet = new Packet ();
Ptr<Packet> p = Ptr<Packet>(packet);
Ptr<Node> srcNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> srcDev = CreateObject<LeoMockNetDevice> ();
srcDev->SetNode (srcNode);
srcDev->SetDeviceType (LeoMockNetDevice::SAT);
srcDev->SetAddress (Mac48Address::Allocate ());
int32_t srcId = channel->Attach (srcDev);
Ptr<Node> dstNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> dstDev = CreateObject<LeoMockNetDevice> ();
dstDev->SetNode (dstNode);
dstDev->SetDeviceType (LeoMockNetDevice::GND);
dstDev->SetAddress (Mac48Address::Allocate ());
channel->Attach (dstDev);
Address destAddr = dstDev->GetAddress ();
Time txTime;
bool result = channel->TransmitStart (p, srcId, destAddr, txTime);
NS_TEST_ASSERT_MSG_EQ (result, true, "space to ground transmission failed");
}
};
LeoMockChannelTransmitSpaceGroundTestCase::LeoMockChannelTransmitSpaceGroundTestCase ()
: TestCase ("Test transmission to known destination")
{
}
LeoMockChannelTransmitSpaceGroundTestCase::~LeoMockChannelTransmitSpaceGroundTestCase ()
{
}
void
LeoMockChannelTransmitSpaceGroundTestCase::DoRun (void)
{
Ptr<LeoMockChannel> channel = CreateObject<LeoMockChannel> ();
channel->SetAttribute ("PropagationDelay", StringValue ("ns3::ConstantSpeedPropagationDelayModel"));
channel->SetAttribute ("PropagationLoss", StringValue ("ns3::LeoPropagationLossModel"));
Packet *packet = new Packet ();
Ptr<Packet> p = Ptr<Packet>(packet);
Ptr<Node> srcNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> srcDev = CreateObject<LeoMockNetDevice> ();
srcDev->SetNode (srcNode);
srcDev->SetDeviceType (LeoMockNetDevice::SAT);
srcDev->SetAddress (Mac48Address::Allocate ());
int32_t srcId = channel->Attach (srcDev);
Ptr<Node> dstNode = CreateObject<Node> ();
Ptr<LeoMockNetDevice> dstDev = CreateObject<LeoMockNetDevice> ();
dstDev->SetNode (dstNode);
dstDev->SetDeviceType (LeoMockNetDevice::GND);
dstDev->SetAddress (Mac48Address::Allocate ());
channel->Attach (dstDev);
Address destAddr = dstDev->GetAddress ();
Time txTime;
bool result = channel->TransmitStart (p, srcId, destAddr, txTime);
NS_TEST_ASSERT_MSG_EQ (result, true, "Space to ground destination delivers");
}
class LeoMockChannelTestSuite : public TestSuite
{
public:
@ -154,11 +116,9 @@ public:
LeoMockChannelTestSuite::LeoMockChannelTestSuite ()
: TestSuite ("leo-mock-channel", UNIT)
{
// TestDuration for TestCase can be QUICK, EXTENSIVE or TAKES_FOREVER
AddTestCase (new LeoMockChannelTransmitUnknownTestCase, TestCase::QUICK);
AddTestCase (new LeoMockChannelTransmitKnownTestCase, TestCase::QUICK);
AddTestCase (new LeoMockChannelTransmitSpaceGroundTestCase, TestCase::QUICK);
}
// Do not forget to allocate an instance of this TestSuite
static LeoMockChannelTestSuite islMockChannelTestSuite;