/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
 * Copyright (c) 2011-2015  Regents of the University of California.
 *
 * This file is part of ndnSIM. See AUTHORS for complete list of ndnSIM authors and
 * contributors.
 *
 * ndnSIM is free software: you can redistribute it and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software Foundation,
 * either version 3 of the License, or (at your option) any later version.
 *
 * ndnSIM 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
 * ndnSIM, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
 **/

#include "helper/ndn-strategy-choice-helper.hpp"
#include "helper/ndn-face-container.hpp"
#include "model/ndn-net-device-face.hpp"

#include "ns3/node-container.h"
#include "ns3/point-to-point-net-device.h"
#include "ns3/point-to-point-grid.h"
#include "ns3/node.h"
#include "ns3/core-module.h"
#include "ns3/network-module.h"
#include "ns3/point-to-point-module.h"
#include "ns3/ndnSIM-module.h"

#include "../tests-common.hpp"

namespace ns3 {
namespace ndn {

BOOST_FIXTURE_TEST_SUITE(TestStrategyChoiceHelper, CleanupFixture)

BOOST_AUTO_TEST_CASE(GridScenario)
{
  ofstream file("/tmp/topo3.txt");
  file  << "router\n\n"
        << "#node city  y x mpi-partition\n"
        << "A3  NA  10 10 1\n"
        << "B3  NA  20 10 1\n"
        << "C3  NA  10 20 1\n"
        << "D3  NA  20 20 1\n\n"
        << "link\n\n"
        << "# from  to  capacity  metric  delay queue\n"
        << "A3      B3  10Mbps    100 1ms 500\n"
        << "A3      C3  10Mbps    200 1ms 500\n"
        << "B3      D3  10Mbps    100 1ms 500\n"
        << "C3      D3  10Mbps    200 1ms 500\n";
  file.close();

  AnnotatedTopologyReader topologyReader("");
  topologyReader.SetFileName("/tmp/topo3.txt");
  topologyReader.Read();

  // Install NDN stack on all nodes
  ndn::StackHelper ndnHelper;
  Ptr<FaceContainer> node_faceContainer = ndnHelper.InstallAll();
  GlobalRoutingHelper ndnGlobalRoutingHelper;
  ndnGlobalRoutingHelper.InstallAll();

  // Install different forwarding strategies
  StrategyChoiceHelper::Install(Names::Find<Node>("A3"), "/prefix",
                                "/localhost/nfd/strategy/broadcast");
  StrategyChoiceHelper::Install(Names::Find<Node>("B3"), "/prefix",
                                "/localhost/nfd/strategy/best-route");
  StrategyChoiceHelper::Install(Names::Find<Node>("C3"), "/prefix",
                                "/localhost/nfd/strategy/broadcast");
  StrategyChoiceHelper::Install(Names::Find<Node>("D3"), "/prefix",
                                "/localhost/nfd/strategy/best-route");

  AppHelper consumerHelper("ns3::ndn::ConsumerCbr");
  consumerHelper.SetPrefix("/prefix");
  consumerHelper.SetAttribute("Frequency", StringValue("100")); // 100 interests a second
  consumerHelper.Install(Names::Find<Node>("A3"));

  AppHelper producerHelper("ns3::ndn::Producer");
  producerHelper.SetPrefix("/prefix");
  producerHelper.SetAttribute("PayloadSize", StringValue("1024"));
  producerHelper.Install(Names::Find<Node>("D3"));

  // Add /prefix origins to ndn::GlobalRouter
  ndnGlobalRoutingHelper.AddOrigins("/prefix", Names::Find<Node>("D3"));

  // Calculate and install FIBs
  GlobalRoutingHelper::CalculateRoutes();

  // Node0 is consumer which is connected to intermediate Node1 and Node2

  FaceContainer::Iterator IntermediateNode1Face1Iterator = node_faceContainer->Begin() + 2; // Node1 with strategy broadcast
  FaceContainer::Iterator IntermediateNode1Face2Iterator = node_faceContainer->Begin() + 3;

  FaceContainer::Iterator IntermediateNode2Face1Iterator = node_faceContainer->Begin() + 4; // Node2 with strategy best route
  FaceContainer::Iterator IntermediateNode2Face2Iterator = node_faceContainer->Begin() + 5;

  FaceContainer::Iterator ProducerFaceIterator1 = node_faceContainer->Begin() + 6;
  FaceContainer::Iterator ProducerFaceIterator2 = node_faceContainer->Begin() + 7;

  auto Producer_netDeviceFace1 = std::dynamic_pointer_cast<NetDeviceFace>(node_faceContainer->Get(ProducerFaceIterator1));
  auto Producer_netDeviceFace2 = std::dynamic_pointer_cast<NetDeviceFace>(node_faceContainer->Get(ProducerFaceIterator2));

  auto IntermediateNode1_netDeviceFace1 = std::dynamic_pointer_cast<NetDeviceFace>(node_faceContainer->Get(IntermediateNode1Face1Iterator));
  auto IntermediateNode1_netDeviceFace2 = std::dynamic_pointer_cast<NetDeviceFace>(node_faceContainer->Get(IntermediateNode1Face2Iterator));

  auto IntermediateNode2_netDeviceFace1 = std::dynamic_pointer_cast<NetDeviceFace>(node_faceContainer->Get(IntermediateNode2Face1Iterator));
  auto IntermediateNode2_netDeviceFace2 = std::dynamic_pointer_cast<NetDeviceFace>(node_faceContainer->Get(IntermediateNode2Face2Iterator));

  Simulator::Stop(Seconds(5.0));
  Simulator::Run();

  ::ndn::nfd::FaceStatus producerFace1Status = Producer_netDeviceFace1->getFaceStatus();
  ::ndn::nfd::FaceStatus producerFace2Status = Producer_netDeviceFace2->getFaceStatus();

  ::ndn::nfd::FaceStatus IntermediateNode1Face1Status = IntermediateNode1_netDeviceFace1->getFaceStatus();
  ::ndn::nfd::FaceStatus IntermediateNode1Face2Status = IntermediateNode1_netDeviceFace2->getFaceStatus();

  ::ndn::nfd::FaceStatus IntermediateNode2Face1Status = IntermediateNode2_netDeviceFace1->getFaceStatus();
  ::ndn::nfd::FaceStatus IntermediateNode2Face2Status = IntermediateNode2_netDeviceFace2->getFaceStatus();

  BOOST_CHECK_EQUAL(producerFace1Status.getNInInterests(), 500);
  BOOST_CHECK_EQUAL(producerFace2Status.getNInInterests(), 0);
  BOOST_CHECK_EQUAL(IntermediateNode1Face1Status.getNInInterests(), 500);
  BOOST_CHECK_EQUAL(IntermediateNode1Face2Status.getNInInterests(), 0);
  BOOST_CHECK_EQUAL(IntermediateNode2Face1Status.getNInInterests(), 0);
  BOOST_CHECK_EQUAL(IntermediateNode2Face2Status.getNInInterests(), 0);
}

BOOST_AUTO_TEST_SUITE_END()

} // namespace ndn
} // namespace ns3