OlsrMprTestCase::OlsrMprTestCase ()

  : TestCase ("Check OLSR MPR computing mechanism")

{

}

OlsrMprTestCase::~OlsrMprTestCase ()

{

}

bool

OlsrMprTestCase::DoRun ()

{

  /*

      * Create a 3x3 grid like the following:

      *      3---6---9

      *      |\ /|\ /|

      *      | X | X |

      *      |/ \|/ \|

      *      2---5---8

      *      |\ /|\ /|

      *      | X | X |

      *      |/ \|/ \|

      *      1---4---7

      * PrepareTopology fills all 2-hop neighbors of station 1 and creates a routing protocol

      * We are the station number 2. Obvious, that an only MPR in this case is 5 

      */

   Ptr<RoutingProtocol> m_protocol = CreateObject<RoutingProtocol> ();

   m_protocol->m_mainAddress = Ipv4Address ("10.0.0.2");

   // we fill all possible 2-hop neighborhood

   TwoHopNeighborTuple tuple;

   tuple.expirationTime = Seconds (3600);

   // All neighbor stations which are seen from station 5

   tuple.neighborMainAddr = Ipv4Address ("10.0.0.5");

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.1");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.2");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.3");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.4");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.6");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.7");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.8");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.9");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   // All neighbor stations which are seen from station 4

   tuple.neighborMainAddr = Ipv4Address ("10.0.0.4");

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.1");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.2");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.5");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.8");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.7");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   // All neighbor stations which are seen from station 6

   tuple.neighborMainAddr = Ipv4Address ("10.0.0.6");

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.3");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.2");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.5");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.8");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.9");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   // All neighbor stations which are seen from station 1

   tuple.neighborMainAddr = Ipv4Address ("10.0.0.1");

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.2");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.5");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.4");

   // All neighbor stations which are seen from station 3

   tuple.neighborMainAddr = Ipv4Address ("10.0.0.3");

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.2");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.5");

   m_protocol->m_state.InsertTwoHopNeighborTuple (tuple);

   tuple.twoHopNeighborAddr = Ipv4Address ("10.0.0.6");

  // First, we fill all neighbors

  // If neighbors willingness = OLSR_WILL_DEFAULT, an only station number 5 will be an MPR

  NeighborTuple neigbor;

  neigbor.status = NeighborTuple::STATUS_SYM;

  neigbor.willingness = OLSR_WILL_DEFAULT;

  neigbor.neighborMainAddr = Ipv4Address ("10.0.0.3");

  m_protocol->m_state.InsertNeighborTuple (neigbor);

  neigbor.neighborMainAddr = Ipv4Address ("10.0.0.6");

  m_protocol->m_state.InsertNeighborTuple (neigbor);

  neigbor.neighborMainAddr = Ipv4Address ("10.0.0.5");

  m_protocol->m_state.InsertNeighborTuple (neigbor);

  neigbor.neighborMainAddr = Ipv4Address ("10.0.0.4");

  m_protocol->m_state.InsertNeighborTuple (neigbor);

  neigbor.neighborMainAddr = Ipv4Address ("10.0.0.1");

  m_protocol->m_state.InsertNeighborTuple (neigbor);

  //Now, calculateMPR

  m_protocol->MprComputation ();

  //Check results

  NS_TEST_ASSERT_MSG_EQ (m_protocol->m_state.FindMprAddress (Ipv4Address ("10.0.0.5")), true, "MPR is incorrect!");

  NS_TEST_ASSERT_MSG_EQ (m_protocol->m_state.GetMprSet ().size (), 1 , "An only address must be chosen!\n");

  return false;

}

static class OlsrProtocolTestSuite : public TestSuite

{

public:

  OlsrProtocolTestSuite ();

} g_olsrProtocolTestSuite;

OlsrProtocolTestSuite::OlsrProtocolTestSuite()

  : TestSuite("Routing-OLSR", UNIT)

{

  AddTestCase (new OlsrMprTestCase ());

}

#include "ns3/test.h"

class RoutingProtocol;

/// Testcase for MPR computation mechanism

class OlsrMprTestCase : public TestCase {

public:

  OlsrMprTestCase ();

  ~OlsrMprTestCase ();

  virtual bool DoRun (void);

  ;

};

  friend class OlsrMprTestCase;

MprSet

OlsrState::GetMprSet () const

{

  return m_mprSet;

}

  MprSet GetMprSet () const;