*          Sébastien Deronne <sebastien.deronne@gmail.com> (Case for bug 730)

#include "ns3/nqos-wifi-mac-helper.h"

#include "ns3/yans-wifi-helper.h"

#include "ns3/internet-stack-helper.h"

#include "ns3/ipv4-address-helper.h"

#include "ns3/packet-sink-helper.h"

#include "ns3/on-off-helper.h"

#include "ns3/mobility-helper.h"

#include "ns3/string.h"

/**

 * TODO description

 *

 * See \bugid{730}

 */

class Bug730TestCase : public TestCase

{

public:

  Bug730TestCase ();

  virtual ~Bug730TestCase ();

  virtual void DoRun (void);

private:

  void Receive (std::string context, Ptr<const Packet> p, const Address &adr);

  uint32_t m_received;

};

Bug730TestCase::Bug730TestCase ()

  : TestCase ("Test case for Bug 730"),

    m_received (0)

{

}

Bug730TestCase::~Bug730TestCase ()

{

}

void

Bug730TestCase::Receive (std::string context, Ptr<const Packet> p, const Address &adr)

{

  std::cout << Simulator::Now ().GetSeconds () << ": " << context << " size=" << p->GetSize() << std::endl;

  if ((p->GetSize()==1460) && (Simulator::Now () > Seconds (20)))

    {

      m_received++;

    }

}

void

Bug730TestCase::DoRun (void)

{

  m_received = 0;

  Config::SetDefault ("ns3::WifiRemoteStationManager::FragmentationThreshold", StringValue ("2304"));

  Config::SetDefault ("ns3::TcpSocket::DelAckCount", UintegerValue (2));

  Config::SetDefault ("ns3::TcpSocket::SegmentSize", UintegerValue (1460));

  NodeContainer wifiStaNode;

  wifiStaNode.Create (1);

  NodeContainer wifiApNode;

  wifiApNode.Create(1);

  YansWifiChannelHelper channel = YansWifiChannelHelper::Default ();

  YansWifiPhyHelper phy = YansWifiPhyHelper::Default ();

  phy.SetChannel (channel.Create());

  WifiHelper wifi = WifiHelper::Default ();

  wifi.SetStandard (WIFI_PHY_STANDARD_80211b);

  wifi.SetRemoteStationManager ("ns3::ConstantRateWifiManager",

                                "DataMode", StringValue ("DsssRate1Mbps"));

  NqosWifiMacHelper mac = NqosWifiMacHelper::Default ();

  Ssid ssid = Ssid ("ns-3-ssid");

  mac.SetType ("ns3::StaWifiMac",

               "Ssid", SsidValue (ssid),

               "ActiveProbing", BooleanValue (false));

  NetDeviceContainer staDevices;

  staDevices = wifi.Install (phy, mac, wifiStaNode);

  mac.SetType ("ns3::ApWifiMac",

               "Ssid", SsidValue (ssid),

               "BeaconGeneration", BooleanValue (true));

  NetDeviceContainer apDevices;

  apDevices = wifi.Install (phy, mac, wifiApNode);

  MobilityHelper mobility;

  Ptr<ListPositionAllocator> positionAlloc = CreateObject<ListPositionAllocator> ();

  positionAlloc->Add (Vector (0.0, 0.0, 0.0));

  positionAlloc->Add (Vector (1.0, 0.0, 0.0));

  mobility.SetPositionAllocator (positionAlloc);

  mobility.SetMobilityModel ("ns3::ConstantPositionMobilityModel");

  mobility.Install (wifiApNode);

  mobility.Install (wifiStaNode);

  InternetStackHelper stack;

  stack.Install (wifiApNode);

  stack.Install (wifiStaNode);

  Ipv4AddressHelper address;

  address.SetBase ("10.1.1.0", "255.255.255.0");

  Ipv4InterfaceContainer StaInterface;

  StaInterface = address.Assign (staDevices);

  Ipv4InterfaceContainer ApInterface;

  ApInterface = address.Assign (apDevices);

  Address sinkLocalAddress (InetSocketAddress (Ipv4Address::GetAny (), 21));

  PacketSinkHelper sinkHelper ("ns3::TcpSocketFactory", sinkLocalAddress);

  ApplicationContainer sinkApp = sinkHelper.Install (wifiApNode.Get (0));

  sinkApp.Start (Seconds (1.0));

  sinkApp.Stop (Seconds (51.0));

  OnOffHelper sourceHelper ("ns3::TcpSocketFactory", Address ());

  sourceHelper.SetAttribute ("OnTime", StringValue ("ns3::ConstantRandomVariable[Constant=1]"));

  sourceHelper.SetAttribute ("OffTime", StringValue ("ns3::ConstantRandomVariable[Constant=0]"));

  AddressValue remoteAddress (InetSocketAddress (ApInterface.GetAddress (0), 21));

  sourceHelper.SetAttribute ("Remote", remoteAddress);

  sourceHelper.SetAttribute ("PacketSize", UintegerValue (1460));

  sourceHelper.SetAttribute ("DataRate", StringValue ("10Mb/s"));

  ApplicationContainer sourceApp;

  sourceApp.Add (sourceHelper.Install (wifiStaNode.Get(0)));

  sourceApp.Start (Seconds (1.0));

  sourceApp.Stop (Seconds (51.0));

  Config::Connect ("/NodeList/*/ApplicationList/0/$ns3::PacketSink/Rx", MakeCallback (&Bug730TestCase::Receive, this));

  Simulator::Schedule (Seconds (10.0), Config::Set, "/NodeList/0/DeviceList/0/RemoteStationManager/FragmentationThreshold", StringValue ("800"));

  Simulator::Stop (Seconds(55));

  Simulator::Run ();

  Simulator::Destroy ();

  bool result = (m_received > 0);

  NS_TEST_ASSERT_MSG_EQ (result, true, "");

}

//-----------------------------------------------------------------------------

  AddTestCase (new Bug730TestCase, TestCase::QUICK); //Bug 730