/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright 2007 University of Washington

 * Copyright (C) 1999, 2000 Kunihiro Ishiguro, Toshiaki Takada

 * 

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Authors:  Tom Henderson (tomhend@u.washington.edu)

 * 

 * Kunihiro Ishigura, Toshiaki Takada (GNU Zebra) are attributed authors

 * of the quagga 0.99.7/src/ospfd/ospf_spf.c code which was ported here

 */

#include "ns3/test.h"

#include "ns3/global-route-manager-impl.h"

#include "ns3/candidate-queue.h"

#include "ns3/simulator.h"

#include <cstdlib> // for rand()

using namespace ns3;

class GlobalRouteManagerImplTestCase : public TestCase

{

public:

  GlobalRouteManagerImplTestCase();

  virtual void DoRun (void);

};

GlobalRouteManagerImplTestCase::GlobalRouteManagerImplTestCase()

  : TestCase ("GlobalRouteManagerImplTestCase")

{

}

void

GlobalRouteManagerImplTestCase::DoRun (void)

{

  CandidateQueue candidate;

  for (int i = 0; i < 100; ++i)

    {

      SPFVertex *v = new SPFVertex;

      v->SetDistanceFromRoot (std::rand () % 100);

      candidate.Push (v);

    }

  for (int i = 0; i < 100; ++i)

    {

      SPFVertex *v = candidate.Pop ();

      delete v;

      v = 0;

    }

  // Build fake link state database; four routers (0-3), 3 point-to-point

  // links

  //

  //   n0

  //      \ link 0

  //       \          link 2

  //        n2 -------------------------n3

  //       /

  //      / link 1

  //    n1

  //

  //  link0:  10.1.1.1/30, 10.1.1.2/30

  //  link1:  10.1.2.1/30, 10.1.2.2/30

  //  link2:  10.1.3.1/30, 10.1.3.2/30

  //

  // Router 0

  GlobalRoutingLinkRecord* lr0 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::PointToPoint,

      "0.0.0.2", // router ID 0.0.0.2

      "10.1.1.1", // local ID

      1);       // metric

  GlobalRoutingLinkRecord* lr1 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::StubNetwork,

      "10.1.1.1",

      "255.255.255.252",

      1);

  GlobalRoutingLSA* lsa0 = new GlobalRoutingLSA ();

  lsa0->SetLSType (GlobalRoutingLSA::RouterLSA);

  lsa0->SetLinkStateId ("0.0.0.0");

  lsa0->SetAdvertisingRouter ("0.0.0.0");

  lsa0->AddLinkRecord (lr0);

  lsa0->AddLinkRecord (lr1);

  // Router 1

  GlobalRoutingLinkRecord* lr2 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::PointToPoint,

      "0.0.0.2",

      "10.1.2.1",

      1);

  GlobalRoutingLinkRecord* lr3 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::StubNetwork,

      "10.1.2.1",

      "255.255.255.252",

      1);

  GlobalRoutingLSA* lsa1 = new GlobalRoutingLSA ();

  lsa1->SetLSType (GlobalRoutingLSA::RouterLSA);

  lsa1->SetLinkStateId ("0.0.0.1");

  lsa1->SetAdvertisingRouter ("0.0.0.1");

  lsa1->AddLinkRecord (lr2);

  lsa1->AddLinkRecord (lr3);

  // Router 2 

  GlobalRoutingLinkRecord* lr4 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::PointToPoint,

      "0.0.0.0",

      "10.1.1.2",

      1);

  GlobalRoutingLinkRecord* lr5 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::StubNetwork,

      "10.1.1.2",

      "255.255.255.252",

      1);

  GlobalRoutingLinkRecord* lr6 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::PointToPoint,

      "0.0.0.1",

      "10.1.2.2",

      1);

  GlobalRoutingLinkRecord* lr7 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::StubNetwork,

      "10.1.2.2",

      "255.255.255.252",

      1);

  GlobalRoutingLinkRecord* lr8 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::PointToPoint,

      "0.0.0.3",

      "10.1.3.2",

      1);

  GlobalRoutingLinkRecord* lr9 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::StubNetwork,

      "10.1.3.2",

      "255.255.255.252",

      1);

  GlobalRoutingLSA* lsa2 = new GlobalRoutingLSA ();

  lsa2->SetLSType (GlobalRoutingLSA::RouterLSA);

  lsa2->SetLinkStateId ("0.0.0.2");

  lsa2->SetAdvertisingRouter ("0.0.0.2");

  lsa2->AddLinkRecord (lr4);

  lsa2->AddLinkRecord (lr5);

  lsa2->AddLinkRecord (lr6);

  lsa2->AddLinkRecord (lr7);

  lsa2->AddLinkRecord (lr8);

  lsa2->AddLinkRecord (lr9);

  // Router 3

  GlobalRoutingLinkRecord* lr10 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::PointToPoint,

      "0.0.0.2",

      "10.1.2.1",

      1);

  GlobalRoutingLinkRecord* lr11 = new GlobalRoutingLinkRecord (

      GlobalRoutingLinkRecord::StubNetwork,

      "10.1.2.1",

      "255.255.255.252",

      1);

  GlobalRoutingLSA* lsa3 = new GlobalRoutingLSA ();

  lsa3->SetLSType (GlobalRoutingLSA::RouterLSA);

  lsa3->SetLinkStateId ("0.0.0.3");

  lsa3->SetAdvertisingRouter ("0.0.0.3");

  lsa3->AddLinkRecord (lr10);

  lsa3->AddLinkRecord (lr11);

  // Test the database 

  GlobalRouteManagerLSDB* srmlsdb = new GlobalRouteManagerLSDB ();

  srmlsdb->Insert (lsa0->GetLinkStateId (), lsa0);

  srmlsdb->Insert (lsa1->GetLinkStateId (), lsa1);

  srmlsdb->Insert (lsa2->GetLinkStateId (), lsa2);

  srmlsdb->Insert (lsa3->GetLinkStateId (), lsa3);

  NS_ASSERT (lsa2 == srmlsdb->GetLSA (lsa2->GetLinkStateId ()));

  // next, calculate routes based on the manually created LSDB

  GlobalRouteManagerImpl* srm = new GlobalRouteManagerImpl ();

  srm->DebugUseLsdb (srmlsdb);  // manually add in an LSDB

  // Note-- this will succeed without any nodes in the topology

  // because the NodeList is empty

  srm->DebugSPFCalculate (lsa0->GetLinkStateId ());  // node n0

  Simulator::Run ();

  Simulator::Destroy ();

  // This delete clears the srm, which deletes the LSDB, which clears 

  // all of the LSAs, which each destroys the attached LinkRecords.

  delete srm;

  // No testing has actually been done other than making sure that this code

  // does not crash

}

static class GlobalRouteManagerImplTestSuite : public TestSuite

{

public:

  GlobalRouteManagerImplTestSuite()

    : TestSuite ("global-route-manager-impl", UNIT)

  {

    AddTestCase (new GlobalRouteManagerImplTestCase (), TestCase::QUICK);

  }

} g_globalRoutingManagerImplTestSuite;