From 6935a997e61eb6397ff00fbc7bf0abed09a9b844 Mon Sep 17 00:00:00 2001
From: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
Date: Mon, 31 Mar 2008 14:34:38 -0700
Subject: [PATCH] do not use InternetNode

---
 src/internet-node/udp-socket.cc | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/internet-node/udp-socket.cc b/src/internet-node/udp-socket.cc
index e1da6bfec..fe86b455e 100644
--- a/src/internet-node/udp-socket.cc
+++ b/src/internet-node/udp-socket.cc
@@ -352,13 +352,13 @@ UdpSocket::ForwardUp (Ptr<Packet> packet, Ipv4Address ipv4, uint16_t port)
 #ifdef RUN_SELF_TESTS
 
 #include "ns3/test.h"
-#include "ns3/internet-node.h"
 #include "ns3/socket-factory.h"
 #include "ns3/udp.h"
 #include "ns3/simulator.h"
 #include "ns3/simple-channel.h"
 #include "ns3/simple-net-device.h"
 #include "ns3/drop-tail-queue.h"
+#include "internet-stack.h"
 #include <string>
 
 namespace ns3 {
@@ -399,7 +399,8 @@ UdpSocketTest::RunTests (void)
   // Create topology
   
   // Receiver Node
-  Ptr<Node> rxNode = CreateObject<InternetNode> ();
+  Ptr<Node> rxNode = CreateObject<Node> ();
+  AddInternetStack (rxNode);
   Ptr<SimpleNetDevice> rxDev1, rxDev2;
   { // first interface
     rxDev1 = CreateObject<SimpleNetDevice> ();
@@ -424,7 +425,8 @@ UdpSocketTest::RunTests (void)
   }
   
   // Sender Node
-  Ptr<Node> txNode = CreateObject<InternetNode> ();
+  Ptr<Node> txNode = CreateObject<Node> ();
+  AddInternetStack (txNode);
   Ptr<SimpleNetDevice> txDev1;
   {
     txDev1 = CreateObject<SimpleNetDevice> ();